def create_pent_context():
edges = [KvetchDbEdgeDefinition(
edge_name='user_to_todo_edge',
edge_id=9283,
from_id_attr='user_id')]
pool = KvetchDbSingleConnectionPool(MagnusConn.get_unittest_conn())
shard = KvetchDbShard(pool=pool)
pent_context = PentContext(
config=get_todo_config(),
kvetch=Kvetch(shards=[shard], edges=edges, indexes=[]),
)
return pent_context
async def gen_time_get_things(ids):
conn = MagnusConn.get_unittest_conn()
shard = KvetchDbShard(
pool=KvetchDbSingleConnectionPool(conn),
)
start = timer()
for obj_id in ids:
obj = await shard.gen_object(obj_id)
end = timer()
reps = len(ids)
db_get_microseconds = (end - start) * 1000 * 1000
print('%s microseconds per db get' % (db_get_microseconds/reps))
def db_edge_and_index_shard():
related_edge = KvetchDbEdgeDefinition(edge_name='related_edge',
edge_id=12345,
from_id_attr='related_id')
num_index = KvetchDbIndexDefinition(indexed_attr='num',
indexed_type_id=1000,
sql_type_of_index='INT',
index_name='num_index')
shard = KvetchDbShard(pool=KvetchDbSingleConnectionPool(
MagnusConn.get_unittest_conn()), )
edges = {'related_edge': related_edge}
indexes = [num_index]
drop_shard_db_tables(shard, indexes)
init_shard_db_tables(shard, indexes)
return shard, edges, indexes
def db_single_edge_shard():
related_edge = KvetchDbEdgeDefinition(
edge_name='related_edge',
edge_id=12345,
from_id_attr='related_id',
)
shard = KvetchDbShard(pool=KvetchDbSingleConnectionPool(
MagnusConn.get_unittest_conn()), )
indexes = []
edges = {
'related_edge': related_edge,
}
drop_shard_db_tables(shard, indexes)
init_shard_db_tables(shard, indexes)
return (shard, edges, indexes)
def db_context():
edges = [
KvetchDbEdgeDefinition(
edge_name='user_to_todo_edge',
edge_id=9283,
from_id_attr='user_id',
)
]
shards = [
KvetchDbShard(pool=KvetchDbSingleConnectionPool(
MagnusConn.get_unittest_conn()), )
]
drop_shard_db_tables(shards[0], [])
init_shard_db_tables(shards[0], [])
return PentContext(kvetch=Kvetch(shards=shards, edges=edges, indexes=[]),
config=get_todo_config())
async def gen_time_insert_things(reps):
conn = MagnusConn.get_unittest_conn()
shard = KvetchDbShard(
pool=KvetchDbSingleConnectionPool(conn),
)
drop_shard_db_tables(shard, {})
init_shard_db_tables(shard, {})
start = timer()
ids = []
for i in range(0, reps):
new_id = uuid4()
ids.append(new_id)
data = {'num': i, 'str': 'some string' + str(i)}
await shard.gen_insert_object(new_id, 1000, data)
end = timer()
db_insert_microseconds = (end - start) * 1000 * 1000
print('%s microseconds per db insert' % (db_insert_microseconds/reps))
return ids
def create_hcris_db_kvetch(conn):
shards = [KvetchDbShard(pool=KvetchDbSingleConnectionPool(conn))]
indexes = get_indexes()
init_shard_db_tables(shards[0], indexes)
return Kvetch(shards=shards, edges=get_edges(), indexes=indexes)