57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"math/rand"
|
|
"net/http"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/twitchtv/twirp"
|
|
|
|
"mcquay.me/hwt"
|
|
pb "mcquay.me/hwt/rpc/hwt"
|
|
)
|
|
|
|
const usage = "hwtc <server address> [subject]"
|
|
|
|
func main() {
|
|
if len(os.Args) < 3 {
|
|
fmt.Fprintf(os.Stderr, "%v\n", usage)
|
|
os.Exit(1)
|
|
}
|
|
|
|
c := pb.NewHelloWorldProtobufClient(fmt.Sprintf("http://%s", os.Args[1]), &http.Client{})
|
|
|
|
h := http.Header{}
|
|
h.Set("sm-auth", hwt.PSK)
|
|
ctx := context.Background()
|
|
ctx, err := twirp.WithHTTPRequestHeaders(ctx, h)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "setting twirp headers: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
s := make(chan bool, 10)
|
|
for i := 0; ; i++ {
|
|
s <- true
|
|
go func(j int) {
|
|
sleep := time.Duration(250+rand.Intn(500)) * time.Millisecond
|
|
for {
|
|
resp, err := c.Hello(ctx, &pb.HelloReq{Subject: strings.Join(os.Args[2:], " ")})
|
|
if err == nil {
|
|
fmt.Printf("0x%08x: %#v\n", j, resp)
|
|
break
|
|
}
|
|
fmt.Fprintf(os.Stderr, "sleeping %v because: %v\n", sleep, err)
|
|
time.Sleep(sleep)
|
|
sleep *= 2
|
|
continue
|
|
}
|
|
<-s
|
|
}(i)
|
|
}
|
|
}
|