modelRT/handler/async_task_progress_update_handler.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,
	})
}