mirror of https://mirror.osredm.com/root/redis.git
52 lines
1.4 KiB
Tcl
52 lines
1.4 KiB
Tcl
#
|
|
# Copyright (c) 2009-Present, Redis Ltd.
|
|
# All rights reserved.
|
|
#
|
|
# Copyright (c) 2024-present, Valkey contributors.
|
|
# All rights reserved.
|
|
#
|
|
# Licensed under your choice of (a) the Redis Source Available License 2.0
|
|
# (RSALv2); or (b) the Server Side Public License v1 (SSPLv1); or (c) the
|
|
# GNU Affero General Public License v3 (AGPLv3).
|
|
#
|
|
# Portions of this file are available under BSD3 terms; see REDISCONTRIBUTIONS for more information.
|
|
#
|
|
|
|
source tests/support/redis.tcl
|
|
|
|
set ::tlsdir "tests/tls"
|
|
|
|
# Continuously sends SET commands to the server. If key is omitted, a random key
|
|
# is used for every SET command. The value is always random.
|
|
proc gen_write_load {host port seconds tls {key ""} {size 0} {sleep 0}} {
|
|
set start_time [clock seconds]
|
|
set r [redis $host $port 1 $tls]
|
|
$r client setname LOAD_HANDLER
|
|
$r select 9
|
|
|
|
# fixed size value
|
|
if {$size != 0} {
|
|
set value [string repeat "x" $size]
|
|
}
|
|
|
|
while 1 {
|
|
if {$size == 0} {
|
|
set value [expr rand()]
|
|
}
|
|
|
|
if {$key == ""} {
|
|
$r set [expr rand()] $value
|
|
} else {
|
|
$r set $key $value
|
|
}
|
|
if {[clock seconds]-$start_time > $seconds} {
|
|
exit 0
|
|
}
|
|
if {$sleep ne 0} {
|
|
after $sleep
|
|
}
|
|
}
|
|
}
|
|
|
|
gen_write_load [lindex $argv 0] [lindex $argv 1] [lindex $argv 2] [lindex $argv 3] [lindex $argv 4] [lindex $argv 5] [lindex $argv 6]
|