frp-panel/services/clientrpc/rpc_service.go

package clientrpc

import (
	"context"

	"github.com/VaalaCat/frp-panel/pb"
	"github.com/VaalaCat/frp-panel/services/app"
	"github.com/VaalaCat/frp-panel/utils/logger"
)

type ClientRPCHandler interface {
	Run()
	Stop()
	GetCli() pb.MasterClient
}

type clientRPCHandler struct {
	appInstance  app.Application
	rpcClient    app.MasterClient
	done         chan bool
	handerFunc   func(appInstance app.Application, req *pb.ServerMessage) *pb.ClientMessage
	clientID     string
	clientSecret string
	event        pb.Event
}

func NewClientRPCHandler(
	appInstance app.Application,
	clientID,
	clientSecret string,
	event pb.Event,
	handerFunc func(appInstance app.Application, req *pb.ServerMessage) *pb.ClientMessage,
) app.ClientRPCHandler {
	rpcCli := appInstance.GetMasterCli()
	done := make(chan bool)
	return &clientRPCHandler{
		appInstance:  appInstance,
		rpcClient:    rpcCli,
		done:         done,
		handerFunc:   handerFunc,
		clientID:     clientID,
		clientSecret: clientSecret,
		event:        event,
	}
}

func (s *clientRPCHandler) Run() {
	defer func() {
		if err := recover(); err != nil {
			logger.Logger(context.Background()).Fatalf("client rpc handler panic: %v", err)
		}
	}()

	startClientRpcHandler(s.appInstance, s.rpcClient, s.done, s.clientID, s.clientSecret, s.event, s.handerFunc)
}

func (s *clientRPCHandler) Stop() {
	close(s.done)
}

func (s *clientRPCHandler) GetCli() app.MasterClient {
	return s.rpcClient
}