|
@@ -14,7 +14,10 @@ import (
|
|
|
|
|
|
//服务地址配置
|
|
//服务地址配置
|
|
var (
|
|
var (
|
|
- rdserver = flag.String("rd", "127.0.0.1:10021", "服务治理地址")
|
|
|
|
|
|
+ rdserver = flag.String("rd", "127.0.0.1:10021", "服务治理地址")
|
|
|
|
+ balancetype = flag.Int("balance", 0, "负载策略 0=随机 1=服务器压力均衡 2=轮训占用")
|
|
|
|
+ threads = flag.Int("threads", 20, "压力并发数")
|
|
|
|
+ requests = flag.Int("reqs", 100, "单个线程服务请求次数")
|
|
)
|
|
)
|
|
|
|
|
|
func init() {
|
|
func init() {
|
|
@@ -35,8 +38,8 @@ func run(thread int, wg *sync.WaitGroup) {
|
|
var client proto.ServiceClient
|
|
var client proto.ServiceClient
|
|
client = proto.NewServiceClient(conn)
|
|
client = proto.NewServiceClient(conn)
|
|
|
|
|
|
- for i := 0; i < 20; i++ {
|
|
|
|
- repl, err := client.Apply(context.Background(), &proto.ApplyReqData{Name: "demo", Balance: 2})
|
|
|
|
|
|
+ for i := 0; i < *requests; i++ {
|
|
|
|
+ repl, err := client.Apply(context.Background(), &proto.ApplyReqData{Name: "demo", Balance: int32(*balancetype)})
|
|
if err != nil {
|
|
if err != nil {
|
|
log.Println("出错了")
|
|
log.Println("出错了")
|
|
log.Fatalln(err.Error())
|
|
log.Fatalln(err.Error())
|
|
@@ -71,7 +74,7 @@ func run(thread int, wg *sync.WaitGroup) {
|
|
|
|
|
|
func main() {
|
|
func main() {
|
|
wg := new(sync.WaitGroup)
|
|
wg := new(sync.WaitGroup)
|
|
- for i := 0; i < 3; i++ {
|
|
|
|
|
|
+ for i := 0; i < *threads; i++ {
|
|
wg.Add(1)
|
|
wg.Add(1)
|
|
go run(i, wg)
|
|
go run(i, wg)
|
|
}
|
|
}
|