def do_init_db(db): try: self.node_id, self.root_id, self.root_inum, self.store, self.port = yield db.DoFn( "select node.id,root.id,root.inode,node.files,node.port from node,root where root.id=node.root and node.name=${name}", name=node, ) except NoData: raise RuntimeError("data for '%s' is missing" % (self.node,)) nnodes, = yield db.DoFn( "select count(*) from node where root=${root} and id != ${node}", root=self.root_id, node=self.node_id ) self.single_node = not nnodes try: mode, = yield db.DoFn("select mode from inode where id=${inode}", inode=self.root_inum) except NoData: raise RuntimeError("database has not been initialized: inode %d is missing" % (self.inode,)) if mode == 0: yield db.Do( "update inode set mode=${dir} where id=${inode}", dir=stat.S_IFDIR | stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO, inode=self.root_inum, ) self.info = Info() yield self.info._load(db) if self.info.version != DBVERSION: raise RuntimeError("Need database version %s, got %s" % (DBVERSION, self.info.version)) root = SqlInode(self, self.root_inum) yield root._load(db) returnValue(root)
def do_work2(db): queue = DeferredQueue() defs = [] nworkers = len(entries)//5+1 if nworkers > self.nworkers: nworkers = self.nworkers for i in range(nworkers): d = self.fetch(db,queue) d.addErrback(log.err,"fetch()") defs.append(d) workers = set() for id,inum,typ in entries: if not self.running: break trace('copyrun',"%d: %s",inum,typ) self.last_entry = id if typ == 'd': def dt(inum): path = build_path(self.fs.store,inum, create=False) try: os.unlink(path) except EnvironmentError as e: if e.errno != errno.ENOENT: raise yield deferToThread(dt,inum) else: inode = SqlInode(self.fs,inum) yield inode._load(db) if typ == 'f': if inum in workers: trace('copyrun',"%d: in workers",inum,typ) continue workers.add(inum) queue.put((id,inode)) elif typ == 't': if inode.cache: yield inode.cache.trim(inode.size) else: raise RuntimeError("Typ '%s' not found (inode %d)" % (typ,inum)) continue for i in range(nworkers): queue.put(None) yield DeferredList(defs)
def do_work(db): seq = self.seq if seq is None: seq, = yield db.DoFn("select event.id from event where event.typ='s' and event.node in (select id from node where root=${root}) order by event.id desc limit 1", root=self.fs.root_id) last,do_copy = yield db.DoFn("select event,autocopy from node where id=${node}", node=self.fs.node_id) if seq == last: self.seq = None if self.do_readonly: self.do_readonly = False self.fs.readonly = False self.fs.copier.trigger() return if self.do_readonly and self.fs.readonly: upd = "" trace('eventcollect',"Events from %d to %d (initial sync)",last+1,seq) else: upd = " and node != ${node}" trace('eventcollect',"Events from %d to %d",last+1,seq) it = yield db.DoSelect("select event.id,event.inode,event.node,event.typ,event.range from event join inode on inode.id=event.inode where inode.typ='f' and event.id>${min} and event.id<=${max} and event.node in (select id from node where root=${root}"+upd+") order by event.id limit 100", root=self.fs.root_id, node=self.fs.node_id, min=last,max=seq, _empty=True) last = seq n=0 for event,inum,node,typ,r in it: n+=1 last = event if r: r = Range(r) trace('eventcollect',"%d: ev=%d node=%d typ=%s range=%s",inum,event,node,typ, "?" if r is None else str(r)) if typ == 'i' or typ == 's' or typ == 'f': continue inode = SqlInode(self.fs,inum) if inode.inum is None: # locally deleted. # Temporarily create a fake cache record so that it'll be skipped when we continue. trace('eventcollect',"%s: deleted (ev=%d)", inode,eid) yield db.Do("replace into cache(inode,node,event) values(${inode},${node},${event})", inode=inum,node=self.fs.node_id,event=seq) continue yield inode._load(db) if typ == 'c': if node != self.fs.node_id: r = r.replace(self.fs.node_id,node) inode.cache.known.add_range(r,replace=True) elif typ == 't': yield db.Do("replace into todo(node,inode,typ) values(${node},${inode},'t')", inode=inode.inum, node=self.fs.node_id, _empty=True) continue elif typ == 'd': yield db.Do("replace into todo(node,inode,typ) values(${node},${inode},'d')", inode=inode.inum, node=self.fs.node_id, _empty=True) yield db.Do("delete from fail where node=${node} and inode=${inode}", inode=inode.inum, node=self.fs.node_id, _empty=True) continue elif typ == 'n': yield inode.cache.trim(0, do_file=False) if not do_copy: yield db.Do("replace into todo(node,inode,typ) values(${node},${inode},'t')", inode=inode.inum, node=self.fs.node_id, _empty=True) else: raise ValueError("event record %d: unknown type %s" % (event,repr(typ))) self.fs.changer.note(inode.cache,event) #self.fs.changer.now(inode.cache,event,db) # TODO only do this if we don't just cache if do_copy: trace('eventcollect',"TODO: inode %d",inode.inum) yield db.Do("replace into todo(node,inode,typ) values(${node},${inode},'f')", inode=inode.inum, node=self.fs.node_id, _empty=True) if not self.fs.readonly: db.call_committed(self.fs.copier.trigger) # yield inode.cache._close() yield db.Do("update node set event=${event} where id=${node}", node=self.fs.node_id, event=last, _empty=True) if n > 90: self.restart = 2