benchmark nutcracker

nutcracker_tester.go

package main

import (
	"code.google.com/p/go-uuid/uuid"
	"flag"
	"fmt"
	"github.com/garyburd/redigo/redis"
	"strings"
	"sync"
	"time"
)

func newPool(server string, n int) *redis.Pool {
	return &redis.Pool{
		MaxIdle:     2 * n,
		IdleTimeout: 240 * time.Second,
		Dial: func() (redis.Conn, error) {
			c, err := redis.Dial("tcp", server)
			if err != nil {
				return nil, err
			}
			return c, err
		},
	}
}

func tester(pool *redis.Pool) {
	defer wg.Done()
	conn := pool.Get()
	defer conn.Close()
	conn.Send("SET", uuid.New(), "1", "EX", "86400")
	if err := conn.Flush(); err != nil {
		fmt.Println(err)
	}
}

var wg *sync.WaitGroup = &sync.WaitGroup{}

func main() {
	var n, p, port int
	var ips string
	flag.StringVar(&ips, "u", "10.1.201.88", "servers")
	flag.IntVar(&p, "p", 12, "port range")
	flag.IntVar(&n, "n", 100, "req number")
	flag.IntVar(&port, "port", 4001, "port begin")
	flag.Parse()

	IPS := strings.Split(ips, ",")
	pools := make([]*redis.Pool, len(IPS)*p)

	num := make([]struct{}, p)
	req := make([]struct{}, n)
	wg.Add(n * len(IPS) * len(num))
	for i, ip := range IPS {
		for j, _ := range num {
			server := fmt.Sprintf("%s:%d", ip, port+j)
			pools[i*p+j] = newPool(server, n)
		}
	}
	begin := time.Now().UnixNano()
	for _, pool := range pools {
		for _ = range req {
			go tester(pool)
		}
	}
	wg.Wait()
	end := time.Now().UnixNano() - begin
	fmt.Println(end, n*len(IPS)*p, float64(n*len(IPS)*p)/(float64(end)*0.000000001))
}