2022-12-20 21:45:52 +08:00
|
|
|
|
package rpc
|
|
|
|
|
|
|
|
|
|
|
|
import (
|
2023-04-09 13:35:34 +08:00
|
|
|
|
"github.com/furryboard/spider-scheduler/pkg/conf"
|
|
|
|
|
|
"github.com/furryboard/spider-scheduler/pkg/log"
|
|
|
|
|
|
"github.com/furryboard/spider-scheduler/rpc/pb"
|
2022-12-20 21:45:52 +08:00
|
|
|
|
"google.golang.org/grpc"
|
|
|
|
|
|
"google.golang.org/grpc/credentials/insecure"
|
|
|
|
|
|
"strconv"
|
|
|
|
|
|
"sync"
|
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
var (
|
|
|
|
|
|
Conn *grpc.ClientConn
|
|
|
|
|
|
biliAPI pb.BiliAPIClient
|
|
|
|
|
|
once sync.Once
|
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
func SpiderCore() pb.BiliAPIClient {
|
|
|
|
|
|
once.Do(InitBiliAPI)
|
|
|
|
|
|
return biliAPI
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
func InitBiliAPI() {
|
|
|
|
|
|
addr := conf.Conf.SpiderCore.Host + ":" + strconv.Itoa(int(conf.Conf.SpiderCore.Port))
|
|
|
|
|
|
Conn, err := grpc.Dial(addr, grpc.WithTransportCredentials(insecure.NewCredentials()))
|
|
|
|
|
|
if err != nil {
|
|
|
|
|
|
log.Logger().Fatalf("连接到RPC服务器时发生错误：%s", err)
|
|
|
|
|
|
}
|
|
|
|
|
|
biliAPI = pb.NewBiliAPIClient(Conn)
|
|
|
|
|
|
}
|
