前言

在linux上我們會有一些批量插入redis mock數據的需求，下面通過bash腳本來自動化完成整個過程。

步驟

1.安裝腳本依賴redis-cli（Ubuntu）

（1）安裝redis-cli：運行以下命令以安裝Redis命令行工具：

sudo apt install redis-tools

（2）驗證redis-cli安裝：安裝完成后，你可以運行以下命令來驗證redis-cli是否成功安裝：

redis-cli --version

2.腳本詳情

（1）腳本支持string、zset、hash 3種數據類型的插入、讀取、刪除操作

#!/bin/bash

redis="$1"
operation="$2"
iterations="$3"

# Function to write string keys
write_string_keys() {
    for i in $(seq 1 "$iterations"); do
        key="key_$i"
        $redis set "$key" "$key"
    done
}

# Function to query string keys
query_string_keys() {
    for i in $(seq 1 "$iterations"); do
        key="key_$i"
        $redis get "$key"
    done
}

# Function to clean up string keys
cleanup_string_keys() {
    for i in $(seq 1 "$iterations"); do
        key="key_$i"
        $redis del "$key"
    done
}

# Function to write Zset keys
write_zset_keys() {
    for i in $(seq 1 "$iterations"); do
        zset_name="myzset$i"
        for j in $(seq 1 100); do
            key="key_$j"
            value="$zset_name-value_$j"
            $redis zadd "$zset_name" "$j" "$key:$value"
        done
    done
}

# Function to query Zset keys
query_zset_keys() {
    for i in $(seq 1 "$iterations"); do
        zset_name="myzset$i"
        for j in $(seq 1 100); do
            key="key_$j"
            value="$zset_name-value_$j"
            $redis zscore "$zset_name" "$key:$value"
        done
    done
}

# Function to clean up Zset keys
cleanup_zset_keys() {
    for i in $(seq 1 "$iterations"); do
        zset_name="myzset$i"
        $redis del "$zset_name"
    done
}

# Function to write hash keys
write_hash_keys() {
    for i in $(seq 1 "$iterations"); do
        hash_name="myhash$i"
        for j in $(seq 1 100); do
            key="key_$j"
            field="field_$j"
            value="$hash_name-value_$j"
            $redis hset "$hash_name" "$field:$key" "$value"
        done
    done
}

# Function to query hash keys
query_hash_keys() {
    for i in $(seq 1 "$iterations"); do
        hash_name="myhash$i"
        for j in $(seq 1 100); do
            key="key_$j"
            field="field_$j"
            value="$hash_name-value_$j"
            $redis hget "$hash_name" "$field:$key"
        done
    done
}

# Function to clean up hash keys
cleanup_hash_keys() {
    for i in $(seq 1 "$iterations"); do
        hash_name="myhash$i"
        $redis del "$hash_name"
    done
}

# Execute the selected operation
case "$operation" in
"write_string")
    write_string_keys
    ;;
"query_string")
    query_string_keys
    ;;
"cleanup_string")
    cleanup_string_keys
    ;;
"write_zset")
    write_zset_keys
    ;;
"query_zset")
    query_zset_keys
    ;;
"cleanup_zset")
    cleanup_zset_keys
    ;;
"write_hash")
    write_hash_keys
    ;;
"query_hash")
    query_hash_keys
    ;;
"cleanup_hash")
    cleanup_hash_keys
    ;;
*)
    echo "Invalid operation. Available options: write_string, query_string, cleanup_string, write_zset, query_zset, cleanup_zset, write_hash, query_hash, cleanup_hash"
    exit 1
    ;;
esac

（2）把上面的腳本保存為linux?redis_mock.sh文件

（3）使用命令

自己把“redis-cli -h 127.0.0.1 -p 6379”，放到${host}的位置，然后執行

# redis: redis-cli -h 127.0.0.1 -p 6379

# 寫入
sudo sh redis_mock.sh "${host}" write_string 100
sudo sh redis_mock.sh "${host}" write_zset 100
sudo sh redis_mock.sh "${host}" write_hash 100

# 查詢
sudo sh redis_mock.sh "${host}" query_string 100
sudo sh redis_mock.sh "${host}" query_zset 100
sudo sh redis_mock.sh "${host}" query_hash 100

# 查詢并保存到本地
sudo touch a.txt 
sudo chmod 777 a.txt
cat /dev/null > a.txt

sudo sh redis_mock.sh "${host}" query_string 100 | tee a.txt
sudo sh redis_mock.sh "${host}" query_zset 100 | tee a.txt
sudo sh redis_mock.sh "${host}" query_hash 100 | tee a.txt

# 統計本地查詢到文件的結果
統計查到的String類型條數：grep -o "key_" a.txt | wc -l
統計查到的hash、zset條數：grep -vE '^$' a.txt | wc -l

# 清理
sudo sh redis_mock.sh "${host}" cleanup_string 100
sudo sh redis_mock.sh "${host}" cleanup_zset 100
sudo sh redis_mock.sh "${host}" cleanup_hash 100

總結

一個臨時在linux上對redis生成小批量mock數據的腳本命令，優點是簡單自動化，大數據量不推薦。