ETCD 压力测试脚本

#!/bin/bash

# etcd 集群手动性能压测

# 用途：使用基本的 put/get 命令进行性能测试

set -e

ENDPOINTS=""

ETCD_USER=""

ETCD_PASSWORD=""

# 颜色输出

RED='\033[0;31m'

GREEN='\033[0;32m'

YELLOW='\033[1;33m'

BLUE='\033[0;34m'

CYAN='\033[0;36m'

NC='\033[0m'

echo -e "${BLUE}╔════════════════════════════════════════╗${NC}"

echo -e "${BLUE}║ etcd 集群性能压测（手动模式） ║${NC}"

echo -e "${BLUE}╚════════════════════════════════════════╝${NC}"

echo ""

# 检测是否启用了认证

echo -e "${CYAN}检查是否启用认证...${NC}"

if docker exec etcd1 etcdctl --endpoints=${ENDPOINTS} get /test 2>&1 | grep -q "user name is empty"; then

echo -e "${YELLOW}检测到已启用认证${NC}"

AUTH_PARAMS="--user=${ETCD_USER}:${ETCD_PASSWORD}"

else

echo -e "${YELLOW}未启用认证${NC}"

AUTH_PARAMS=""

fi

echo ""

# 测试 1: 写入性能测试

echo -e "${GREEN}═══════════════════════════════════════${NC}"

echo -e "${GREEN}测试 1: 写入性能测试${NC}"

echo -e "${GREEN}═══════════════════════════════════════${NC}"

echo -e "${CYAN}写入 1000 个 key-value 对...${NC}"

START_TIME=$(date +%s.%N)

for i in {1..1000}; do

docker exec etcd1 etcdctl put \

--endpoints=${ENDPOINTS} \

${AUTH_PARAMS} \

/benchmark/test/key${i} "value${i}" > /dev/null 2>&1

# 每 100 个显示一次进度

if [ $((i % 100)) -eq 0 ]; then

echo -ne "\r${YELLOW}进度: ${i}/1000${NC}"

fi

done

END_TIME=$(date +%s.%N)

DURATION=$(echo "$END_TIME - $START_TIME" | bc)

QPS=$(echo "scale=2; 1000 / $DURATION" | bc)

echo -ne "\r"

echo -e "${GREEN}✓ 写入完成${NC}"

echo -e " 总耗时: ${DURATION} 秒"

echo -e " QPS: ${QPS} 次/秒"

echo ""

sleep 1

# 测试 2: 单点读性能测试

echo -e "${GREEN}═══════════════════════════════════════${NC}"

echo -e "${GREEN}测试 2: 单点读性能测试${NC}"

echo -e "${GREEN}═══════════════════════════════════════${NC}"

echo -e "${CYAN}读取 1000 次...${NC}"

START_TIME=$(date +%s.%N)

for i in {1..1000}; do

docker exec etcd1 etcdctl get \

--endpoints=${ENDPOINTS} \

${AUTH_PARAMS} \

/benchmark/test/key${i} > /dev/null 2>&1

if [ $((i % 100)) -eq 0 ]; then

echo -ne "\r${YELLOW}进度: ${i}/1000${NC}"

fi

done

END_TIME=$(date +%s.%N)

DURATION=$(echo "$END_TIME - $START_TIME" | bc)

QPS=$(echo "scale=2; 1000 / $DURATION" | bc)

echo -ne "\r"

echo -e "${GREEN}✓ 读取完成${NC}"

echo -e " 总耗时: ${DURATION} 秒"

echo -e " QPS: ${QPS} 次/秒"

echo ""

sleep 1

# 测试 3: 范围查询性能

echo -e "${GREEN}═══════════════════════════════════════${NC}"

echo -e "${GREEN}测试 3: 范围查询性能测试${NC}"

echo -e "${GREEN}═══════════════════════════════════════${NC}"

echo -e "${CYAN}执行 100 次前缀查询...${NC}"

START_TIME=$(date +%s.%N)

for i in {1..100}; do

docker exec etcd1 etcdctl get \

--endpoints=${ENDPOINTS} \

${AUTH_PARAMS} \

--prefix \

/benchmark/test/ > /dev/null 2>&1

if [ $((i % 10)) -eq 0 ]; then

echo -ne "\r${YELLOW}进度: ${i}/100${NC}"

fi

done

END_TIME=$(date +%s.%N)

DURATION=$(echo "$END_TIME - $START_TIME" | bc)

QPS=$(echo "scale=2; 100 / $DURATION" | bc)

echo -ne "\r"

echo -e "${GREEN}✓ 范围查询完成${NC}"

echo -e " 总耗时: ${DURATION} 秒"

echo -e " QPS: ${QPS} 次/秒"

echo ""

sleep 1

# 测试 4: 删除性能

echo -e "${GREEN}═══════════════════════════════════════${NC}"

echo -e "${GREEN}测试 4: 删除性能测试${NC}"

echo -e "${GREEN}═══════════════════════════════════════${NC}"

echo -e "${CYAN}删除 1000 个 key...${NC}"

START_TIME=$(date +%s.%N)

for i in {1..1000}; do

docker exec etcd1 etcdctl del \

--endpoints=${ENDPOINTS} \

${AUTH_PARAMS} \

/benchmark/test/key${i} > /dev/null 2>&1

if [ $((i % 100)) -eq 0 ]; then

echo -ne "\r${YELLOW}进度: ${i}/1000${NC}"

fi

done

END_TIME=$(date +%s.%N)

DURATION=$(echo "$END_TIME - $START_TIME" | bc)

QPS=$(echo "scale=2; 1000 / $DURATION" | bc)

echo -ne "\r"

echo -e "${GREEN}✓ 删除完成${NC}"

echo -e " 总耗时: ${DURATION} 秒"

echo -e " QPS: ${QPS} 次/秒"

echo ""

# 显示集群状态

echo -e "${GREEN}═══════════════════════════════════════${NC}"

echo -e "${GREEN}测试完成 - 集群状态${NC}"

echo -e "${GREEN}═══════════════════════════════════════${NC}"

echo ""

echo -e "${CYAN}集群健康:${NC}"

docker exec etcd1 etcdctl --endpoints=${ENDPOINTS} ${AUTH_PARAMS} endpoint health

echo ""

echo -e "${CYAN}集群状态:${NC}"

docker exec etcd1 etcdctl --endpoints=${ENDPOINTS} ${AUTH_PARAMS} endpoint status -w table

echo ""

echo -e "${CYAN}数据库大小:${NC}"

docker exec etcd1 etcdctl --endpoints=${ENDPOINTS} ${AUTH_PARAMS} endpoint status --write-out=json | \

jq -r '.[] | "\(.Endpoint): \(.Status.dbSize / 1024 / 1024 | floor)MB"' 2>/dev/null || \

docker exec etcd1 etcdctl --endpoints=${ENDPOINTS} ${AUTH_PARAMS} endpoint status

echo ""

echo -e "${GREEN}════════════════════════════════════════${NC}"

echo -e "${GREEN}所有测试完成！${NC}"

echo -e "${GREEN}════════════════════════════════════════${NC}"