1
- import redis
2
1
import time
3
- from functools import wraps
4
2
from argparse import ArgumentParser
3
+ from functools import wraps
4
+
5
+ import redis
5
6
6
7
7
8
def parse_args ():
8
9
parser = ArgumentParser ()
9
- parser .add_argument ('-n' ,
10
- type = int ,
11
- help = 'Total number of requests (default 100000)' ,
12
- default = 100000 )
13
- parser .add_argument ('-P' ,
14
- type = int ,
15
- help = ('Pipeline <numreq> requests.'
16
- ' Default 1 (no pipeline).' ),
17
- default = 1 )
18
- parser .add_argument ('-s' ,
19
- type = int ,
20
- help = 'Data size of SET/GET value in bytes (default 2)' ,
21
- default = 2 )
10
+ parser .add_argument (
11
+ "-n" , type = int , help = "Total number of requests (default 100000)" , default = 100000
12
+ )
13
+ parser .add_argument (
14
+ "-P" ,
15
+ type = int ,
16
+ help = ("Pipeline <numreq> requests." " Default 1 (no pipeline)." ),
17
+ default = 1 ,
18
+ )
19
+ parser .add_argument (
20
+ "-s" ,
21
+ type = int ,
22
+ help = "Data size of SET/GET value in bytes (default 2)" ,
23
+ default = 2 ,
24
+ )
22
25
23
26
args = parser .parse_args ()
24
27
return args
@@ -45,15 +48,16 @@ def wrapper(*args, **kwargs):
45
48
start = time .monotonic ()
46
49
ret = func (* args , ** kwargs )
47
50
duration = time .monotonic () - start
48
- if ' num' in kwargs :
49
- count = kwargs [' num' ]
51
+ if " num" in kwargs :
52
+ count = kwargs [" num" ]
50
53
else :
51
54
count = args [1 ]
52
- print (f' { func .__name__ } - { count } Requests' )
53
- print (f' Duration = { duration } ' )
54
- print (f' Rate = { count / duration } ' )
55
+ print (f" { func .__name__ } - { count } Requests" )
56
+ print (f" Duration = { duration } " )
57
+ print (f" Rate = { count / duration } " )
55
58
print ()
56
59
return ret
60
+
57
61
return wrapper
58
62
59
63
@@ -62,9 +66,9 @@ def set_str(conn, num, pipeline_size, data_size):
62
66
if pipeline_size > 1 :
63
67
conn = conn .pipeline ()
64
68
65
- set_data = 'a' .ljust (data_size , '0' )
69
+ set_data = "a" .ljust (data_size , "0" )
66
70
for i in range (num ):
67
- conn .set (f' set_str:{ i } ' , set_data )
71
+ conn .set (f" set_str:{ i } " , set_data )
68
72
if pipeline_size > 1 and i % pipeline_size == 0 :
69
73
conn .execute ()
70
74
@@ -79,7 +83,7 @@ def set_int(conn, num, pipeline_size, data_size):
79
83
80
84
set_data = 10 ** (data_size - 1 )
81
85
for i in range (num ):
82
- conn .set (f' set_int:{ i } ' , set_data )
86
+ conn .set (f" set_int:{ i } " , set_data )
83
87
if pipeline_size > 1 and i % pipeline_size == 0 :
84
88
conn .execute ()
85
89
@@ -93,7 +97,7 @@ def get_str(conn, num, pipeline_size, data_size):
93
97
conn = conn .pipeline ()
94
98
95
99
for i in range (num ):
96
- conn .get (f' set_str:{ i } ' )
100
+ conn .get (f" set_str:{ i } " )
97
101
if pipeline_size > 1 and i % pipeline_size == 0 :
98
102
conn .execute ()
99
103
@@ -107,7 +111,7 @@ def get_int(conn, num, pipeline_size, data_size):
107
111
conn = conn .pipeline ()
108
112
109
113
for i in range (num ):
110
- conn .get (f' set_int:{ i } ' )
114
+ conn .get (f" set_int:{ i } " )
111
115
if pipeline_size > 1 and i % pipeline_size == 0 :
112
116
conn .execute ()
113
117
@@ -121,7 +125,7 @@ def incr(conn, num, pipeline_size, *args, **kwargs):
121
125
conn = conn .pipeline ()
122
126
123
127
for i in range (num ):
124
- conn .incr (' incr_key' )
128
+ conn .incr (" incr_key" )
125
129
if pipeline_size > 1 and i % pipeline_size == 0 :
126
130
conn .execute ()
127
131
@@ -136,7 +140,7 @@ def lpush(conn, num, pipeline_size, data_size):
136
140
137
141
set_data = 10 ** (data_size - 1 )
138
142
for i in range (num ):
139
- conn .lpush (' lpush_key' , set_data )
143
+ conn .lpush (" lpush_key" , set_data )
140
144
if pipeline_size > 1 and i % pipeline_size == 0 :
141
145
conn .execute ()
142
146
@@ -150,7 +154,7 @@ def lrange_300(conn, num, pipeline_size, data_size):
150
154
conn = conn .pipeline ()
151
155
152
156
for i in range (num ):
153
- conn .lrange (' lpush_key' , i , i + 300 )
157
+ conn .lrange (" lpush_key" , i , i + 300 )
154
158
if pipeline_size > 1 and i % pipeline_size == 0 :
155
159
conn .execute ()
156
160
@@ -163,7 +167,7 @@ def lpop(conn, num, pipeline_size, data_size):
163
167
if pipeline_size > 1 :
164
168
conn = conn .pipeline ()
165
169
for i in range (num ):
166
- conn .lpop (' lpush_key' )
170
+ conn .lpop (" lpush_key" )
167
171
if pipeline_size > 1 and i % pipeline_size == 0 :
168
172
conn .execute ()
169
173
if pipeline_size > 1 :
@@ -175,17 +179,15 @@ def hmset(conn, num, pipeline_size, data_size):
175
179
if pipeline_size > 1 :
176
180
conn = conn .pipeline ()
177
181
178
- set_data = {'str_value' : 'string' ,
179
- 'int_value' : 123456 ,
180
- 'float_value' : 123456.0 }
182
+ set_data = {"str_value" : "string" , "int_value" : 123456 , "float_value" : 123456.0 }
181
183
for i in range (num ):
182
- conn .hmset (' hmset_key' , set_data )
184
+ conn .hmset (" hmset_key" , set_data )
183
185
if pipeline_size > 1 and i % pipeline_size == 0 :
184
186
conn .execute ()
185
187
186
188
if pipeline_size > 1 :
187
189
conn .execute ()
188
190
189
191
190
- if __name__ == ' __main__' :
192
+ if __name__ == " __main__" :
191
193
run ()
0 commit comments