diff --git a/api/desc/core/pcm-core.api b/api/desc/core/pcm-core.api index 0d4fcc8c..76600544 100644 --- a/api/desc/core/pcm-core.api +++ b/api/desc/core/pcm-core.api @@ -565,11 +565,12 @@ type NodeAsset { ParticipantId int64 `json:"ParticipantId"` // 集群动态信息id } -type crackProgressReq { - CrackTaskId string `form:"crackTaskId"` +type crackProgressResp { + crackProgressList []CrackProgress `json:"crackProgressList"` } -type crackProgressResp { +type CrackProgress { + Name string `json:"name"` Progress string `json:"progress"` Current string `json:"current"` Total string `json:"total"` diff --git a/api/desc/pcm.api b/api/desc/pcm.api index c850e6f4..a5f260c3 100644 --- a/api/desc/pcm.api +++ b/api/desc/pcm.api @@ -119,7 +119,7 @@ service pcm { @doc "Hashcat Crack Progress" @handler crackProgressHandler - get /core/crack/progress (crackProgressReq) returns (crackProgressResp) + get /core/crack/progress returns (crackProgressResp) @doc "Resource Center Information" @handler resourceCenterInfoHandler diff --git a/api/internal/handler/core/crackprogresshandler.go b/api/internal/handler/core/crackprogresshandler.go index ddebe890..14c00021 100644 --- a/api/internal/handler/core/crackprogresshandler.go +++ b/api/internal/handler/core/crackprogresshandler.go @@ -1,28 +1,17 @@ package core import ( + "gitlink.org.cn/jcce-pcm/pcm-coordinator/pkg/repository/result" "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/svc" - "gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/types" ) func CrackProgressHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { - var req types.CrackProgressReq - if err := httpx.Parse(r, &req); err != nil { - httpx.ErrorCtx(r.Context(), w, err) - return - } - l := core.NewCrackProgressLogic(r.Context(), svcCtx) - resp, err := l.CrackProgress(&req) - if err != nil { - httpx.ErrorCtx(r.Context(), w, err) - } else { - httpx.OkJsonCtx(r.Context(), w, resp) - } + resp, err := l.CrackProgress() + result.HttpResult(r, w, resp, err) } } diff --git a/api/internal/logic/core/crackprogresslogic.go b/api/internal/logic/core/crackprogresslogic.go index 52bd0ec5..91bc9b50 100644 --- a/api/internal/logic/core/crackprogresslogic.go +++ b/api/internal/logic/core/crackprogresslogic.go @@ -2,7 +2,6 @@ package core import ( "context" - "gitlink.org.cn/jcce-pcm/pcm-coordinator/pkg/models" "strings" "gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/svc" @@ -25,24 +24,29 @@ func NewCrackProgressLogic(ctx context.Context, svcCtx *svc.ServiceContext) *Cra } } -func (l *CrackProgressLogic) CrackProgress(req *types.CrackProgressReq) (resp *types.CrackProgressResp, err error) { +func (l *CrackProgressLogic) CrackProgress() (resp *types.CrackProgressResp, err error) { + // todo: add your logic here and delete this line result := types.CrackProgressResp{} - var hashcat *models.THashcat - tx := l.svcCtx.DbEngin.Where("crack_task_id = ?", req.CrackTaskId).Order("id desc").Limit(1).Find(&hashcat) + var crackProgressList []*types.CrackProgress + tx := l.svcCtx.DbEngin.Raw("SELECT h.id, h.speed, h.crack_progress as progress,c.`name` \nFROM t_hashcat h \nJOIN ( \n SELECT crack_task_id, MAX(id) as max_id \n FROM t_hashcat \n GROUP BY crack_task_id \n) AS sub_query ON h.crack_task_id = sub_query.crack_task_id AND h.id = sub_query.max_id \nJOIN cloud c ON h.crack_task_id = c.`name`").Scan(&crackProgressList) if tx.Error != nil { return nil, tx.Error } - if len(hashcat.Speed) != 0 { - str1 := strings.Split(hashcat.Speed, "@") - result.Speed = str1[0] - } - if len(hashcat.CrackProgress) != 0 { - str1 := strings.Split(hashcat.CrackProgress, "/") - result.Current = str1[0] - str2 := strings.Split(str1[1], " (") - result.Total = str2[0] - str3 := strings.Split(str2[1], "%") - result.Progress = str3[0] + for _, crackProgress := range crackProgressList { + if len(crackProgress.Speed) != 0 { + str1 := strings.Split(crackProgress.Speed, "@") + crackProgress.Speed = str1[0] + } + if len(crackProgress.Progress) != 0 { + str1 := strings.Split(crackProgress.Progress, "/") + crackProgress.Current = str1[0] + str2 := strings.Split(str1[1], " (") + crackProgress.Total = str2[0] + str3 := strings.Split(str2[1], "%") + crackProgress.Progress = str3[0] + } + result.CrackProgressList = append(result.CrackProgressList, *crackProgress) } + return &result, nil } diff --git a/api/internal/types/types.go b/api/internal/types/types.go index 37a4de44..5c017a79 100644 --- a/api/internal/types/types.go +++ b/api/internal/types/types.go @@ -517,11 +517,12 @@ type NodeAsset struct { ParticipantId int64 `json:"ParticipantId"` // 集群动态信息id } -type CrackProgressReq struct { - CrackTaskId string `form:"crackTaskId"` +type CrackProgressResp struct { + CrackProgressList []CrackProgress `json:"crackProgressList"` } -type CrackProgressResp struct { +type CrackProgress struct { + Name string `json:"name"` Progress string `json:"progress"` Current string `json:"current"` Total string `json:"total"`