stuff

Signed-off-by: Brendan Dougherty <[email protected]>

examples/common/env.sh

@@ -84,3 +84,4 @@ if [[ -n ${BASH} ]]; then
     shopt -s expand_aliases
 fi
 
+export GOMAXPROCS=4

examples/common/scripts/vtgate-up.sh

@@ -41,6 +41,7 @@ vtgate \
   --pid_file $VTDATAROOT/tmp/vtgate.pid \
   --enable_buffer \
   --mysql_auth_server_impl none \
+  --pprof-http \
   > $VTDATAROOT/tmp/vtgate.out 2>&1 &
 
 # Block waiting for vtgate to be listening

examples/common/scripts/vttablet-up.sh

@@ -54,6 +54,7 @@ vttablet \
  --service_map 'grpc-queryservice,grpc-tabletmanager,grpc-updatestream' \
  --pid_file $VTDATAROOT/$tablet_dir/vttablet.pid \
  --heartbeat_on_demand_duration=5s \
+  --pprof-http \
  > $VTDATAROOT/$tablet_dir/vttablet.out 2>&1 &
 
 # Block waiting for the tablet to be listening

examples/local/bench.sh

@@ -0,0 +1,13 @@
+#!/bin/bash
+
+source ../common/env.sh
+
+vtbench \
+  --config-file-not-found-handling ignore \
+	--protocol mysql \
+	--host 127.0.0.1 \
+	--port 15306 \
+	--db commerce@primary \
+	--sql "select * from customer where customer_id = :random_int" \
+	--threads 10 \
+	--count 100000

go/cmd/vtbench/cli/vtbench.go

@@ -26,7 +26,6 @@ import (
 	"github.com/spf13/cobra"
 
 	"vitess.io/vitess/go/acl"
-	"vitess.io/vitess/go/vt/dbconfigs"
 	"vitess.io/vitess/go/vt/grpccommon"
 	"vitess.io/vitess/go/vt/logutil"
 	"vitess.io/vitess/go/vt/servenv"
@@ -191,23 +190,23 @@ func run(cmd *cobra.Command, args []string) error {
 		return errors.New("vtbench requires either host/port or unix_socket")
 	}
 
-	var password string
-	if clientProto == vtbench.MySQL {
-		var err error
-		_, password, err = dbconfigs.GetCredentialsServer().GetUserAndPassword(user)
-		if err != nil {
-			return fmt.Errorf("error reading password for user %v from file: %w", user, err)
-		}
-	}
+	//var password string
+	//if clientProto == vtbench.MySQL {
+	//	var err error
+	//	_, password, err = dbconfigs.GetCredentialsServer().GetUserAndPassword(user)
+	//	if err != nil {
+	//		return fmt.Errorf("error reading password for user %v from file: %w", user, err)
+	//	}
+	//}
 
 	connParams := vtbench.ConnParams{
 		Hosts:      strings.Split(host, ","),
 		Port:       port,
 		UnixSocket: unixSocket,
 		Protocol:   clientProto,
 		DB:         db,
-		Username:   user,
-		Password:   password,
+		//Username:   user,
+		//Password:   password,
 	}
 
 	b := vtbench.NewBench(threads, count, connParams, sql)

go/vtbench/vtbench.go

@@ -19,6 +19,7 @@ package vtbench
 import (
 	"context"
 	"fmt"
+	"math/rand/v2"
 	"strings"
 	"sync"
 	"time"
@@ -180,6 +181,7 @@ func (b *Bench) createConns(ctx context.Context) error {
 
 func (b *Bench) getQuery(i int) (string, map[string]*querypb.BindVariable) {
 	query := strings.Replace(b.Query, ":thread", fmt.Sprintf("%d", i), -1)
+	query = strings.Replace(query, ":random_int", fmt.Sprintf("%d", rand.Int()), -1)
 	bindVars := make(map[string]*querypb.BindVariable)
 	return query, bindVars
 }