55 lines
1.5 KiB
Go
55 lines
1.5 KiB
Go
// Package handler provides HTTP handlers for various endpoints.
|
|
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"modelRT/database"
|
|
"modelRT/logger"
|
|
"modelRT/network"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// AsyncTaskProgressUpdateHandler handles updating task progress (internal use, not exposed via API)
|
|
func AsyncTaskProgressUpdateHandler(c *gin.Context) {
|
|
ctx := c.Request.Context()
|
|
var request network.AsyncTaskProgressUpdate
|
|
|
|
if err := c.ShouldBindJSON(&request); err != nil {
|
|
logger.Error(ctx, "failed to unmarshal async task progress update request", "error", err)
|
|
c.JSON(http.StatusOK, network.FailureResponse{
|
|
Code: http.StatusBadRequest,
|
|
Msg: "invalid request parameters",
|
|
})
|
|
return
|
|
}
|
|
|
|
pgClient := database.GetPostgresDBClient()
|
|
if pgClient == nil {
|
|
logger.Error(ctx, "database connection not found in context")
|
|
c.JSON(http.StatusOK, network.FailureResponse{
|
|
Code: http.StatusInternalServerError,
|
|
Msg: "database connection error",
|
|
})
|
|
return
|
|
}
|
|
|
|
// Update task progress
|
|
err := database.UpdateAsyncTaskProgress(ctx, pgClient, request.TaskID, request.Progress)
|
|
if err != nil {
|
|
logger.Error(ctx, "failed to update async task progress", "task_id", request.TaskID, "error", err)
|
|
c.JSON(http.StatusOK, network.FailureResponse{
|
|
Code: http.StatusInternalServerError,
|
|
Msg: "failed to update task progress",
|
|
})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, network.SuccessResponse{
|
|
Code: 2000,
|
|
Msg: "task progress updated successfully",
|
|
Payload: nil,
|
|
})
|
|
}
|