File: invalidate_stresstest.py

package info (click to toggle)
sqlalchemy 1.2.18%2Bds1-2
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 16,080 kB
  • sloc: python: 239,496; ansic: 1,345; makefile: 264; xml: 17
file content (59 lines) | stat: -rw-r--r-- 1,218 bytes parent folder | download
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
52
53
54
55
56
57
58
59
from __future__ import print_function

import logging
import random
import sys

import gevent.monkey

from sqlalchemy import create_engine
from sqlalchemy import event


gevent.monkey.patch_all()  # noqa

logging.basicConfig()  # noqa
# logging.getLogger("sqlalchemy.pool").setLevel(logging.INFO)

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 Exception:
            # 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()