first commit

apps/gateway/internal/logic/whitelistr/addwhitelistlogic.go

@@ -0,0 +1,46 @@
+package whitelistr
+
+import (
+	"context"
+	"errors"
+	"tianyuan-api/apps/gateway/internal/validator"
+	"tianyuan-api/apps/sentinel/client/whitelist"
+
+	"tianyuan-api/apps/gateway/internal/svc"
+	"tianyuan-api/apps/gateway/internal/types"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type AddWhitelistLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewAddWhitelistLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AddWhitelistLogic {
+	return &AddWhitelistLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *AddWhitelistLogic) AddWhitelist(req *types.AddWhitelistReq) error {
+	userId, ok := l.ctx.Value("userId").(int64)
+	if !ok {
+		return errors.New("无法获取 userId")
+	}
+	isIp := validator.IsValidIPAddress(req.Ip)
+	if !isIp {
+		return errors.New("请输入正确的IP地址")
+	}
+	_, err := l.svcCtx.WhitelistRpc.CreateWhitelist(l.ctx, &whitelist.CreateWhitelistRequest{
+		UserId:      userId,
+		WhitelistIp: req.Ip,
+	})
+	if err != nil {
+		return err
+	}
+	return nil
+}

apps/gateway/internal/logic/whitelistr/deletewhitelistlogic.go

@@ -0,0 +1,35 @@
+package whitelistr
+
+import (
+	"context"
+	"tianyuan-api/apps/sentinel/client/whitelist"
+
+	"tianyuan-api/apps/gateway/internal/svc"
+	"tianyuan-api/apps/gateway/internal/types"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type DeleteWhitelistLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewDeleteWhitelistLogic(ctx context.Context, svcCtx *svc.ServiceContext) *DeleteWhitelistLogic {
+	return &DeleteWhitelistLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *DeleteWhitelistLogic) DeleteWhitelist(req *types.DeleteWhitelistReq) error {
+	_, err := l.svcCtx.WhitelistRpc.DeleteWhitelist(l.ctx, &whitelist.DeleteWhitelistRequest{
+		Id: req.Id,
+	})
+	if err != nil {
+		return err
+	}
+	return nil
+}

apps/gateway/internal/logic/whitelistr/getwhitelistlistlogic.go

@@ -0,0 +1,55 @@
+package whitelistr
+
+import (
+	"context"
+	"errors"
+	"tianyuan-api/apps/sentinel/client/whitelist"
+
+	"tianyuan-api/apps/gateway/internal/svc"
+	"tianyuan-api/apps/gateway/internal/types"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type GetWhitelistListLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewGetWhitelistListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetWhitelistListLogic {
+	return &GetWhitelistListLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *GetWhitelistListLogic) GetWhitelistList(req *types.GetWhitelistListReq) (resp *types.GetWhitelistListResp, err error) {
+	userId, ok := l.ctx.Value("userId").(int64)
+	if !ok {
+		return nil, errors.New("无法获取 userId")
+	}
+	whitelistPageResp, err := l.svcCtx.WhitelistRpc.GetWhitePageList(l.ctx, &whitelist.WhitePageListRequest{
+		UserId:   userId,
+		Page:     req.Page,
+		PageSize: req.PageSize,
+	})
+	if err != nil {
+		return nil, err
+	}
+	var list []types.WhitelistItem
+	for _, up := range whitelistPageResp.Whitelists {
+		list = append(list, types.WhitelistItem{
+			Id:          up.Id,
+			WhitelistIp: up.WhitelistIp,
+			CreatedAt:   up.CreatedAt,
+			UpdatedAt:   up.UpdatedAt,
+		})
+	}
+	resp = &types.GetWhitelistListResp{
+		Total: whitelistPageResp.Total,
+		List:  list,
+	}
+	return resp, nil
+}