forgot couple files

2 files changed, 188 insertions, 0 deletions

diff --git a/doc/skytools3.txt b/doc/skytools3.txt
new file mode 100644
index 00000000..6eacdf5e
--- /dev/null
+++ b/doc/skytools3.txt
@@ -0,0 +1,19 @@
+
+Keep old design from Skytools 2.x:
+----------------------------------
+
+* worker process connects to only 2 databases
+* worker process pulls data, no pushing with LISTEN/NOTIFY
+* relaxed attitude about tables
+  - can be added/removed any time.
+  - inital sync happens table-by-table, no attempt is made to keep
+    consistend picture between tables when syncing.
+
+New features in 3.0:
+--------------------
+
+* Queue content are kept in sync between all nodes in node-set.
+* parallel copy - during inital sync several tables can be
+  copied at the same time.
+
+
diff --git a/python/pgq/setinfo.py b/python/pgq/setinfo.py
new file mode 100644
index 00000000..8013e114
--- /dev/null
+++ b/python/pgq/setinfo.py
@@ -0,0 +1,169 @@
+#! /usr/bin/env python
+
+import skytools
+
+__all__ = ['MemberInfo', 'NodeInfo', 'SetInfo',
+        'ROOT', 'BRANCH', 'LEAF', 'COMBINED_ROOT',
+        'COMBINED_BRANCH', 'MERGE_LEAF']
+
+ROOT = 'root'
+BRANCH = 'branch'
+LEAF = 'leaf'
+COMBINED_ROOT = 'combined-root'
+COMBINED_BRANCH = 'combined-branch'
+MERGE_LEAF = 'merge-leaf'
+
+action_map = {
+'process-batch':   {'root':0, 'branch':1, 'leaf':1, 'combined-root':0, 'combined-branch':1, 'merge-leaf-to-root':1, 'merge-leaf-to-branch':1},
+'process-events':  {'root':0, 'branch':1, 'leaf':1, 'combined-root':0, 'combined-branch':1, 'merge-leaf-to-root':1, 'merge-leaf-to-branch':0},
+'copy-events':     {'root':0, 'branch':1, 'leaf':0, 'combined-root':0, 'combined-branch':1, 'merge-leaf-to-root':0, 'merge-leaf-to-branch':0},
+'tick-event':      {'root':0, 'branch':0, 'leaf':0, 'combined-root':0, 'combined-branch':0, 'merge-leaf-to-root':1, 'merge-leaf-to-branch':0},
+'global-wm-event': {'root':1, 'branch':0, 'leaf':0, 'combined-root':1, 'combined-branch':0, 'merge-leaf-to-root':0, 'merge-leaf-to-branch':0},
+'wait-behind':     {'root':0, 'branch':0, 'leaf':0, 'combined-root':0, 'combined-branch':0, 'merge-leaf-to-root':0, 'merge-leaf-to-branch':1},
+'sync-part-pos':   {'root':0, 'branch':0, 'leaf':0, 'combined-root':0, 'combined-branch':1, 'merge-leaf-to-root':0, 'merge-leaf-to-branch':0},
+'local-wm-publish':{'root':0, 'branch':1, 'leaf':1, 'combined-root':0, 'combined-branch':1, 'merge-leaf-to-root':1, 'merge-leaf-to-branch':1},
+}
+
+class MemberInfo:
+    def __init__(self, row):
+        self.name = row['node_name']
+        self.location = row['node_location']
+        self.dead = row['dead']
+
+class NodeInfo:
+    def __init__(self, row, main_worker = True):
+        self.member_map = {}
+        self.main_worker = main_worker
+
+        self.name = row['node_name']
+        self.type = row['node_type']
+        self.queue_name = row['queue_name']
+        self.global_watermark = row['global_watermark']
+        self.local_watermark = row['local_watermark']
+        self.completed_tick = row['completed_tick']
+        self.provider_node = row['provider_node']
+        self.provider_location = row['provider_location']
+        self.paused = row['paused']
+        self.resync = row['resync']
+        self.up_to_date = row['up_to_date']
+        self.combined_set = row['combined_set']
+        self.combined_type = row['combined_type']
+        self.combined_queue = row['combined_queue']
+        self.worker_name = row['worker_name']
+
+        self._row = row
+
+    def need_action(self, action_name):
+        if not self.main_worker:
+            return action_name in ('process-batch', 'process-events')
+
+        typ = self.type
+        if type == 'merge-leaf':
+            if self.target_type == 'combined-branch':
+                typ += "merge-leaf-to-branch"
+            elif self.target_type == 'combined-root':
+                typ += "merge-leaf-to-root"
+            else:
+                raise Exception('bad target type')
+
+        try:
+            return action_map[action_name][typ]
+        except KeyError, d:
+            raise Exception('need_action(name=%s, type=%s) unknown' % (action_name, typ))
+
+    def get_target_queue(self):
+        qname = None
+        if self.type == 'merge-leaf':
+            qname = self.combined_queue
+        else:
+            qname = self.queue_name
+        if qname is None:
+            raise Exception("no target queue")
+        return qname
+
+    def get_infolines(self):
+        return ['somestuff = 100.2', '']
+
+class SetInfo:
+    def __init__(self, set_name, info_row, member_rows):
+        self.root_info = info_row
+        self.set_name = set_name
+        self.member_map = {}
+        self.node_map = {}
+        self.root_name = info_row['node_name']
+        self.root_type = info_row['node_type']
+        self.global_watermark = info_row['global_watermark']
+
+        for r in member_rows:
+            n = MemberInfo(r)
+            self.member_map[n.name] = n
+
+    def get_member(self, name):
+        return self.member_map.get(name)
+
+    def get_node(self, name):
+        return self.node_map.get(name)
+
+    def get_root_node(self):
+        return self.get_node(self.root_name)
+
+    def add_node(self, node):
+        self.node_map[node.name] = node
+
+    _DATAFMT = "%-30s%s"
+    def print_tree(self):
+        self._prepare_tree()
+        root = self.get_root_node()
+        self._tree_calc(root)
+        datalines = self._print_node(root, '', [])
+        for ln in datalines:
+            print self._DATAFMT % (' ', ln)
+
+    def _print_node(self, node, pfx, datalines):
+        for ln in datalines:
+            print self._DATAFMT % (_setpfx(pfx, '|'), ln)
+        datalines = node.get_infolines()
+        print "%s%s" % (_setpfx(pfx, '+--'), node.name)
+
+        for i, n in enumerate(node.child_list):
+            sfx = ((i < len(node.child_list) - 1) and '  |' or '   ')
+            tmppfx = pfx + sfx
+            datalines = self._print_node(n, tmppfx, datalines)
+
+        return datalines
+
+    def _prepare_tree(self):
+        for node in self.node_map.itervalues():
+            node.total_childs = 0
+            node.levels = 0
+            node.child_list = []
+        for node in self.node_map.itervalues():
+            if node.provider_node:
+                p = self.node_map[node.provider_node]
+                p.child_list.append(node)
+
+    def _tree_calc(self, node):
+        total = len(node.child_list)
+        levels = 1
+        for subnode in node.child_list:
+            self._tree_calc(subnode)
+            total += subnode.total_childs
+            if levels < subnode.levels + 1:
+                levels = subnode.levels + 1
+        node.total_childs = total
+        node.levels = levels
+        node.child_list.sort(_cmp_node)
+
+def _setpfx(pfx, sfx):
+    if pfx:
+        pfx = pfx[:-1] + sfx
+    return pfx
+
+
+def _cmp_node(n1, n2):
+    # returns neg if n1 smaller
+    cmp = n1.levels - n2.levels
+    if cmp:
+        return cmp
+    return n1.total_childs - n2.total_childs
+