85 lines
1.8 KiB
Go
85 lines
1.8 KiB
Go
package svc
|
|
|
|
import (
|
|
"context"
|
|
store "lone-services/rpc/store/pb"
|
|
user "lone-services/rpc/user/pb"
|
|
"lone-services/services/equipment/internal/dao"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
func GetUser(id int64, name string) dao.UserInfo {
|
|
var userInfo dao.UserInfo
|
|
cli, err := GetRpcClient(name)
|
|
if err != nil {
|
|
logx.Error(err)
|
|
return userInfo
|
|
}
|
|
|
|
client := user.NewUserClient(cli.Conn())
|
|
info, err := client.InternalGetUserInfo(context.Background(), &user.InternalGetUserInfoReq{UserId: id})
|
|
if err != nil {
|
|
logx.Errorf("get user err: %v", err)
|
|
return userInfo
|
|
}
|
|
|
|
userInfo.Id = info.Id
|
|
|
|
return userInfo
|
|
}
|
|
|
|
func GetStore(id int64, name string) dao.StoreInfo {
|
|
var storeInfo dao.StoreInfo
|
|
cli, err := GetRpcClient(name)
|
|
if err != nil {
|
|
logx.Error(err)
|
|
return storeInfo
|
|
}
|
|
|
|
client := store.NewStoreClient(cli.Conn())
|
|
info, err := client.GetInfoById(context.Background(), &store.IdRequest{Id: id})
|
|
if err != nil {
|
|
logx.Errorf("get store err: %v", err)
|
|
return storeInfo
|
|
}
|
|
|
|
storeInfo.Id = info.Id
|
|
|
|
return storeInfo
|
|
}
|
|
|
|
func GetStores(ids []int64, name string) []*store.InfoResponse {
|
|
cli, err := GetRpcClient(name)
|
|
if err != nil {
|
|
logx.Error(err)
|
|
return nil
|
|
}
|
|
|
|
client := store.NewStoreClient(cli.Conn())
|
|
items, err := client.GetInfoByIds(context.Background(), &store.IdsRequest{Ids: ids})
|
|
if err != nil {
|
|
logx.Errorf("get store err: %v", err)
|
|
return nil
|
|
}
|
|
|
|
return items.Items
|
|
}
|
|
|
|
func GetUsers(ids []int64, name string) []*user.UserBriefItem {
|
|
cli, err := GetRpcClient(name)
|
|
if err != nil {
|
|
logx.Error(err)
|
|
return nil
|
|
}
|
|
|
|
client := user.NewUserClient(cli.Conn())
|
|
items, err := client.InternalUsersByIds(context.Background(), &user.InternalUsersByIdsReq{Ids: ids})
|
|
if err != nil {
|
|
logx.Errorf("get store err: %v", err)
|
|
return nil
|
|
}
|
|
|
|
return items.Items
|
|
}
|