@@ -220,17 +220,26 @@ def prepare_memtier_benchmark_parameters(
220
220
resp_version = None ,
221
221
override_memtier_test_time = 0 ,
222
222
override_test_runs = 1 ,
223
+ unix_socket = "" ,
223
224
):
224
225
arbitrary_command = False
225
226
benchmark_command = [
226
227
full_benchmark_path ,
227
- "--port" ,
228
- f"{ port } " ,
229
- "--server" ,
230
- f"{ server } " ,
231
228
"--json-out-file" ,
232
229
local_benchmark_output_filename ,
233
230
]
231
+ if unix_socket != "" :
232
+ benchmark_command .extend (["--unix-socket" , unix_socket ])
233
+ logging .info (f"Using UNIX SOCKET to connect { unix_socket } " )
234
+ else :
235
+ benchmark_command .extend (
236
+ [
237
+ "--port" ,
238
+ f"{ port } " ,
239
+ "--server" ,
240
+ f"{ server } " ,
241
+ ]
242
+ )
234
243
if password is not None :
235
244
benchmark_command .extend (["--authenticate" , password ])
236
245
if tls_enabled :
@@ -430,6 +439,7 @@ def delete_temporary_files(
430
439
431
440
port = args .db_server_port
432
441
host = args .db_server_host
442
+ unix_socket = args .unix_socket
433
443
password = args .db_server_password
434
444
oss_cluster_api_enabled = args .cluster_mode
435
445
ssl_cert_reqs = "required"
@@ -606,7 +616,7 @@ def delete_temporary_files(
606
616
)
607
617
continue
608
618
if "preload_tool" in benchmark_config ["dbconfig" ]:
609
- if args .skip_tests_with_preload_via_tool is False :
619
+ if args .skip_tests_with_preload_via_tool is True :
610
620
logging .warning (
611
621
"Skipping test {} giving it implies dataset preload via tool" .format (
612
622
test_name
@@ -648,6 +658,7 @@ def delete_temporary_files(
648
658
args .benchmark_local_install ,
649
659
password ,
650
660
oss_cluster_api_enabled ,
661
+ unix_socket ,
651
662
)
652
663
if res is False :
653
664
logging .warning (
@@ -748,6 +759,7 @@ def delete_temporary_files(
748
759
resp_version ,
749
760
override_memtier_test_time ,
750
761
override_test_runs ,
762
+ unix_socket ,
751
763
)
752
764
753
765
if (
@@ -1151,6 +1163,7 @@ def data_prepopulation_step(
1151
1163
benchmark_local_install = False ,
1152
1164
password = None ,
1153
1165
oss_cluster_api_enabled = False ,
1166
+ unix_socket = "" ,
1154
1167
):
1155
1168
result = True
1156
1169
# setup the benchmark
@@ -1193,6 +1206,8 @@ def data_prepopulation_step(
1193
1206
tls_cacert ,
1194
1207
resp_version ,
1195
1208
override_memtier_test_time_preload ,
1209
+ 1 ,
1210
+ unix_socket ,
1196
1211
)
1197
1212
if arbitrary_command is True and oss_cluster_api_enabled :
1198
1213
logging .warning (
0 commit comments