27 lines
695 B
Go
27 lines
695 B
Go
// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
|
|
|
|
package financeutils
|
|
|
|
import (
|
|
"context"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
|
"github.com/TeaOSLab/EdgeUser/internal/rpc"
|
|
)
|
|
|
|
// FindUserBalance 读取用户余额
|
|
func FindUserBalance(ctx context.Context) (total float64, err error) {
|
|
rpcClient, err := rpc.SharedRPC()
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
accountResp, err := rpcClient.UserAccountRPC().FindEnabledUserAccountWithUserId(ctx, &pb.FindEnabledUserAccountWithUserIdRequest{})
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
if accountResp.UserAccount != nil {
|
|
return accountResp.UserAccount.Total, nil
|
|
}
|
|
return 0, nil
|
|
}
|