pymssql_gevent_test.py

import datetime
import gevent
import gevent.socket
import os
import pymssql
#import random

server = os.getenv("PYMSSQL_TEST_SERVER")
user = os.getenv("PYMSSQL_TEST_USERNAME")
password = os.getenv("PYMSSQL_TEST_PASSWORD")
database = os.getenv("PYMSSQL_TEST_DATABASE")

def run(num):
#    gevent.sleep(random.randint(0, 2) * 0.001)

    now = datetime.datetime.now()
    print("%s connecting at time: %s" % (num, now))
    conn = pymssql.connect(host=server,
                           database=database,
                           user=user,
                           password=password)
    cur = conn.cursor()
    cur.execute("""
    WAITFOR DELAY '00:00:05'  -- sleep for 5 seconds
    SELECT CURRENT_TIMESTAMP
    """)
    row = cur.fetchone()
    print("    CURRENT_TIMESTAMP = %r" % (row[0],))
    conn.close()

def do_test():
    greenlets = []

    dt1 = datetime.datetime.now()

    for i in range(5):
        gevent.sleep(1)
        greenlets.append(gevent.spawn(run, i))

    gevent.joinall(greenlets)

    dt2 = datetime.datetime.now()

    print("Done running - elapsed time: %s" % (dt2 - dt1))


print("**** Running test with NO wait_callback...\n")
do_test()

print("\n***** Running test WITH gevent.sleep wait_callback...\n")
def wait_callback(read_fileno):
    # print("    *** wait_callback called with read_fileno = %r" % (read_fileno,))
    gevent.sleep()

pymssql.set_wait_callback(wait_callback)

do_test()

print("\n***** Running test WITH gevent.socket.wait_read wait_callback...\n")
def wait_callback(read_fileno):
    # print("    *** wait_callback called with read_fileno = %r" % (read_fileno,))
    gevent.socket.wait_read(read_fileno)

pymssql.set_wait_callback(wait_callback)

do_test()