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)
}
}
