Skip to content

Commit 376dfa3

Browse files
committed
2.19.08.08.01 添加对 text 类型的支持
1 parent 9d6cb84 commit 376dfa3

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

bin/mtls-perf-bench

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@ def parse_cmd_arags() -> argparse.ArgumentParser:
5050
parser.add_argument('--ints',type=int,default=0,help="int column counts")
5151
parser.add_argument('--floats',type=int,default=0,help="float column counts")
5252
parser.add_argument('--doubles',type=int,default=0,help='double column counts')
53+
parser.add_argument('--texts',type=int,default=0,help="text column counts")
5354
parser.add_argument('--varchars',type=int,default=0,help="varchar column counts")
5455
parser.add_argument('--varchar-length',type=int,default=128,help="varchar column length default 128")
5556
parser.add_argument('--decimals',type=int,default=0,help="decimal column counts")
@@ -110,8 +111,12 @@ def create(args:argparse.ArgumentParser):
110111
logger.error("decimal-precision argument must big then 0")
111112
sys.exit(3)
112113
for d in range(args.decimals):
113-
columns.append(f"dm{d} decimal({args.decimal_precision},{args.decimal_scale})")
114-
114+
columns.append(f"dm{d} decimal({args.decimal_precision},{args.decimal_scale}) not null")
115+
# 检查 text 字段的数量
116+
if args.texts >=1:
117+
for t in range(args.texts):
118+
columns.append(f"t{t} text not null")
119+
115120
# 如果没有指定任何类型的列,那么直接退出
116121
if len(columns) == 1 and args.auto_primary_key==True:
117122
logger.error(f"do not have any columns in table {args.database}.{args.table}")
@@ -204,6 +209,10 @@ def insert_sql(args:argparse.ArgumentParser) -> tuple:
204209
if args.decimals >= 1:
205210
for dm in range(args.decimals):
206211
columns.append( (f"dm{dm}",get_float_value) )
212+
213+
if args.texts >= 1:
214+
for t in range(args.texts):
215+
columns.append( (f"t{t}",get_str_value) )
207216

208217
if len(columns) == 0:
209218
logger.error(f"columns counts equal 0")
@@ -257,6 +266,7 @@ def drop(args):
257266
logger.error(str(err))
258267
sys.exit(3)
259268

269+
260270

261271

262272
def main():

0 commit comments

Comments
 (0)