File: invalidate_stresstest.py

package info (click to toggle)
sqlalchemy 1.0.15%2Bds1-1
  • links: PTS, VCS
  • area: main
  • in suites: stretch
  • size: 13,056 kB
  • ctags: 26,600
  • sloc: python: 169,901; ansic: 1,346; makefile: 260; xml: 17
file content (51 lines) | stat: -rw-r--r-- 1,177 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
import gevent.monkey
gevent.monkey.patch_all()

import logging
logging.basicConfig()
#logging.getLogger("sqlalchemy.pool").setLevel(logging.INFO)
from sqlalchemy import event
import random
import sys
from sqlalchemy import create_engine
import traceback

engine = create_engine('mysql+pymysql://scott:tiger@localhost/test',
                       pool_size=50, max_overflow=0)

@event.listens_for(engine, "connect")
def conn(*arg):
    print "new connection!"

def worker():
    while True:
        conn = engine.connect()
        try:
            trans = conn.begin()
            for i in range(5):
                conn.execute("SELECT 1+1")
                gevent.sleep(random.random() * 1.01)

        except:
            #traceback.print_exc()
            sys.stderr.write('X')
        else:
            conn.close()
            sys.stderr.write('.')


def main():
    for i in range(40):
        gevent.spawn(worker)

    gevent.sleep(3)

    while True:
        result = list(engine.execute("show processlist"))
        engine.execute("kill %d" % result[-2][0])
        print "\n\n\n BOOM!!!!! \n\n\n"
        gevent.sleep(5)
        print(engine.pool.status())


main()