zzzeek/test.py

## test.py
import asyncio
import contextlib
import random

import yappi

# install sqlalchemy from github master:
# pip install git+https://github.com/sqlalchemy/sqlalchemy/
from sqlalchemy import cast
from sqlalchemy import Column
from sqlalchemy import Integer
from sqlalchemy import MetaData
from sqlalchemy import String
from sqlalchemy import Table
from sqlalchemy.ext.asyncio import create_async_engine


@contextlib.contextmanager
def profiled():
    yappi.set_context_backend("greenlet")
    yappi.set_clock_type("cpu")

    yappi.start()
    yield
    yappi.stop()

    print("## Function stats:")
    yappi.get_func_stats().print_all()

    print("\n## Greenlet stats:")
    yappi.get_greenlet_stats().print_all()


meta = MetaData()

t1 = Table(
    "t1", meta, Column("id", Integer, primary_key=True), Column("name", String)
)


async def async_setup(engine):
    # engine is an instance of AsyncEngine

    # conn is an instance of AsyncConnection
    async with engine.begin() as conn:
        await conn.run_sync(meta.drop_all)
        await conn.run_sync(meta.create_all)

        await conn.execute(
            t1.insert(), [{"name": "some name %d" % i} for i in range(500)]
        )


async def run_queries(engine, index):
    async with engine.connect() as conn:

        result = await conn.execute(
            t1.select()
            .where(cast(t1.c.name, String).like("some name%"))
            .where(t1.c.id % 17 == index)
        )

        rows = result.fetchall()
        assert len(rows) >= 29

async def async_bench(engine):
    tasks = [
        asyncio.create_task(run_queries(engine, random.randint(1, 16)))
        for i in range(1000)
    ]
    for task in tasks:
        await task


engine = create_async_engine(
    "postgresql+asyncpg://scott:tiger@localhost/test",
)

asyncio.get_event_loop().run_until_complete(async_setup(engine))

with profiled():
    asyncio.get_event_loop().run_until_complete(async_bench(engine))
	import asyncio
	import contextlib
	import random

	import yappi

	# install sqlalchemy from github master:
	# pip install git+https://github.com/sqlalchemy/sqlalchemy/
	from sqlalchemy import cast
	from sqlalchemy import Column
	from sqlalchemy import Integer
	from sqlalchemy import MetaData
	from sqlalchemy import String
	from sqlalchemy import Table
	from sqlalchemy.ext.asyncio import create_async_engine


	@contextlib.contextmanager
	def profiled():
	yappi.set_context_backend("greenlet")
	yappi.set_clock_type("cpu")

	yappi.start()
	yield
	yappi.stop()

	print("## Function stats:")
	yappi.get_func_stats().print_all()

	print("\n## Greenlet stats:")
	yappi.get_greenlet_stats().print_all()


	meta = MetaData()

	t1 = Table(
	"t1", meta, Column("id", Integer, primary_key=True), Column("name", String)
	)


	async def async_setup(engine):
	# engine is an instance of AsyncEngine

	# conn is an instance of AsyncConnection
	async with engine.begin() as conn:
	await conn.run_sync(meta.drop_all)
	await conn.run_sync(meta.create_all)

	await conn.execute(
	t1.insert(), [{"name": "some name %d" % i} for i in range(500)]
	)


	async def run_queries(engine, index):
	async with engine.connect() as conn:

	result = await conn.execute(
	t1.select()
	.where(cast(t1.c.name, String).like("some name%"))
	.where(t1.c.id % 17 == index)
	)

	rows = result.fetchall()
	assert len(rows) >= 29

	async def async_bench(engine):
	tasks = [
	asyncio.create_task(run_queries(engine, random.randint(1, 16)))
	for i in range(1000)
	]
	for task in tasks:
	await task


	engine = create_async_engine(
	"postgresql+asyncpg://scott:tiger@localhost/test",
	)

	asyncio.get_event_loop().run_until_complete(async_setup(engine))

	with profiled():
	asyncio.get_event_loop().run_until_complete(async_bench(engine))