Browse Source

model层部分表删除部分字段

Former-commit-id: 269c3fa7e6
pull/9/head
zhangwei 2 years ago
parent
commit
68cde61a44
8 changed files with 19 additions and 19 deletions
  1. +1
    -1
      api/desc/core/pcm-core.api
  2. +9
    -1
      api/internal/handler/core/gethashcathandler.go
  3. +1
    -1
      api/internal/handler/core/scheduletaskbyyamlhandler.go
  4. +3
    -3
      api/internal/logic/core/gethashcatlogic.go
  5. +3
    -4
      api/internal/logic/core/scheduletaskbyyamllogic.go
  6. +0
    -2
      api/internal/logic/core/scheduletasklogic.go
  7. +1
    -1
      api/internal/types/types.go
  8. +1
    -6
      model/taskmodel_gen.go

+ 1
- 1
api/desc/core/pcm-core.api View File

@@ -468,7 +468,7 @@ type SaveHashcatReq {
} }


type getHashcatHandlerReq { type getHashcatHandlerReq {
CrackTaskId int64 `json:"crackTaskId"` // 任务id
CrackTaskId string `path:"crackTaskId"` // 任务id
} }


type getHashcatHandlerResp { type getHashcatHandlerResp {


+ 9
- 1
api/internal/handler/core/gethashcathandler.go View File

@@ -4,14 +4,22 @@ import (
"gitlink.org.cn/jcce-pcm/utils/result" "gitlink.org.cn/jcce-pcm/utils/result"
"net/http" "net/http"


"github.com/zeromicro/go-zero/rest/httpx"
"gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/logic/core" "gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/logic/core"
"gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/svc" "gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/svc"
"gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/types"
) )


func GetHashcatHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { func GetHashcatHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) { return func(w http.ResponseWriter, r *http.Request) {
var req types.GetHashcatHandlerReq
if err := httpx.Parse(r, &req); err != nil {
httpx.ErrorCtx(r.Context(), w, err)
return
}

l := core.NewGetHashcatLogic(r.Context(), svcCtx) l := core.NewGetHashcatLogic(r.Context(), svcCtx)
resp, err := l.GetHashcat()
resp, err := l.GetHashcat(&req)
result.HttpResult(r, w, resp, err) result.HttpResult(r, w, resp, err)
} }
} }

+ 1
- 1
api/internal/handler/core/scheduletaskbyyamlhandler.go View File

@@ -15,7 +15,7 @@ func ScheduleTaskByYamlHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) { return func(w http.ResponseWriter, r *http.Request) {
var req types.ScheduleTaskByYamlReq var req types.ScheduleTaskByYamlReq
if err := httpx.Parse(r, &req); err != nil { if err := httpx.Parse(r, &req); err != nil {
httpx.ErrorCtx(r.Context(), w, err)
result.HttpResult(r, w, nil, err)
return return
} }
// 解析yaml文件 // 解析yaml文件


+ 3
- 3
api/internal/logic/core/gethashcatlogic.go View File

@@ -25,10 +25,10 @@ func NewGetHashcatLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetHas
} }
} }


func (l *GetHashcatLogic) GetHashcat() (resp *types.GetHashcatHandlerResp, err error) {
func (l *GetHashcatLogic) GetHashcat(req *types.GetHashcatHandlerReq) (resp *types.GetHashcatHandlerResp, err error) {
// todo: add your logic here and delete this line
var hashcatList []*model.THashcat var hashcatList []*model.THashcat
tx := l.svcCtx.DbEngin.Find(&hashcatList)
tx := l.svcCtx.DbEngin.Where("crack_task_id = ?", req.CrackTaskId).Find(&hashcatList)
if tx.Error != nil { if tx.Error != nil {
return nil, tx.Error return nil, tx.Error
} }


+ 3
- 4
api/internal/logic/core/scheduletaskbyyamllogic.go View File

@@ -32,17 +32,15 @@ func (l *ScheduleTaskByYamlLogic) ScheduleTaskByYaml(req *types.ScheduleTaskByYa
if err != nil { if err != nil {
return err return err
} }
// construct task info
// 构建任务结构体
taskModel := model.Task{ taskModel := model.Task{
Status: "Saved", Status: "Saved",
Description: req.Description, Description: req.Description,
Name: req.Name, Name: req.Name,
YamlString: string(bytes), YamlString: string(bytes),
StartTime: time.Now(), StartTime: time.Now(),
CreatedTime: time.Now(),
UpdatedTime: time.Now(),
} }
// save the task in mysql and return id
// 保存任务数据到数据库
tx := l.svcCtx.DbEngin.Create(&taskModel) tx := l.svcCtx.DbEngin.Create(&taskModel)
if tx.Error != nil { if tx.Error != nil {
return tx.Error return tx.Error
@@ -51,6 +49,7 @@ func (l *ScheduleTaskByYamlLogic) ScheduleTaskByYaml(req *types.ScheduleTaskByYa
// push message into topic // push message into topic
for _, task := range req.Tasks { for _, task := range req.Tasks {
task.TaskId = taskModel.Id task.TaskId = taskModel.Id
// 将任务数据转换成消息体
reqMessage, err := json.Marshal(task) reqMessage, err := json.Marshal(task)
if err != nil { if err != nil {
logx.Error(err) logx.Error(err)


+ 0
- 2
api/internal/logic/core/scheduletasklogic.go View File

@@ -44,8 +44,6 @@ func (l *ScheduleTaskLogic) ScheduleTask(req *types.ScheduleTaskReq) (err error)
Name: req.Name, Name: req.Name,
YamlString: string(bytes), YamlString: string(bytes),
StartTime: time.Now(), StartTime: time.Now(),
CreatedTime: time.Now(),
UpdatedTime: time.Now(),
} }
// save the task in mysql and return id // save the task in mysql and return id
tx := l.svcCtx.DbEngin.Create(&taskModel) tx := l.svcCtx.DbEngin.Create(&taskModel)


+ 1
- 1
api/internal/types/types.go View File

@@ -437,7 +437,7 @@ type SaveHashcatReq struct {
} }


type GetHashcatHandlerReq struct { type GetHashcatHandlerReq struct {
CrackTaskId int64 `json:"crackTaskId"` // 任务id
CrackTaskId string `path:"crackTaskId"` // 任务id
} }


type GetHashcatHandlerResp struct { type GetHashcatHandlerResp struct {


+ 1
- 6
model/taskmodel_gen.go View File

@@ -44,12 +44,7 @@ type (
EndTime string `db:"end_time"` // 结束运行时间 EndTime string `db:"end_time"` // 结束运行时间
RunningTime int64 `db:"running_time"` // 已运行时间(单位秒) RunningTime int64 `db:"running_time"` // 已运行时间(单位秒)
YamlString string `db:"yaml_string"` YamlString string `db:"yaml_string"`
Result string `db:"result"` // 作业结果
CreatedBy int64 `db:"created_by"` // 创建人
CreatedTime time.Time `db:"created_time"` // 创建时间
UpdatedBy int64 `db:"updated_by"` // 更新人
UpdatedTime time.Time `db:"updated_time"` // 更新时间
DeletedFlag int64 `db:"deleted_flag"` // 是否删除(0-否,1-是)
Result string `db:"result"` // 作业结果
} }
) )




Loading…
Cancel
Save