chore: 并入 iotlcm-admin-svr 历史到 admin-svr/(原提交 49818c6)
This commit is contained in:
@@ -0,0 +1,45 @@
|
||||
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
|
||||
}
|
||||
Reference in New Issue
Block a user