EdgeAPI/internal/rpc/services/clients/service_client_agent_ip.go

// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .

package clients

import (
	"context"
	"github.com/TeaOSLab/EdgeAPI/internal/db/models/clients"
	"github.com/TeaOSLab/EdgeAPI/internal/rpc/services"
	rpcutils "github.com/TeaOSLab/EdgeAPI/internal/rpc/utils"
	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
)

// ClientAgentIPService Agent IP服务
type ClientAgentIPService struct {
	services.BaseService
}

// CreateClientAgentIPs 创建一组IP
func (this *ClientAgentIPService) CreateClientAgentIPs(ctx context.Context, req *pb.CreateClientAgentIPsRequest) (*pb.RPCSuccess, error) {
	_, _, err := this.ValidateNodeId(ctx, rpcutils.UserTypeAdmin, rpcutils.UserTypeDNS, rpcutils.UserTypeNode)
	if err != nil {
		return nil, err
	}

	if len(req.AgentIPs) == 0 {
		return this.Success()
	}

	var tx = this.NullTx()
	for _, agentIP := range req.AgentIPs {
		agentId, err := clients.SharedClientAgentDAO.FindAgentIdWithCode(tx, agentIP.AgentCode)
		if err != nil {
			return nil, err
		}
		if agentId <= 0 {
			continue
		}

		err = clients.SharedClientAgentIPDAO.CreateIP(tx, agentId, agentIP.Ip, agentIP.Ptr)
		if err != nil {
			return nil, err
		}
	}

	return this.Success()
}

// ListClientAgentIPsAfterId 查询最新的IP
func (this *ClientAgentIPService) ListClientAgentIPsAfterId(ctx context.Context, req *pb.ListClientAgentIPsAfterIdRequest) (*pb.ListClientAgentIPsAfterIdResponse, error) {
	_, _, err := this.ValidateNodeId(ctx, rpcutils.UserTypeAdmin, rpcutils.UserTypeDNS, rpcutils.UserTypeNode)
	if err != nil {
		return nil, err
	}

	if req.Size <= 0 {
		req.Size = 10000
	}

	var tx = this.NullTx()
	var agentMap = map[int64]*clients.ClientAgent{} // agentId => agentCode
	agentIPs, err := clients.SharedClientAgentIPDAO.ListIPsAfterId(tx, req.Id, req.Size)
	if err != nil {
		return nil, err
	}

	var pbIPs = []*pb.ClientAgentIP{}
	for _, agentIP := range agentIPs {
		var agentId = int64(agentIP.AgentId)
		agent, ok := agentMap[agentId]
		if !ok {
			agent, err = clients.SharedClientAgentDAO.FindAgent(tx, agentId)
			if err != nil {
				return nil, err
			}
			if agent == nil {
				continue
			}
			agentMap[agentId] = agent
		}

		pbIPs = append(pbIPs, &pb.ClientAgentIP{
			Id:  int64(agentIP.Id),
			Ip:  agentIP.IP,
			Ptr: agentIP.Ptr, // 导出时需要
			ClientAgent: &pb.ClientAgent{
				Id:          agentId,
				Name:        "",
				Code:        agent.Code,
				Description: "",
			},
		})
	}

	return &pb.ListClientAgentIPsAfterIdResponse{
		ClientAgentIPs: pbIPs,
	}, nil
}