redis/tests/helpers/gen_write_load.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
    catch {$r select 9} ;# select 9 will fail in cluster mode

    # 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]