MiFaZhan
46 lines
1.9 KiB
Go
46 lines
1.9 KiB
Go
package iot
|
|
|
|
import (
|
|
"context"
|
|
"strconv"
|
|
|
|
"github.com/gogf/gf/v2/errors/gcode"
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
|
|
v1 "git.5iot.com/iotlcm-public/iotlcm-admin-svr/api/iot/v1"
|
|
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/contexts"
|
|
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/service/fullsyncretry"
|
|
)
|
|
|
|
type FullSyncReplayController struct{ client fullsyncretry.Client }
|
|
|
|
func NewFullSyncReplay() *FullSyncReplayController {
|
|
return &FullSyncReplayController{client: fullsyncretry.FromEnvironment()}
|
|
}
|
|
|
|
func (c *FullSyncReplayController) Replay(ctx context.Context, req *v1.FullSyncReplayReq) (*v1.FullSyncReplayRes, error) {
|
|
userID := contexts.GetUserId(ctx)
|
|
if userID == 0 {
|
|
return nil, gerror.NewCode(gcode.CodeNotAuthorized, "Login required")
|
|
}
|
|
receipt, err := c.client.Replay(ctx, req.SourceRunID, strconv.FormatInt(userID, 10), req.MaxRounds)
|
|
if err != nil {
|
|
g.Log().Warningf(ctx, "full_sync_replay rejected actor=%d sourceRunId=%s maxRounds=%d", userID, req.SourceRunID, req.MaxRounds)
|
|
return nil, err
|
|
}
|
|
g.Log().Infof(ctx, "full_sync_replay accepted actor=%d sourceRunId=%s maxRounds=%d replayRunId=%s", userID, req.SourceRunID, req.MaxRounds, receipt.ReplayRunID)
|
|
return &v1.FullSyncReplayRes{ReplayRunID: receipt.ReplayRunID, Status: receipt.Status}, nil
|
|
}
|
|
|
|
func (c *FullSyncReplayController) Status(ctx context.Context, req *v1.FullSyncReplayStatusReq) (*v1.FullSyncReplayStatusRes, error) {
|
|
if contexts.GetUserId(ctx) == 0 {
|
|
return nil, gerror.NewCode(gcode.CodeNotAuthorized, "Login required")
|
|
}
|
|
status, err := c.client.Get(ctx, req.ReplayRunID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &v1.FullSyncReplayStatusRes{ReplayRunID: status.ReplayRunID, SourceRunID: status.SourceRunID, Status: status.Status, Enqueued: status.Enqueued, Remaining: status.Remaining, RunStatus: status.RunStatus, Error: status.Error}, nil
|
|
}
|