mirror of
https://github.com/TeaOSLab/EdgeAPI.git
synced 2025-11-05 17:40:24 +08:00
75 lines
2.0 KiB
Go
75 lines
2.0 KiB
Go
package apis
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/TeaOSLab/EdgeAPI/internal/configs"
|
|
"github.com/TeaOSLab/EdgeAPI/internal/rpc/admin"
|
|
"github.com/TeaOSLab/EdgeAPI/internal/rpc/dns"
|
|
"github.com/TeaOSLab/EdgeAPI/internal/rpc/log"
|
|
"github.com/TeaOSLab/EdgeAPI/internal/rpc/monitor"
|
|
"github.com/TeaOSLab/EdgeAPI/internal/rpc/node"
|
|
"github.com/TeaOSLab/EdgeAPI/internal/rpc/provider"
|
|
"github.com/TeaOSLab/EdgeAPI/internal/rpc/stat"
|
|
"github.com/TeaOSLab/EdgeAPI/internal/rpc/user"
|
|
"github.com/TeaOSLab/EdgeAPI/internal/utils"
|
|
"github.com/iwind/TeaGo/logs"
|
|
"google.golang.org/grpc"
|
|
"net"
|
|
"os"
|
|
"strconv"
|
|
)
|
|
|
|
var sharedAPIConfig *configs.APIConfig = nil
|
|
|
|
type APINode struct {
|
|
}
|
|
|
|
func NewAPINode() *APINode {
|
|
return &APINode{}
|
|
}
|
|
|
|
func (this *APINode) Start() {
|
|
logs.Println("[API]start api node, pid: " + strconv.Itoa(os.Getpid()))
|
|
|
|
config, err := configs.SharedAPIConfig()
|
|
if err != nil {
|
|
logs.Println("[API]start failed: " + err.Error())
|
|
return
|
|
}
|
|
sharedAPIConfig = config
|
|
|
|
// 设置rlimit
|
|
_ = utils.SetRLimit(1024 * 1024)
|
|
|
|
// 监听RPC服务
|
|
logs.Println("[API]start rpc: " + config.RPC.Listen)
|
|
err = this.listenRPC()
|
|
if err != nil {
|
|
logs.Println(err.Error())
|
|
return
|
|
}
|
|
}
|
|
|
|
// 启动RPC监听
|
|
func (this *APINode) listenRPC() error {
|
|
listener, err := net.Listen("tcp", sharedAPIConfig.RPC.Listen)
|
|
if err != nil {
|
|
return errors.New("[API]listen rpc failed: " + err.Error())
|
|
}
|
|
rpcServer := grpc.NewServer()
|
|
dns.RegisterServiceServer(rpcServer, &dns.Service{})
|
|
log.RegisterServiceServer(rpcServer, &log.Service{})
|
|
monitor.RegisterServiceServer(rpcServer, &monitor.Service{})
|
|
node.RegisterServiceServer(rpcServer, &node.Service{})
|
|
provider.RegisterServiceServer(rpcServer, &provider.Service{})
|
|
stat.RegisterServiceServer(rpcServer, &stat.Service{})
|
|
user.RegisterServiceServer(rpcServer, &user.Service{})
|
|
admin.RegisterServiceServer(rpcServer, &admin.Service{})
|
|
err = rpcServer.Serve(listener)
|
|
if err != nil {
|
|
return errors.New("[API]start rpc failed: " + err.Error())
|
|
}
|
|
|
|
return nil
|
|
}
|