From dee975c73c5736beab8c2c6857271b61d8c6c660 Mon Sep 17 00:00:00 2001
From: tzwang <wtz1003@sina.com>
Date: Tue, 24 Jun 2025 17:11:20 +0800
Subject: [PATCH] update api

---
 http/apis/trainjob.go    | 31 +++++++++++++++++++++++++++++++
 http/model/trainjob.go   |  4 ++++
 http/router/router.go    |  1 +
 http/service/trainjob.go | 18 ++++++++++++++++++
 4 files changed, 54 insertions(+)

diff --git a/http/apis/trainjob.go b/http/apis/trainjob.go
index 94405ee..e3b8eba 100644
--- a/http/apis/trainjob.go
+++ b/http/apis/trainjob.go
@@ -110,3 +110,34 @@ func TrainJobDetail(ctx *gin.Context) {
 
 	model.Response(ctx, http.StatusOK, common.SUCCESS, resp.Payload)
 }
+
+func TrainJobLog(ctx *gin.Context) {
+	var param model.TrainJobLog
+	if err := ctx.BindJSON(&param); err != nil {
+		model.Response(ctx, http.StatusBadRequest, common.INVALIDPARAMS, err)
+		return
+	}
+
+	ip := ctx.Query(common.IPADDR)
+	if ip == "" {
+		model.Response(ctx, 401, "addr为必填字段", nil)
+		return
+	}
+	token := ctx.Query(common.ACCESSTOKEN)
+	if token == "" {
+		model.Response(ctx, 401, "token为必填字段", nil)
+		return
+	}
+
+	resp, err := service.TrainJobLog(ip, token, &param)
+	if err != nil {
+		model.Response(ctx, 500, common.INVOKEERROR, err)
+		return
+	}
+
+	if resp == "" {
+		model.Response(ctx, 500, common.INVOKEERROR, nil)
+	}
+
+	model.Response(ctx, http.StatusOK, common.SUCCESS, resp)
+}
diff --git a/http/model/trainjob.go b/http/model/trainjob.go
index 3991950..c31406f 100644
--- a/http/model/trainjob.go
+++ b/http/model/trainjob.go
@@ -148,3 +148,7 @@ type KS struct {
 	Key   string `json:"key"`
 	State string `json:"state"`
 }
+
+type TrainJobLog struct {
+	JobId string `json:"jobId"`
+}
diff --git a/http/router/router.go b/http/router/router.go
index 4df941c..fb708f3 100644
--- a/http/router/router.go
+++ b/http/router/router.go
@@ -69,6 +69,7 @@ func Create() (*gin.Engine, error) {
 		job.POST("/create", apis.CreateTrainJob)
 		job.GET("/list", apis.GetTrainJobList)
 		job.GET("/detail", apis.TrainJobDetail)
+		job.GET("/log", apis.TrainJobLog)
 
 		//billing
 		bill := v1.Group("bill")
diff --git a/http/service/trainjob.go b/http/service/trainjob.go
index 9452eae..86c4044 100644
--- a/http/service/trainjob.go
+++ b/http/service/trainjob.go
@@ -62,3 +62,21 @@ func TrainJobDetail(ip, token string, param *model.TrainJobDetailParam) (*model.
 	}
 	return resp, nil
 }
+
+func TrainJobLog(ip, token string, param *model.TrainJobLog) (string, error) {
+	var reqUrl = ip + common.Forward_Slash + common.GetTrainJobLog
+
+	req := common.GetRestyRequest()
+	resp, err := req.
+		SetHeader("Authorization", "Bearer "+token).
+		SetPathParam("taskId", param.JobId).
+		SetPathParam("taskNum", "task0").
+		SetPathParam("num", "0").
+		SetBody(param).
+		Get(reqUrl)
+
+	if err != nil {
+		return "", err
+	}
+	return resp.String(), nil
+}