func workerGetSetRand(client *memcache.Client, wg *sync.WaitGroup, ch <-chan int) {
defer wg.Done()
var item memcache.Item
itemsCount := *requestsCount / *workersCount
for _ = range ch {
n := rand.Intn(itemsCount)
item.Key = []byte(fmt.Sprintf("%s_%d", *key, n))
v := []byte(fmt.Sprintf("%s_%d", *value, n))
if rand.Float32() < 0.5 {
err := client.Get(&item)
if err == memcache.ErrCacheMiss {
continue
}
if err != nil {
log.Fatalf("Error in Client.Get(): [%s]", err)
}
if !bytes.Equal(item.Value, v) {
log.Fatalf("Unexpected value=[%s] obtained. Expected [%s]", item.Value, v)
}
} else {
item.Value = v
if err := client.Set(&item); err != nil {
log.Fatalf("Error in Client.Set(): [%s]", err)
}
}
}
}
func workerSet(client *memcache.Client, wg *sync.WaitGroup, ch <-chan int) {
defer wg.Done()
var item memcache.Item
for i := range ch {
item.Key = []byte(fmt.Sprintf("%s_%d", *key, i))
item.Value = []byte(fmt.Sprintf("%s_%d", *value, i))
if err := client.Set(&item); err != nil {
log.Fatalf("Error in Client.Set(): [%s]", err)
}
}
}
func workerGetMiss(client *memcache.Client, wg *sync.WaitGroup, ch <-chan int) {
defer wg.Done()
item := memcache.Item{
Key: []byte(*key),
}
for _ = range ch {
if err := client.Get(&item); err != memcache.ErrCacheMiss {
log.Fatalf("Error in Client.Get(): [%s]", err)
}
}
}
func workerGetHit(client *memcache.Client, wg *sync.WaitGroup, ch <-chan int) {
defer wg.Done()
item := memcache.Item{
Key: []byte(*key),
Value: []byte(*value),
}
valueOrig := item.Value
item.Value = nil
for _ = range ch {
if err := client.Get(&item); err != nil {
log.Fatalf("Error in Client.Get(): [%s]", err)
}
if !bytes.Equal(valueOrig, item.Value) {
log.Fatalf("Unexpected value read=[%s]. Expected=[%s]", item.Value, valueOrig)
}
}
}
func main() {
flag.Parse()
runtime.GOMAXPROCS(*goMaxProcs)
client := memcache.Client{
ServerAddr: *serverAddr,
ConnectionsCount: *connectionsCount,
MaxPendingRequestsCount: *maxPendingRequestsCount,
ReadBufferSize: *readBufferSize,
WriteBufferSize: *writeBufferSize,
OSReadBufferSize: *osReadBufferSize,
OSWriteBufferSize: *osWriteBufferSize,
}
client.Start()
defer client.Stop()
fmt.Printf("Config:\n")
fmt.Printf("serverAddr=[%s]\n", *serverAddr)
fmt.Printf("connectionsCount=[%d]\n", *connectionsCount)
fmt.Printf("goMaxProcs=[%d]\n", *goMaxProcs)
fmt.Printf("key=[%s]\n", *key)
fmt.Printf("maxPendingRequestsCount=[%d]\n", *maxPendingRequestsCount)
fmt.Printf("osReadBufferSize=[%d]\n", *osReadBufferSize)
fmt.Printf("osWriteBufferSize=[%d]\n", *osWriteBufferSize)
fmt.Printf("requestsCount=[%d]\n", *requestsCount)
fmt.Printf("readBufferSize=[%d]\n", *readBufferSize)
fmt.Printf("value=[%s]\n", *value)
fmt.Printf("workerMode=[%s]\n", *workerMode)
fmt.Printf("workersCount=[%d]\n", *workersCount)
fmt.Printf("writeBufferSize=[%d]\n", *writeBufferSize)
fmt.Printf("\n")
fmt.Printf("Preparing...")
ch := make(chan int, *requestsCount)
for i := 0; i < *requestsCount; i++ {
ch <- i
}
close(ch)
fmt.Printf("done\n")
fmt.Printf("starting...")
startTime := time.Now()
wg := sync.WaitGroup{}
defer func() {
wg.Wait()
duration := float64(time.Since(startTime)) / float64(time.Second)
fmt.Printf("done! %.3f seconds, %.0f qps\n", duration, float64(*requestsCount)/duration)
}()
item := memcache.Item{
Key: []byte(*key),
Value: []byte(*value),
}
worker := workerGetMiss
switch *workerMode {
case "GetHit":
if err := client.Set(&item); err != nil {
log.Fatalf("Error in Client.Set(): [%s]", err)
}
worker = workerGetHit
case "GetMiss":
if err := client.Delete(item.Key); err != nil && err != memcache.ErrCacheMiss {
log.Fatalf("Cannot delete item with key=[%s]: [%s]", item.Key, err)
}
worker = workerGetMiss
case "Set":
worker = workerSet
case "GetSetRand":
worker = workerGetSetRand
default:
log.Fatalf("Unknown workerMode=[%s]", *workerMode)
}
for i := 0; i < *workersCount; i++ {
wg.Add(1)
go worker(&client, &wg, ch)
}
}