gistfile1.py

from __future__ import print_function

import signal

import gevent
import zmq.green as zmq
import psycopg2
from psycogreen.gevent import patch_psycopg

patch_psycopg()


def main():
    job1 = gevent.spawn(greenloop, 'tcp://*:3333')
    job2 = gevent.spawn(greenloop, 'tcp://*:3334')
    job3 = gevent.spawn(greenpsycopg)

    gevent.signal(signal.SIGTERM, gevent.killall, [job1, job2, job3])

    gevent.joinall([job1, job2, job3])


def greenpsycopg():
    connection = psycopg2.connect(
        database='test', user='postgres', password='postgres',
        host='localhost', port=5433)
    cur = connection.cursor()

    while True:
        cur.execute('SELECT pg_sleep(5), 1 as r')
        print(cur.fetchall())


def greenloop(address):
    ctx = zmq.Context()
    socket = ctx.socket(zmq.PULL)
    socket.bind(address)

    while True:
        try:
            msg = socket.recv()
        except gevent.GreenletExit:
            print('EXIT', address)
            return

        print(address, msg)


main()