实现IP名单管理

2025-12-26 15:06:37 +08:00 · 2020-11-07 19:40:24 +08:00
parent 5d0f353232
commit 725ff1d9ef
11 changed files with 567 additions and 0 deletions
--- a/internal/rpc/services/service_ip_item.go
+++ b/internal/rpc/services/service_ip_item.go
@@ -0,0 +1,124 @@
+package services
+
+import (
+	"context"
+	"github.com/TeaOSLab/EdgeAPI/internal/db/models"
+	rpcutils "github.com/TeaOSLab/EdgeAPI/internal/rpc/utils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+)
+
+// IP条目相关服务
+type IPItemService struct {
+}
+
+// 创建IP
+func (this *IPItemService) CreateIPItem(ctx context.Context, req *pb.CreateIPItemRequest) (*pb.CreateIPItemResponse, error) {
+	// 校验请求
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	itemId, err := models.SharedIPItemDAO.CreateIPItem(req.IpListId, req.IpFrom, req.IpTo, req.ExpiredAt, req.Reason)
+	if err != nil {
+		return nil, err
+	}
+	return &pb.CreateIPItemResponse{IpItemId: itemId}, nil
+}
+
+// 修改IP
+func (this *IPItemService) UpdateIPItem(ctx context.Context, req *pb.UpdateIPItemRequest) (*pb.RPCUpdateSuccess, error) {
+	// 校验请求
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	err = models.SharedIPItemDAO.UpdateIPItem(req.IpItemId, req.IpFrom, req.IpTo, req.ExpiredAt, req.Reason)
+	if err != nil {
+		return nil, err
+	}
+	return rpcutils.RPCUpdateSuccess()
+}
+
+// 删除IP
+func (this *IPItemService) DeleteIPItem(ctx context.Context, req *pb.DeleteIPItemRequest) (*pb.RPCDeleteSuccess, error) {
+	// 校验请求
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	err = models.SharedIPItemDAO.DisableIPItem(req.IpItemId)
+	if err != nil {
+		return nil, err
+	}
+	return rpcutils.RPCDeleteSuccess()
+}
+
+// 计算IP数量
+func (this *IPItemService) CountIPItemsWithListId(ctx context.Context, req *pb.CountIPItemsWithListIdRequest) (*pb.CountIPItemsWithListIdResponse, error) {
+	// 校验请求
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	count, err := models.SharedIPItemDAO.CountIPItemsWithListId(req.IpListId)
+	if err != nil {
+		return nil, err
+	}
+	return &pb.CountIPItemsWithListIdResponse{Count: count}, nil
+}
+
+// 列出单页的IP
+func (this *IPItemService) ListIPItemsWithListId(ctx context.Context, req *pb.ListIPItemsWithListIdRequest) (*pb.ListIPItemsWithListIdResponse, error) {
+	// 校验请求
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	items, err := models.SharedIPItemDAO.ListIPItemsWithListId(req.IpListId, req.Offset, req.Size)
+	if err != nil {
+		return nil, err
+	}
+	result := []*pb.IPItem{}
+	for _, item := range items {
+		result = append(result, &pb.IPItem{
+			Id:        int64(item.Id),
+			IpFrom:    item.IpFrom,
+			IpTo:      item.IpTo,
+			Version:   int64(item.Version),
+			ExpiredAt: int64(item.ExpiredAt),
+			Reason:    item.Reason,
+		})
+	}
+
+	return &pb.ListIPItemsWithListIdResponse{IpItems: result}, nil
+}
+
+// 查找单个IP
+func (this *IPItemService) FindEnabledIPItem(ctx context.Context, req *pb.FindEnabledIPItemRequest) (*pb.FindEnabledIPItemResponse, error) {
+	// 校验请求
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	item, err := models.SharedIPItemDAO.FindEnabledIPItem(req.IpItemId)
+	if err != nil {
+		return nil, err
+	}
+	if item == nil {
+		return &pb.FindEnabledIPItemResponse{IpItem: nil}, nil
+	}
+	return &pb.FindEnabledIPItemResponse{IpItem: &pb.IPItem{
+		Id:        int64(item.Id),
+		IpFrom:    item.IpFrom,
+		IpTo:      item.IpTo,
+		Version:   int64(item.Version),
+		ExpiredAt: int64(item.ExpiredAt),
+		Reason:    item.Reason,
+	}}, nil
+}
--- a/internal/rpc/services/service_ip_list.go
+++ b/internal/rpc/services/service_ip_list.go
@@ -0,0 +1,67 @@
+package services
+
+import (
+	"context"
+	"github.com/TeaOSLab/EdgeAPI/internal/db/models"
+	rpcutils "github.com/TeaOSLab/EdgeAPI/internal/rpc/utils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+)
+
+// IP名单相关服务
+type IPListService struct {
+}
+
+// 创建IP列表
+func (this *IPListService) CreateIPList(ctx context.Context, req *pb.CreateIPListRequest) (*pb.CreateIPListResponse, error) {
+	// 校验请求
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	listId, err := models.SharedIPListDAO.CreateIPList(req.Type, req.Name, req.Code, req.TimeoutJSON)
+	if err != nil {
+		return nil, err
+	}
+	return &pb.CreateIPListResponse{IpListId: listId}, nil
+}
+
+// 修改IP列表
+func (this *IPListService) UpdateIPList(ctx context.Context, req *pb.UpdateIPListRequest) (*pb.RPCUpdateSuccess, error) {
+	// 校验请求
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	err = models.SharedIPListDAO.UpdateIPList(req.IpListId, req.Name, req.Code, req.TimeoutJSON)
+	if err != nil {
+		return nil, err
+	}
+	return rpcutils.RPCUpdateSuccess()
+}
+
+// 查找IP列表
+func (this *IPListService) FindEnabledIPList(ctx context.Context, req *pb.FindEnabledIPListRequest) (*pb.FindEnabledIPListResponse, error) {
+	// 校验请求
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	list, err := models.SharedIPListDAO.FindEnabledIPList(req.IpListId)
+	if err != nil {
+		return nil, err
+	}
+	if list == nil {
+		return &pb.FindEnabledIPListResponse{IpList: nil}, nil
+	}
+	return &pb.FindEnabledIPListResponse{IpList: &pb.IPList{
+		Id:          int64(list.Id),
+		IsOn:        list.IsOn == 1,
+		Type:        list.Type,
+		Name:        list.Name,
+		Code:        list.Code,
+		TimeoutJSON: []byte(list.Timeout),
+	}}, nil
+}