liwei_3d/go_dreamfactory
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

update

Browse Source
This commit is contained in:
meixiongfeng 2022-12-06 14:18:43 +08:00
parent 15a5e3063f
commit 91242479e6
4 changed files with 87 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

86
stress/robot/robot.go
View File

@ -203,7 +203,14 @@ func (r *Robot) MessageRsp(mainType, subType string) (bool, int64) {
}
//fmt.Printf("接收消息=====loginData:%v,userExpand:%v\n", resp.Data, resp.Ex)
return true, byteLen
} else if msg.MainType == "goutmet" && msg.SubType == "getranduser" {
} else if msg.MainType == "gourmet" && msg.SubType == "getranduser" {
resp := &pb.GourmetGetRandUserResp{}
if !comm.ProtoUnmarshal(msg, resp) {
return false, byteLen
}
return false, byteLen
} else if msg.MainType == "gourmet" && msg.SubType == "getlist" {
resp := &pb.GourmetGetRandUserResp{}
if !comm.ProtoUnmarshal(msg, resp) {
return false, byteLen
@ -220,12 +227,12 @@ func (r *Robot) CloseHandler() {
}
func (r *Robot) GetMessagedata() []byte {
_, data, err := r.ws.ReadMessage()
func (r *Robot) GetMessagedata() (data []byte, err error) {
_, data, err = r.ws.ReadMessage()
if err != nil {
fmt.Printf("readMessage err:%v", err)
}
return data
return data, err
}
func (r *Robot) Login() error {
msg := &pb.UserMessage{MainType: "user", SubType: "login"}
@ -242,8 +249,8 @@ func (r *Robot) Login() error {
}
func (r Robot) GetRandUser() bool {
mainType := "goutmet"
subType := "getlist"
mainType := "gourmet"
subType := "getranduser"
msg := &pb.UserMessage{MainType: mainType, SubType: subType}
rsp := &pb.GourmetGetRandUserReq{
@ -258,12 +265,12 @@ func (r Robot) GetRandUser() bool {
}
}
for i := 0; i < 100; i++ {
for {
if b, _ := r.MessageRsp(mainType, subType); b {
return true
}
}
return true
//return true
}
func (r *Robot) Create() bool {
mainType := "user"
@ -287,24 +294,22 @@ func (r *Robot) Create() bool {
}
}
for i := 0; i < 100; i++ { // 防止丢包
for { // 防止丢包
if b, _ := r.MessageRsp(mainType, subType); b {
return true
}
}
return false
//return false
}
// 获取美食家信息列表数据
func (r Robot) GetGourmetList() int64 {
var byteLen int64
mainType := "goutmet"
subType := "getranduser"
func (r Robot) GetGourmetList() {
mainType := "gourmet"
subType := "getlist"
msg := &pb.UserMessage{MainType: mainType, SubType: subType}
rsp := &pb.GourmetGetRandUserReq{
People: 3,
}
rsp := &pb.GourmetGetListReq{}
msg.Sec = r.BuildSecStr()
if comm.ProtoMarshal(rsp, msg) {
data, _ := proto.Marshal(msg)
@ -314,30 +319,31 @@ func (r Robot) GetGourmetList() int64 {
}
}
for i := 0; i < 100; i++ {
var msg *pb.UserMessage = &pb.UserMessage{}
_, data, err := r.ws.ReadMessage()
if err != nil {
fmt.Printf("readMessage err:%v", err)
continue
}
byteLen += int64(len(data))
if err = proto.Unmarshal(data, msg); err != nil {
fmt.Printf("unmarshal err:%v", err)
}
if msg.MainType == "goutmet" && msg.SubType == "getlist" {
resp := &pb.GourmetGetListResp{}
if !comm.ProtoUnmarshal(msg, resp) { //反序列化失败
break
}
break
}
}
return byteLen
// for {
// var msg *pb.UserMessage = &pb.UserMessage{}
// _, data, err := r.ws.ReadMessage()
// if err != nil {
// fmt.Printf("readMessage err:%v", err)
// continue
// }
// if err = proto.Unmarshal(data, msg); err != nil {
// fmt.Printf("unmarshal err:%v", err)
// }
// if msg.MainType == mainType && msg.SubType == subType {
// byteLen += int64(len(data))
// resp := &pb.GourmetGetListResp{}
// if !comm.ProtoUnmarshal(msg, resp) { //反序列化失败
// break
// }
// break
// }
// }
return
}
// addItem
func (r Robot) AddGrormetItem() {
func (r Robot) AddGrormetItem() bool {
mainType := "gm"
subType := "cmd"
msg := &pb.UserMessage{MainType: mainType, SubType: subType}
@ -353,4 +359,8 @@ func (r Robot) AddGrormetItem() {
fmt.Printf("WriteMessage err:%v", err)
}
}
return true
}
func (r *Robot) SetAccount(account string) {
r.account = account
}

3
stress/server/dispose.go
View File

@ -6,6 +6,7 @@ import (
"fmt"
"go_dreamfactory/stress/model"
"go_dreamfactory/stress/robot"
"strconv"
"sync"
"time"
@ -53,7 +54,7 @@ func Dispose(ctx context.Context, concurrency, totalNumber uint64, request *mode
case 1:
// 连接以后再启动协程
r := robot.NewRobot(request.URL)
r.SetAccount("00000" + strconv.Itoa(int(i)))
go golink.WebSocket(ctx, i, ch, totalNumber, &wg, request, r)
case 2:
// 并发建立长链接

40
stress/server/golink/websocket_link.go
View File

@ -3,13 +3,17 @@ package golink
import (
"context"
"fmt"
"sync"
"time"
"go_dreamfactory/comm"
"go_dreamfactory/pb"
"go_dreamfactory/stress/helper"
"go_dreamfactory/stress/model"
"go_dreamfactory/stress/robot"
"github.com/golang/protobuf/proto"
)
const (
@ -23,7 +27,7 @@ var (
)
func init() {
keepAlive = false
keepAlive = true
}
// WebSocket webSocket go link
@ -75,12 +79,13 @@ func webSocketRequest(chanID uint64, ch chan<- *model.RequestResults, i uint64,
errCode = model.HTTPOk
byteLen int64 // 协议传输字节数
)
// 需要发送的数据
head := &pb.UserMessage{MainType: "user", SubType: "login"}
r.SendToClient(head, &pb.UserLoginReq{})
// 美食馆增加道具
r.AddGrormetItem()
//time.Sleep(1 * time.Millisecond)
// for {
// if b, len := r.MessageRsp("user", "login"); b {
// byteLen += len
@ -89,10 +94,37 @@ func webSocketRequest(chanID uint64, ch chan<- *model.RequestResults, i uint64,
// break
// }
// }
// r.Create()
startTime := time.Now()
//r.Create()
//isSucceed = r.AddGrormetItem()
//startTime := time.Now()
// r.GetRandUser() // 获取随机在线玩家数据
byteLen = r.GetGourmetList()
r.GetGourmetList()
for {
var msg *pb.UserMessage = &pb.UserMessage{}
data, err := r.GetMessagedata()
if err != nil {
fmt.Printf("readMessage err:%v", err)
isSucceed = false
break
}
if err = proto.Unmarshal(data, msg); err != nil {
fmt.Printf("unmarshal err:%v", err)
}
if msg.MainType == "gourmet" && msg.SubType == "getlist" {
byteLen += int64(len(data))
isSucceed = true
resp := &pb.GourmetGetListResp{}
if !comm.ProtoUnmarshal(msg, resp) { //反序列化失败
break
}
break
}
}
requestTime := uint64(helper.DiffNano(startTime))
requestResults := &model.RequestResults{
ID: "",

2
stress/stress.go
View File

@ -79,7 +79,7 @@ func main() {
flag.Parse()
statistics.InitLog()
concurrency = 12
concurrency = 5
totalNumber = 1
debugStr = "false"
requestURL = "ws://10.0.0.85:7891/gateway"