amo/cmd/amocli/updater.go

package main

//import "surdeus.su/core/ss/u"
import "surdeus.su/core/amo"
import "surdeus.su/core/cli/mtool"
import "encoding/json"
import "os"
import "log"
import "time"

type Updater[V, VR any] interface {
	UpdateValues(*amo.Client, []V) (VR, error)
	GetNameMul() string
	GetFuncName() string
}

func RunUpdater[V, VR any, U Updater[V, VR]](
	u U, flags *mtool.Flags,
) {
	now := time.Now()
	var (
		opts DefaultFlags
	)
	MakeDefaultFlags(&opts, flags)
	flags.Parse()

	client, err := amo.NewClient(opts.SecretPath)
	if err != nil {
		log.Fatalf("NewAmoClient(...): %s\n", err)
	}

	values := []V{}
	dec := json.NewDecoder(os.Stdin)
	err = dec.Decode(&values)
	if err != nil {
		log.Fatalf("json.Decode(...): %s\n", err)
	}

	finish := RunForSliceInThreads[V](
		opts.Threads, opts.MEPR,
		values, func(thread int, s []V){
			_, err := u.UpdateValues(client, s)
			if err != nil {
				log.Printf(
					"thread(%d): %s(...): %s\n",
					thread, u.GetFuncName(), err,
				)
				return
			}
			if opts.Verbose {
				log.Printf(
					"thread(%d): Updated %d %s\n",
					thread, len(s), u.GetNameMul(),
				)
			}
		},
	)
	<-finish
	if opts.Verbose {
		rm := client.API.RequestsMade()
		/*log.Printf(
			"Summarized got %d %s\n",
			len(finalValues), g.GetNameMul(),
		)*/
		log.Printf(
			"Made %d requests in process\n",
			rm,
		)
		took := time.Since(now).Seconds()
		log.Printf(
			"Took %f seconds\n",
			took,
		)
		log.Printf("RPS = %f\n", float64(rm)/took)
	}
}