#!/usr/bin/python2
from __future__ import print_function
import os
import sys
import struct
import xattr
from threading import Thread, Lock
from Queue import Queue
from subprocess import check_output

NUM_THREADS = 30

base_dir = os.path.join(sys.argv[1], "O", "0")
print_lock = Lock()
obj_q = Queue()
ost_index = int(check_output(["zfs", "get", "-o", "value", "-H", "lustre:index", sys.argv[1]]))

def check_obj():
    while True:
        obj_path = obj_q.get(block=True)
        if obj_path is None:
            break
        if not os.path.isfile(obj_path):
            continue
        lma = xattr.getxattr(obj_path, "trusted.lma")
        compat, incompat, seq, oid, version = struct.unpack('<LLQLL', lma)
        if seq == 0x100000000:
            new_seq = seq | (ost_index << 16)
            new_lma = struct.pack('<LLQLL', compat, incompat, new_seq, oid, version)
            xattr.setxattr(obj_path, "trusted.lma", new_lma)
            with print_lock:
                print(os.path.basename(obj_path))

workers = [Thread(target=check_obj) for i in range(NUM_THREADS)]
for w in workers:
    w.start()

for sub_dir in os.listdir(base_dir):
    for obj in os.listdir(os.path.join(base_dir, sub_dir)):
        obj_q.put(os.path.join(base_dir, sub_dir, obj))

for w in workers:
    obj_q.put(None)
for w in workers:
    w.join()