You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
278 lines
12 KiB
278 lines
12 KiB
package dao |
|
|
|
import ( |
|
"strconv" |
|
"strings" |
|
|
|
"go-common/app/admin/ep/melloi/conf" |
|
"go-common/app/admin/ep/melloi/model" |
|
) |
|
|
|
// AddScene Add Scene |
|
func (d *Dao) AddScene(scene *model.Scene) (sceneId int, err error) { |
|
scene.IsActive = true |
|
err = d.DB.Table("scene").Create(scene).Error |
|
sceneId = scene.ID |
|
return |
|
} |
|
|
|
// QueryDraft Query Draft |
|
func (d *Dao) QueryDraft(scene *model.Scene) (res *model.QueryDraft, err error) { |
|
res = &model.QueryDraft{} |
|
err = d.DB.Table("scene").Select("id as scene_id, scene_name").Where(scene).Where("is_draft = 1 and is_active = 1").Count(&res.Total).Order("mtime desc").Find(&res.Drafts).Error |
|
return |
|
} |
|
|
|
// UpdateScene Update Scene |
|
func (d *Dao) UpdateScene(scene *model.Scene, scriptIDList []int) (fusing int, err error) { |
|
if err = d.DB.Exec("update scene set fusing = ? where id = ?", scene.Fusing, scene.ID).Error; err != nil { |
|
return |
|
} |
|
if err = d.DB.Model(&model.Scene{}).Where("ID = ?", scene.ID).Update(scene).Error; err != nil { |
|
return |
|
} |
|
if scene.IsBatch { //选择已有接口点击确定时执行 |
|
if scene.Fusing == 0 { |
|
err = d.DB.Model(&model.Script{}).Where("scene_id = ? and id in (?)", scene.ID, scriptIDList).Update("fusing", conf.Conf.Melloi.DefaultFusing).Error |
|
} else { |
|
err = d.DB.Model(&model.Script{}).Where("scene_id = ? and id in (?)", scene.ID, scriptIDList).Update("fusing", scene.Fusing).Error |
|
} |
|
} else if scene.IsUpdate { // |
|
err = d.DB.Model(&model.Script{}).Where("scene_id = ?", scene.ID).Update("fusing", scene.Fusing).Error |
|
} |
|
fusing = scene.Fusing |
|
return |
|
} |
|
|
|
// SaveScene Save Scene |
|
func (d *Dao) SaveScene(scene *model.Scene) error { |
|
return d.DB.Model(&model.Scene{}).Where("ID = ?", scene.ID).Update("is_draft", 0).Update("scene_type", scene.SceneType).Error |
|
} |
|
|
|
// SaveOrder Save Order |
|
func (d *Dao) SaveOrder(reqList []*model.GroupOrder, scene *model.Scene) (err error) { |
|
for _, req := range reqList { |
|
if scene.SceneType == 1 { |
|
err = d.DB.Model(&model.Script{}).Where("ID = ?", req.ID).Update("group_id", req.GroupID).Error |
|
} else { |
|
err = d.DB.Model(&model.Script{}).Where("ID = ?", req.ID).Update("run_order", req.RunOrder).Error |
|
} |
|
} |
|
return |
|
} |
|
|
|
// QueryGroupId Query GroupId |
|
func (d *Dao) QueryGroupId(script *model.Script) (res *model.QueryRelation, groupId int, err error) { |
|
res = &model.QueryRelation{} |
|
//select group_id, count(group_id) as count from script where scene_id = 1 and group_id in (select group_id from script where id = 777) group by group_id |
|
err = d.DB.Table("script").Select("group_id").Where("id = ?", script.ID).Find(&res.RelationList).Error |
|
groupId = res.RelationList[0].GroupID |
|
return |
|
} |
|
|
|
// QueryRelation Query Relation |
|
func (d *Dao) QueryRelation(groupId int, script *model.Script) (res *model.QueryRelation, err error) { |
|
res = &model.QueryRelation{} |
|
err = d.DB.Table("script").Select("group_id, count(group_id) as count").Where("scene_id = ? and group_id = ?", script.SceneID, groupId).Find(&res.RelationList).Error |
|
return |
|
} |
|
|
|
// QueryAPI Query API |
|
func (d *Dao) QueryAPI(scene *model.Scene) (res *model.QueryAPIs, err error) { |
|
res = &model.QueryAPIs{} |
|
if err = d.DB.Table(model.Script{}.TableName()).Select("id").Where("scene_id = ? and active = 1", scene.ID).Count(&res.Total).Error; err != nil { |
|
return |
|
} |
|
if res.Total > 0 { |
|
//select scene.scene_name, scene.scene_type, scene.department, scene.project, scene.app, script.group_id, script.run_order, script.id as test_id, script.test_name |
|
//from scene INNER JOIN script on scene.id = script.scene_id and script.active = 1 where script.scene_id = 1 |
|
gDB := d.DB.Table(model.Scene{}.TableName()).Select("scene.id as scene_id, scene.scene_name, scene.scene_type, scene.department, " + |
|
"scene.project, scene.app, script.group_id, script.run_order, script.id, script.test_name, script.url, script.output_params, script.threads_sum, script.load_time"). |
|
Joins("inner join script on scene.id = script.scene_id and script.active = 1") |
|
if scene.SceneType == 2 { |
|
err = gDB.Where(scene).Count(&res.Total).Order("group_id, run_order").Find(&res).Find(&res.APIs).Error |
|
res.SceneType = 2 |
|
} else { |
|
err = gDB.Where(scene).Count(&res.Total).Order("script.id").Find(&res).Find(&res.APIs).Error |
|
res.SceneType = 1 |
|
} |
|
} else { |
|
gDB := d.DB.Table(model.Scene{}.TableName()).Select("id as scene_id, scene_name, department, project, app") |
|
err = gDB.Where("id = ?", scene.ID).Find(&res).Error |
|
//查询无结果时，APIs返回空数组 |
|
res.APIs = []*model.TestAPI{} |
|
} |
|
return |
|
} |
|
|
|
// DeleteAPI Delete API |
|
func (d *Dao) DeleteAPI(script *model.Script) error { |
|
return d.DB.Model(&model.Script{}).Where("ID = ?", script.ID).Update("active", 0).Error |
|
} |
|
|
|
// AddConfig Add Config |
|
func (d *Dao) AddConfig(script *model.Script) error { |
|
return d.DB.Exec("update script set threads_sum = ?, load_time = ?, ready_time = ? where scene_id = ? and group_id = ?", script.ThreadsSum, script.LoadTime, script.ReadyTime, script.SceneID, script.GroupID).Error |
|
} |
|
|
|
// QueryTree Query Tree |
|
func (d *Dao) QueryTree(script *model.Script) (res *model.ShowTree, err error) { |
|
res = &model.ShowTree{} |
|
err = d.DB.Table(model.Script{}.TableName()).Select("department, project, app").Where(script).Count(&res.IsShow).Order("id").Limit(1).Find(&res.Tree).Error |
|
if res.IsShow > 1 { |
|
res.IsShow = 1 |
|
} |
|
return |
|
} |
|
|
|
//QueryScenesByPage query scripts by page |
|
func (d *Dao) QueryScenesByPage(scene *model.Scene, pn, ps int32, treeNodes []string) (qsr *model.QuerySceneResponse, err error) { |
|
qsr = &model.QuerySceneResponse{} |
|
gDB := d.DB.Table(model.Scene{}.TableName()).Where("app in (?)", treeNodes).Where("is_draft = 0") |
|
if scene.Department != "" && scene.Project != "" && scene.APP != "" { |
|
gDB = gDB.Where("department = ? and project = ? and app = ?", scene.Department, scene.Project, scene.APP) |
|
} |
|
if scene.Department != "" && scene.Project != "" { |
|
gDB = gDB.Where("department = ? and project = ?", scene.Department, scene.Project) |
|
} |
|
if scene.Department != "" { |
|
gDB = gDB.Where("department = ?", scene.Department) |
|
} |
|
if scene.SceneName != "" { |
|
gDB = gDB.Where("scene_name LIKE ? ", "%"+scene.SceneName+"%") |
|
} |
|
if scene.UserName != "" { |
|
gDB = gDB.Where("user_name LIKE ? ", "%"+scene.UserName+"%") |
|
} |
|
|
|
if scene.SceneName == "" && scene.UserName == "" && scene.Department == "" && scene.Project == "" && scene.APP == "" { |
|
gDB = gDB.Where(scene) |
|
} |
|
|
|
err = gDB.Where("is_active = 1").Count(&qsr.TotalSize).Order("ctime desc").Offset((pn - 1) * ps).Limit(ps). |
|
Find(&qsr.Scenes).Error |
|
qsr.PageSize = ps /**/ |
|
qsr.PageNum = pn |
|
return |
|
} |
|
|
|
//QueryScenesByPageWhiteName query scripts by page white name |
|
func (d *Dao) QueryScenesByPageWhiteName(scene *model.Scene, pn, ps int32) (qsr *model.QuerySceneResponse, err error) { |
|
qsr = &model.QuerySceneResponse{} |
|
gDB := d.DB.Table(model.Scene{}.TableName()).Where("is_draft = 0") |
|
if scene.SceneName != "" { |
|
gDB = gDB.Where("scene_name LIKE ? ", "%"+scene.SceneName+"%") |
|
} |
|
if scene.UserName != "" { |
|
gDB = gDB.Where("user_name LIKE ? ", "%"+scene.UserName+"%") |
|
} |
|
|
|
if scene.SceneName == "" && scene.UserName == "" { |
|
gDB = gDB.Where(scene) |
|
} |
|
|
|
err = gDB.Where("is_active = 1").Count(&qsr.TotalSize).Order("mtime desc").Offset((pn - 1) * ps).Limit(ps). |
|
Find(&qsr.Scenes).Error |
|
qsr.PageSize = ps /**/ |
|
qsr.PageNum = pn |
|
return |
|
} |
|
|
|
//QueryScenes query scene |
|
func (d *Dao) QueryScenes(scene *model.Scene, pn int, ps int) (scenes []*model.Scene, err error) { |
|
err = d.DB.Table(model.Scene{}.TableName()).Where(scene).Order("mtime desc").Offset((pn - 1) * ps).Limit(ps).Find(&scenes).Error |
|
return |
|
} |
|
|
|
// QueryExistAPI Query Exist API |
|
func (d *Dao) QueryExistAPI(script *model.Script, pageNum int32, pageSize int32, sceneId int, treeNodes []string) (res *model.APIInfoList, err error) { |
|
res = &model.APIInfoList{} |
|
gDB := d.DB.Table(model.Script{}.TableName()).Where("app in (?)", treeNodes) |
|
if pageSize == 0 && pageNum == 0 { |
|
err = gDB.Where(script).Where("active = 1 and scene_id = 0").Count(&res.TotalSize).Order("id desc").Find(&res.ScriptList).Error |
|
} else { |
|
err = gDB.Where(script).Where("active = 1 and scene_id = 0").Count(&res.TotalSize).Order("id desc").Offset((pageNum - 1) * pageSize).Limit(pageSize).Find(&res.ScriptList).Error |
|
res.PageNum = pageNum |
|
res.PageSize = pageSize |
|
} |
|
res.SceneID = sceneId |
|
return |
|
} |
|
|
|
// QueryGroup Query Group |
|
func (d *Dao) QueryGroup(sceneId int) (res *model.GroupList, err error) { |
|
res = &model.GroupList{} |
|
gDB := d.DB.Table(model.Script{}.TableName()) |
|
//select group_id, count(group_id) from script where scene_id = 1 GROUP BY group_id order by group_id |
|
err = gDB.Select("group_id, threads_sum, load_time, ready_time").Where("scene_id = ? and active = 1", sceneId).Group("group_id").Order("group_id").Find(&res.GroupList).Error |
|
return |
|
} |
|
|
|
// QueryPreview Query Preview |
|
func (d *Dao) QueryPreview(sceneId, groupId int) (res *model.PreviewList, err error) { |
|
res = &model.PreviewList{} |
|
gDB := d.DB.Table(model.Script{}.TableName()) |
|
//select id as test_id, test_name, run_order from script where scene_id = 1 and group_id = 1 order by run_order |
|
err = gDB.Select("id, test_name, run_order, group_id, const_timer, random_timer").Where("scene_id = ? and group_id = ? and active = 1", sceneId, groupId).Order("run_order").Find(&res.PreList).Error |
|
return |
|
} |
|
|
|
// QueryUsefulParams Query Useful Params |
|
func (d *Dao) QueryUsefulParams(sceneId int) (res *model.UsefulParamsList, err error) { |
|
res = &model.UsefulParamsList{} |
|
gDB := d.DB.Table(model.Script{}.TableName()) |
|
err = gDB.Select("output_params").Where("scene_id = ?", sceneId).Where("active = 1 and output_params != '[{\"\":\"\"}]'").Order("id desc").Find(&res.ParamsList).Error |
|
return |
|
} |
|
|
|
// UpdateBindScene Update Bind Scene |
|
func (d *Dao) UpdateBindScene(bindScene *model.BindScene) (err error) { |
|
var id int |
|
tempList := strings.Split(bindScene.ID, ",") |
|
for _, tempId := range tempList { |
|
if id, err = strconv.Atoi(tempId); err != nil { |
|
return err |
|
} |
|
err = d.DB.Model(&model.Script{}).Where("id =?", id).Updates(model.Script{SceneID: bindScene.SceneID}).Error |
|
} |
|
return |
|
} |
|
|
|
// QueryDrawRelation Query Draw Relation |
|
func (d *Dao) QueryDrawRelation(scene *model.Scene) (res *model.SaveOrderReq, err error) { |
|
res = &model.SaveOrderReq{} |
|
gDB := d.DB.Table(model.Script{}.TableName()) |
|
err = gDB.Select("id, test_name, group_id, run_order").Where("scene_id = ? and active = 1", scene.ID).Order("group_id, run_order").Find(&res.GroupOrderList).Error |
|
return |
|
} |
|
|
|
// DeleteDraft Delete Draft |
|
func (d *Dao) DeleteDraft(scene *model.Scene) error { |
|
if scene.ID == 0 { |
|
return d.DB.Model(&model.Scene{}).Where("user_name = ? and is_draft = 1", scene.UserName).Update("is_active", 0).Error |
|
} |
|
return d.DB.Model(&model.Scene{}).Where("user_name = ? and id = ? and is_draft = 1", scene.UserName, scene.ID).Update("is_active", 0).Error |
|
} |
|
|
|
// QueryConfig Query Config |
|
func (d *Dao) QueryConfig(script *model.Script) (res *model.GroupInfo, err error) { |
|
res = &model.GroupInfo{} |
|
gDB := d.DB.Table(model.Script{}.TableName()) |
|
// select distinct group_id, threads_sum, ready_time, load_time from script where scene_id = 282 and active = 1 and group_id = 2 |
|
err = gDB.Select("distinct group_id, threads_sum, ready_time, load_time").Where("scene_id = ? and group_id = ? and active = 1", script.SceneID, script.GroupID).Find(&res).Error |
|
return |
|
} |
|
|
|
// DeleteScene Delete Scene |
|
func (d *Dao) DeleteScene(scene *model.Scene) error { |
|
return d.DB.Model(&model.Scene{}).Where("id = ?", scene.ID).Update("is_active", 0).Error |
|
} |
|
|
|
// QueryFusing Query Fusing |
|
func (d *Dao) QueryFusing(script *model.Script) (res *model.FusingInfoList, err error) { |
|
res = &model.FusingInfoList{} |
|
gDB := d.DB.Table(model.Script{}.TableName()) |
|
//select id as test_id, test_name, run_order from script where scene_id = 1 and group_id = 1 order by run_order |
|
err = gDB.Select("fusing").Where("scene_id = ? and active = 1", script.SceneID).Find(&res.FusingList).Error |
|
return |
|
}
|
|
|