modelRT/handler/async_task_status_update_handler.go

// Package handler provides HTTP handlers for various endpoints.
package handler

import (
	"modelRT/constants"
	"modelRT/database"
	"modelRT/logger"
	"modelRT/network"
	"modelRT/orm"

	"github.com/gin-gonic/gin"
)

// AsyncTaskStatusUpdateHandler handles updating task status (internal use, not exposed via API)
func AsyncTaskStatusUpdateHandler(c *gin.Context) {
	ctx := c.Request.Context()
	var request network.AsyncTaskStatusUpdate

	if err := c.ShouldBindJSON(&request); err != nil {
		logger.Error(ctx, "failed to unmarshal async task status update request", "error", err)
		renderRespFailure(c, constants.RespCodeInvalidParams, "invalid request parameters", nil)
		return
	}

	validStatus := map[string]bool{
		string(orm.AsyncTaskStatusSubmitted): true,
		string(orm.AsyncTaskStatusRunning):   true,
		string(orm.AsyncTaskStatusCompleted): true,
		string(orm.AsyncTaskStatusFailed):    true,
	}

	if !validStatus[request.Status] {
		logger.Error(ctx, "invalid task status", "status", request.Status)
		renderRespFailure(c, constants.RespCodeInvalidParams, "invalid task status", nil)
		return
	}

	pgClient := database.GetPostgresDBClient()
	if pgClient == nil {
		logger.Error(ctx, "database connection not found in context")
		renderRespFailure(c, constants.RespCodeServerError, "database connection error", nil)
		return
	}

	status := orm.AsyncTaskStatus(request.Status)
	err := database.UpdateAsyncTaskStatus(ctx, pgClient, request.TaskID, status)
	if err != nil {
		logger.Error(ctx, "failed to update async task status", "task_id", request.TaskID, "status", request.Status, "error", err)
		renderRespFailure(c, constants.RespCodeServerError, "failed to update task status", nil)
		return
	}

	if request.Status == string(orm.AsyncTaskStatusCompleted) {
		err = database.CompleteAsyncTask(ctx, pgClient, request.TaskID, request.Timestamp)
	} else if request.Status == string(orm.AsyncTaskStatusFailed) {
		err = database.FailAsyncTask(ctx, pgClient, request.TaskID, request.Timestamp)
	}

	if err != nil {
		logger.Error(ctx, "failed to update async task completion timestamp", "task_id", request.TaskID, "error", err)
		renderRespFailure(c, constants.RespCodeServerError, "failed to update task completion timestamp", nil)
		return
	}

	renderRespSuccess(c, constants.RespCodeSuccess, "task status updated successfully", nil)
}