Create & Init Project...

app/interface/main/broadcast/benchmark/push/BUILD

@@ -0,0 +1,40 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_binary",
+    "go_library",
+)
+
+go_binary(
+    name = "push",
+    embed = [":go_default_library"],
+    tags = ["automanaged"],
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["main.go"],
+    importpath = "go-common/app/interface/main/broadcast/benchmark/push",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//library/net/http/blademaster:go_default_library",
+        "//library/net/netutil/breaker:go_default_library",
+        "//library/time:go_default_library",
+    ],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+    visibility = ["//visibility:private"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [":package-srcs"],
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+)

app/interface/main/broadcast/benchmark/push/main.go

@@ -0,0 +1,116 @@
+package main
+
+// Start Commond eg:./push 1 500 127.0.0.1:7831 100
+// first parameter：beginning port
+// second parameter: end port
+// third parameter: comet server ip
+// fourth parameter: runing time
+
+import (
+	"context"
+	"fmt"
+	"log"
+	"net/url"
+	"os"
+	"runtime"
+	"strconv"
+	"time"
+
+	bm "go-common/library/net/http/blademaster"
+	"go-common/library/net/netutil/breaker"
+	xtime "go-common/library/time"
+)
+
+var (
+	lg         *log.Logger
+	httpClient *bm.Client
+	t          int
+)
+
+const TestContent = "{\"test\":1}"
+
+func main() {
+	runtime.GOMAXPROCS(runtime.NumCPU())
+	infoLogfi, err := os.OpenFile("./pushs.log", os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0644)
+	if err != nil {
+		panic(err)
+	}
+	lg = log.New(infoLogfi, "", log.LstdFlags|log.Lshortfile)
+
+	begin, err := strconv.Atoi(os.Args[1])
+	if err != nil {
+		panic(err)
+	}
+	end, err := strconv.Atoi(os.Args[2])
+	if err != nil {
+		panic(err)
+	}
+
+	t, err = strconv.Atoi(os.Args[4])
+	if err != nil {
+		panic(err)
+	}
+
+	num := runtime.NumCPU() * 2
+	lg.Printf("start routine num:%d", num)
+
+	l := end / num
+	b, e := begin, begin+l
+	time.AfterFunc(time.Duration(t)*time.Second, stop)
+	for i := begin; i <= end; i++ {
+		this := i
+		go func() {
+			for {
+				startPush(this, num)
+				time.Sleep(time.Second)
+			}
+		}()
+	}
+
+	for i := 0; i < num; i++ {
+		go startPush(b, e)
+		b += l
+		e += l
+	}
+
+	time.Sleep(9999 * time.Hour)
+}
+
+func init() {
+	httpClient = bm.NewClient(&bm.ClientConfig{
+		App: &bm.App{
+			Key:    "6aa4286456d16b97",
+			Secret: "test",
+		},
+		Dial:      xtime.Duration(time.Second),
+		Timeout:   xtime.Duration(time.Second),
+		KeepAlive: xtime.Duration(time.Second * 10),
+		Breaker: &breaker.Config{
+			Window:    xtime.Duration(time.Second),
+			Sleep:     xtime.Duration(time.Second),
+			Bucket:    10,
+			Ratio:     0.8,
+			Request:   100,
+			SwitchOff: false,
+		},
+	})
+}
+
+func stop() {
+	os.Exit(-1)
+}
+
+func startPush(b, e int) {
+	lg.Printf("start Push key from %d to %d", b, e)
+	for i := b; i < b+e; i++ {
+		params := url.Values{}
+		params.Set("operation", "9")
+		params.Set("keys", strconv.Itoa(b))
+		params.Set("message", TestContent)
+		err := httpClient.Get(context.Background(), fmt.Sprintf("http://%s/x/internal/broadcast/push/keys", os.Args[3]), "", params, nil)
+		if err != nil {
+			lg.Printf("get error (%v)", err)
+			return
+		}
+	}
+}