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
60
61
62
63
64
65
66
67
68
|
from random import choice
import random
import string
import uuid
from db_op import DB_op, DBO_cypher_query
class DBO_flush_db(DBO_cypher_query):
"""
complete DB flush: remove all nodes & links
"""
def __init__(self):
q_arr = ['match (n)',
'optional match (n)-[r]-()',
'delete n,r'
]
super(DBO_flush_db, self).__init__(q_arr)
class DBO_random_data_generation(DB_op):
def __init__(self, lim_n=128, lim_r=256, prob_link_create=0.3):
"""
generate random DB data
@return: tuple consisting of the random node,link labels generated
"""
assert 2 <= lim_n
super(DBO_random_data_generation, self).__init__()
self.n_label = rand_label()
self.r_label = rand_label()
q_arr = ['with 0 as _', # TODO clean: foreach triggers SyntaxException: otherwise
'foreach (rid in range(0,%d)' % (lim_n - 1),
'|',
'create (:%s {id: \'test-id_\' + toString(rid), n_attr_0:toInt(%d * rand())}))' % (self.n_label, lim_n)
]
self.add_statement(q_arr)
q_arr = ['match (s:%s),(d:%s)' % (self.n_label, self.n_label),
'with s,d',
'limit %d' % (lim_r - 1),
'where rand() < %.2f' % (prob_link_create),
'create (s)-[:%s {l_attr_0:toInt(%d * rand())}]->(d)' % (self.r_label, lim_r)]
self.add_statement(q_arr)
@property
def node_set_label(self):
return self.n_label
@property
def link_set_label(self):
return self.r_label
def gen_random_name(size=8, char_set=string.ascii_uppercase + string.digits):
"""
used for random node generation
"""
return ''.join(random.choice(char_set) for _ in range(size))
def rand_label(prefix='T_', length=8):
"""
return a prefixed random label, where the default prefix is 'T_'
"""
char_set = string.ascii_lowercase + string.ascii_uppercase + string.digits
ret = ''.join([choice(string.ascii_lowercase)] + [choice(char_set) for _ in range(length - 1)])
ret = prefix + ret
return ret
|
