final public releaseskytools_2_1

1 files changed, 30 insertions, 0 deletions

diff --git a/scripts/queue_mover.py b/scripts/queue_mover.py
new file mode 100755
index 00000000..129728a3
--- /dev/null
+++ b/scripts/queue_mover.py
@@ -0,0 +1,30 @@
+#! /usr/bin/env python
+
+# this script simply mover events from one queue to another
+
+import sys, os, pgq, skytools
+
+class QueueMover(pgq.SerialConsumer):
+    def __init__(self, args):
+        pgq.SerialConsumer.__init__(self, "queue_mover", "src_db", "dst_db", args)
+
+        self.dst_queue_name = self.cf.get("dst_queue_name")
+
+    def process_remote_batch(self, db, batch_id, ev_list, dst_db):
+
+        # load data
+        rows = []
+        for ev in ev_list:
+            data = [ev.type, ev.data, ev.extra1, ev.extra2, ev.extra3, ev.extra4, ev.time]
+            rows.append(data)
+            ev.tag_done()
+        fields = ['type', 'data', 'extra1', 'extra2', 'extra3', 'extra4', 'time']
+
+        # insert data
+        curs = dst_db.cursor()
+        pgq.bulk_insert_events(curs, rows, fields, self.dst_queue_name)
+
+if __name__ == '__main__':
+    script = QueueMover(sys.argv[1:])
+    script.start()
+