add apis

services/equipment/internal/svc/servicecontext.go

@@ -1,23 +1,98 @@
 package svc
 
 import (
+	"fmt"
+	"lone-services/pkg/discovery"
 	"lone-services/pkg/utils"
 	"lone-services/services/equipment/internal/config"
+	"net"
+	"strconv"
+	"sync"
+	"time"
 
+	"github.com/zeromicro/go-zero/core/logx"
+	"github.com/zeromicro/go-zero/zrpc"
 	"gorm.io/gorm"
 )
 
+type rpcClientCacheEntry struct {
+	cli      zrpc.Client
+	target   string
+	expireAt time.Time
+}
+
+const rpcClientCacheTTL = 30 * time.Second //缓存过期时间
+
+var (
+	rpcCacheLock sync.Mutex
+	rpcCache     = make(map[string]*rpcClientCacheEntry)
+)
+
+func GetRpcClient(serviceName string) (zrpc.Client, error) {
+	if serviceName == utils.StringEmpty {
+		err := fmt.Errorf("rpc serviceName is empty")
+		logx.Error(err)
+		return nil, err
+	}
+
+	cacheKey := serviceName
+
+	rpcCacheLock.Lock()
+	entry, ok := rpcCache[cacheKey]
+	if ok && time.Now().Before(entry.expireAt) {
+		rpcCacheLock.Unlock()
+		return entry.cli, nil
+	}
+	delete(rpcCache, cacheKey)
+	rpcCacheLock.Unlock()
+
+	inst, err := discovery.Pick(serviceName)
+	if err != nil {
+		err = fmt.Errorf("discovery pick %s failed: %w", serviceName, err)
+		logx.Error(err)
+		return nil, err
+	}
+
+	target := net.JoinHostPort(inst.IP, strconv.FormatUint(inst.Port, utils.NumberTen))
+
+	cli := zrpc.MustNewClient(zrpc.RpcClientConf{
+		Target:  target,
+		Timeout: utils.RpcTimeOut, //rpc调用超时5s
+	})
+
+	rpcCacheLock.Lock()
+	rpcCache[cacheKey] = &rpcClientCacheEntry{
+		cli:      cli,
+		target:   target,
+		expireAt: time.Now().Add(rpcClientCacheTTL),
+	}
+	rpcCacheLock.Unlock()
+
+	return cli, nil
+}
+
 type ServiceContext struct {
-	Config config.Config
-	DB     *gorm.DB
-	Prefix string
+	Config       config.Config
+	DB           *gorm.DB
+	Prefix       string
+	StoreSvcName string
+	UserSvcName  string
 }
 
 func NewServiceContext(c config.Config, db *gorm.DB) *ServiceContext {
-
+	storeSvc := utils.GetConfigString("services.store")
+	if storeSvc == utils.StringEmpty {
+		logx.Error("config services.store empty")
+	}
+	userSvc := utils.GetConfigString("services.user")
+	if userSvc == utils.StringEmpty {
+		logx.Error("config services.user empty")
+	}
 	return &ServiceContext{
-		Config: c,
-		DB:     db,
-		Prefix: utils.GetConfigString("mysql.prefix"),
+		Config:       c,
+		DB:           db,
+		Prefix:       utils.GetConfigString("mysql.prefix"),
+		StoreSvcName: storeSvc,
+		UserSvcName:  userSvc,
 	}
 }