added flags to testclient, added optional getrepo call

This commit is contained in:
Philipp Mieden 2019-05-24 17:40:26 +02:00
parent 871c844f7b
commit a5ff003d8f
2 changed files with 42 additions and 14 deletions

View File

@ -100,7 +100,7 @@ func (repo *Repo) getNodes(nodeRequests map[string]*requests.Node, env *requests
for nodeName, nodeRequest := range nodeRequests {
if nodeName == "" || nodeRequest.ID == "" {
Log.Error("invalid node request", zap.Error(errors.New("nodeName or nodeRequest.ID empty")))
Log.Info("invalid node request", zap.Error(errors.New("nodeName or nodeRequest.ID empty")))
continue
}
Log.Debug("adding node", zap.String("name", nodeName), zap.String("requestID", nodeRequest.ID))

View File

@ -1,6 +1,7 @@
package main
import (
"flag"
"log"
"time"
@ -8,24 +9,51 @@ import (
"github.com/foomo/contentserver/client"
)
// https://globus-b.stage.mzg.bestbytes.net/contentserverapi
var (
flagAddr = flag.String("addr", "http://127.0.0.1:9191/contentserver", "set addr")
flagGetRepo = flag.Bool("getRepo", false, "get repo")
flagUpdate = flag.Bool("update", true, "trigger content update")
flagNum = flag.Int("num", 100, "num repititions")
flagDelay = flag.Int("delay", 2, "delay in seconds")
)
func main() {
serverAdr := "http://127.0.0.1:9191/contentserver"
c, errClient := client.NewHTTPClient(serverAdr)
flag.Parse()
c, errClient := client.NewHTTPClient(*flagAddr)
if errClient != nil {
log.Fatal(errClient)
}
for i := 1; i <= 150; i++ {
go func(num int) {
log.Println("start update")
resp, errUpdate := c.Update()
if errUpdate != nil {
spew.Dump(resp)
log.Fatal(errUpdate)
}
log.Println(num, "update done", resp)
}(i)
time.Sleep(2 * time.Second)
for i := 1; i <= *flagNum; i++ {
if *flagUpdate {
go func(num int) {
log.Println("start update")
resp, errUpdate := c.Update()
if errUpdate != nil {
spew.Dump(resp)
log.Fatal(errUpdate)
}
log.Println(num, "update done", resp)
}(i)
}
if *flagGetRepo {
go func(num int) {
log.Println("get repo", num)
_, err := c.GetRepo()
if err != nil {
// spew.Dump(resp)
log.Fatal("failed to get repo")
}
log.Println(num, "get repo done")
}(i)
}
time.Sleep(time.Duration(*flagDelay) * time.Second)
}
log.Println("done")