add support for counter updates

main.go

@@ -14,6 +14,7 @@ import (
 
 var keyspaceName string
 var tableName string
+var counterTableName string
 
 var concurrency int
 
@@ -40,6 +41,11 @@ func PrepareDatabase(session *gocql.Session, replicationFactor int) {
 	if err != nil {
 		log.Fatal(err)
 	}
+
+	err = session.Query("CREATE TABLE IF NOT EXISTS " + keyspaceName + "." + counterTableName + " (pk bigint, ck bigint, c1 counter, c2 counter, c3 counter, c4 counter, c5 counter, PRIMARY KEY(pk, ck)) WITH compression = { }").Exec()
+	if err != nil {
+		log.Fatal(err)
+	}
 }
 
 func GetWorkload(name string, threadId int, partitionOffset int) WorkloadGenerator {
@@ -69,6 +75,8 @@ func GetMode(name string) func(session *gocql.Session, workload WorkloadGenerato
 			return DoWrites
 		}
 		return DoBatchedWrites
+	case "counter_update":
+		return DoCounterUpdates
 	case "read":
 		return DoReads
 	default:
@@ -114,6 +122,7 @@ func main() {
 	flag.StringVar(&keyspaceName, "keyspace", "scylla_bench", "keyspace to use")
 	flag.StringVar(&tableName, "table", "test", "table to use")
 	flag.Parse()
+	counterTableName = "test_counters"
 
 	flag.Usage = func() {
 		fmt.Fprintf(os.Stdout, "Usage:\n%s [options]\n\n", os.Args[0])

modes.go

@@ -135,6 +135,37 @@ func DoBatchedWrites(session *gocql.Session, workload WorkloadGenerator) Result
 	return result
 }
 
+func DoCounterUpdates(session *gocql.Session, workload WorkloadGenerator) Result {
+	query := session.Query("UPDATE " + keyspaceName + "." + counterTableName + " SET c1 = c1 + 1, c2 = c2 + 1, c3 = c3 + 1, c4 = c4 + 1, c5 = c5 + 1 WHERE pk = ? AND ck = ?")
+
+	var operations int
+	latencyHistogram := NewHistogram()
+
+	start := time.Now()
+	for !workload.IsDone() && atomic.LoadUint32(&stopAll) == 0 {
+		operations++
+		pk := workload.NextPartitionKey()
+		ck := workload.NextClusteringKey()
+		bound := query.Bind(pk, ck)
+
+		requestStart := time.Now()
+		err := bound.Exec()
+		requestEnd := time.Now()
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		latency := requestEnd.Sub(requestStart)
+		err = latencyHistogram.RecordValue(latency.Nanoseconds())
+		if err != nil {
+			log.Fatal(err)
+		}
+	}
+	end := time.Now()
+
+	return Result{end.Sub(start), operations, operations, latencyHistogram}
+}
+
 func DoReads(session *gocql.Session, workload WorkloadGenerator) Result {
 	request := fmt.Sprintf("SELECT * FROM %s.%s WHERE pk = ? AND ck >= ? LIMIT %d", keyspaceName, tableName, rowsPerRequest)
 	query := session.Query(request)