golang-对redis进行压力测试

golang-对redis进行压力测试

用golang来检测redis的qps。

直接上代码吧！

package main

import (
	"fmt"
	"github.com/garyburd/redigo/redis"
	"sync"
	"time"
)

var(
	Pool	 redis.Pool			//连接池
	oSynWait sync.WaitGroup		//互斥锁
)

const (
	OnMaxRun 	=	10000	//单线程执行命令数
	AllMaxRun	=	50		//并发数
	TimeFormat  = "2006-01-02 15:04:05.999" //打印时间
	Type = "set"			//指令类型
)


func main() {
	fmt.Println("--------------------------redis压力测试--------------------------")
	GetPool(&Pool)
	OldTime := time.Now()
	fmt.Println("开始",OldTime.Format(TimeFormat))

	oSynWait.Add(AllMaxRun)
	for i := 0; i < AllMaxRun; i ++ {
		go ReadWriteInfo()
	}
	oSynWait.Wait()

	EndTime := time.Now()
	fmt.Println("结束",EndTime.Format(TimeFormat))
	fmt.Printf("并发数: %d ;总数据量：%d ;耗时: %.2fs ;\n类型：%s ;key：1;value：1;\nqps：%.0f\n",
		AllMaxRun,AllMaxRun*OnMaxRun,EndTime.Sub(OldTime).Seconds(),Type,AllMaxRun*OnMaxRun/time.Now().Sub(OldTime).Seconds())
	fmt.Println("-----------------------------测试结束-----------------------------")
}


//取得一个连接池
func GetPool(this *redis.Pool) {
	this.MaxActive = 10
	this.MaxIdle = 10
	this.Wait = true
	this.IdleTimeout = 100*time.Second
	this.Dial = func() (conn redis.Conn, err error) {
		conn,err = redis.Dial("tcp","127.0.0.1:6379")
		if err != nil{
			fmt.Println("连接失败：",err)
		}
		return
	}
}

//存放数据
func ReadWriteInfo() {
	defer oSynWait.Done()
	for i := 0; i <OnMaxRun; i++ {
		conn := Pool.Get()
		//存
		_, err := conn.Do(Type, "1", "1")
		if err != nil {
			fmt.Println("存放数据失败", err)
			return
		}
		//取
		//_, err = conn.Do("get", "1")
		//if err != nil {
		//	fmt.Println("取出数据失败", err)
		//	return
		//}
		//关
		err = conn.Close()
		if err!= nil{
			fmt.Println("关闭连接失败 ",err)
			return
		}
	}
}

结果

和redis自带的测试结果差不多哈。

注意

1.请确保电脑性能能跟的上，在只有单核，2G内存的虚拟机上协程没能完全跑起来，数值很低。
2.并发设置：大概30-50就合适了，多了获得的结果也差不多。