108 lines
2.9 KiB
Go
108 lines
2.9 KiB
Go
package service
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
|
||
"go-common/app/admin/ep/melloi/model"
|
||
"go-common/library/log"
|
||
)
|
||
|
||
//AddSciptSnap add scriptSnap
|
||
func (s *Service) AddSciptSnap(ptestParam model.DoPtestParam, excuteId string) (scriptSnapIds []int, err error) {
|
||
var (
|
||
scripts []*model.Script
|
||
scriptSnapId int
|
||
)
|
||
|
||
//场景脚本快照保存逻辑
|
||
if ptestParam.Type == model.SCENE_SCRIPT_TYPE {
|
||
for _, script := range ptestParam.Scripts {
|
||
script.SceneID = ptestParam.SceneID
|
||
if scriptSnapId, err = s.AddScriptSnapInfo(script, excuteId); err != nil {
|
||
log.Error("s.AddScriptSnapInfo err :(%v)", err)
|
||
return
|
||
}
|
||
scriptSnapIds = append(scriptSnapIds, scriptSnapId)
|
||
}
|
||
return
|
||
}
|
||
|
||
//单场景http 脚本的快照保存
|
||
if scripts, err = s.QueryScripts(&model.Script{ID: ptestParam.ScriptID}, 1, 5); err != nil {
|
||
log.Error(" s.QueryScripts err :(%v)", err)
|
||
return
|
||
}
|
||
if scriptSnapId, err = s.AddScriptSnapInfo(scripts[0], excuteId); err != nil {
|
||
log.Error("s.AddScriptSnapInfo err :(%v)", err)
|
||
return
|
||
}
|
||
scriptSnapIds = append(scriptSnapIds, scriptSnapId)
|
||
return
|
||
}
|
||
|
||
//AddScriptSnapInfo Add scriptSnap Info
|
||
func (s *Service) AddScriptSnapInfo(script *model.Script, excuteId string) (scriptSnapId int, err error) {
|
||
var JSON []byte
|
||
scriptSnap := model.ScriptSnap{ScriptID: script.ID}
|
||
script.ID = 0
|
||
|
||
if JSON, err = json.Marshal(script); err != nil {
|
||
log.Error(" json.Marshal(script) error :(%v)", err)
|
||
return
|
||
}
|
||
if err = json.Unmarshal([]byte(string(JSON)), &scriptSnap); err != nil {
|
||
return
|
||
}
|
||
|
||
scriptSnap.ExecuteID = excuteId
|
||
if scriptSnapId, err = s.dao.AddScriptSnap(&scriptSnap); err != nil {
|
||
log.Error("s.dao.AddScriptSnap error :(%v)", err)
|
||
return
|
||
}
|
||
return
|
||
}
|
||
|
||
//AddSnap add snap
|
||
func (s *Service) AddSnap(c context.Context, ptestParam model.DoPtestParam, executeID, jobName, jobNamed string) (scriptSnapIDs []int, err error) {
|
||
if ptestParam.Type == model.PROTOCOL_GRPC {
|
||
if scriptSnapIDs, err = s.AddGRPCSnap(ptestParam.ScriptID, executeID); err != nil {
|
||
log.Error("save grpc snap failed,(%v)", err)
|
||
s.DeleteJob(context.TODO(), jobName)
|
||
return
|
||
}
|
||
} else {
|
||
//用jobNamed 表示执行id，即是 ExcuteId
|
||
if scriptSnapIDs, err = s.AddSciptSnap(ptestParam, jobNamed); err != nil {
|
||
log.Error("s.AddSciptSnap err :(%v)", err)
|
||
s.DeleteJob(context.TODO(), jobName)
|
||
return
|
||
}
|
||
}
|
||
return
|
||
}
|
||
|
||
// AddGRPCSnap Add GRPC Snap
|
||
func (s *Service) AddGRPCSnap(grpcID int, executeID string) (snapIDs []int, err error) {
|
||
var (
|
||
grpc *model.GRPC
|
||
grpcSnap *model.GRPCSnap
|
||
j []byte
|
||
)
|
||
if grpc, err = s.dao.QueryGRPCByID(grpcID); err != nil {
|
||
return
|
||
}
|
||
grpcSnap = &model.GRPCSnap{GRPCID: grpc.ID, ExecuteID: executeID}
|
||
grpc.ID = 0
|
||
if j, err = json.Marshal(grpc); err != nil {
|
||
return
|
||
}
|
||
if err = json.Unmarshal([]byte(string(j)), grpcSnap); err != nil {
|
||
return
|
||
}
|
||
if err = s.dao.CreateGRPCSnap(grpcSnap); err != nil {
|
||
return
|
||
}
|
||
return []int{grpcSnap.ID}, nil
|
||
}
|