package main

import (
	"os"
	"os/signal"
	"sync"
	"syscall"
	"time"

	client "github.com/pingcap/schrodinger/pkg/test_client"
	"golang.org/x/net/context"
)

var (
	checkInterval = 30 * time.Second
)

func main() {
	client.Info("[pd test] start...")

	ctx, cancel := context.WithCancel(context.Background())
	cli := client.New(ctx, client.ModeOnline)
	_ = cli.PDs()
	sc := make(chan os.Signal, 1)
	signal.Notify(sc,
		syscall.SIGHUP,
		syscall.SIGINT,
		syscall.SIGTERM,
		syscall.SIGQUIT)
	go func() {
		sig := <-sc
		client.Infof("[pd test] Got signal [%s] to exist.", sig)
		cancel()
		os.Exit(0)
	}()
	wg := &sync.WaitGroup{}
	wg.Add(2)
	checkAPI(ctx, cli, wg)
	checkTSO(ctx, cli, wg)
	wg.Wait()
}