single-region-transfer-leader.go

package main

import (
	"bytes"
	"context"
	"flag"
	"fmt"
	"log"
	"math/rand"
	"time"

	"github.com/tikv/client-go/v2/config"
	"github.com/tikv/client-go/v2/rawkv"
)

var (
	pd = flag.String("pd", "127.0.0.1:44279", "address of pd")
)

func main() {
	flag.Parse()

	cli, err := rawkv.NewClient(context.TODO(), []string{*pd}, config.Security{})
	assert(err)

	const N = 10000
	const L = 1024
	const S1 = 50
	const S2 = 5
	const M = 10

	log.Println("load data")
	for i := 0; i < N; i++ {
		err = cli.Put(context.TODO(), []byte(fmt.Sprintf("k%010d", i)), bytes.Repeat([]byte("a"), L))
		assert(err)
	}

	log.Println("start workload1")
	go func() {
		for {
			cli.Scan(context.TODO(), []byte(fmt.Sprintf("k%010d", rand.Intn(N))), nil, 1)
			time.Sleep(S1 * time.Millisecond)
		}
	}()

	log.Println("start workload2")
	go func() {
		for {
			if time.Now().Unix()%300 < 45 {
				cli.Scan(context.TODO(), []byte(fmt.Sprintf("k%010d", rand.Intn(N))), nil, M)
			}
			time.Sleep(S2 * time.Millisecond)
		}
	}()

	select {}
}

func assert(err error) {
	if err != nil {
		panic(err)
	}
}