56 lines
2.0 KiB
Go
56 lines
2.0 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"github.com/topfans/backend/pkg/logger"
|
|
pbUser "github.com/topfans/backend/pkg/proto/user"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
type UserServiceClient interface {
|
|
UpdateCrystalBalance(ctx context.Context, userID, starID int64, delta int64) (int64, error)
|
|
AddExperience(ctx context.Context, userID, starID int64, delta int64) (int64, error)
|
|
}
|
|
|
|
type userServiceClient struct {
|
|
client pbUser.UserSocialService
|
|
}
|
|
|
|
func NewUserServiceClient(client pbUser.UserSocialService) UserServiceClient {
|
|
return &userServiceClient{client: client}
|
|
}
|
|
|
|
func (c *userServiceClient) UpdateCrystalBalance(ctx context.Context, userID, starID int64, delta int64) (int64, error) {
|
|
logger.Logger.Debug("Calling UserService.UpdateCrystalBalance",
|
|
zap.Int64("user_id", userID), zap.Int64("star_id", starID), zap.Int64("delta", delta))
|
|
resp, err := c.client.UpdateCrystalBalance(ctx, &pbUser.UpdateCrystalBalanceRequest{
|
|
UserId: userID, StarId: starID, Delta: delta,
|
|
})
|
|
if err != nil {
|
|
logger.Logger.Error("UserService.UpdateCrystalBalance failed", zap.Error(err))
|
|
return 0, err
|
|
}
|
|
if resp.Base.Code != 0 {
|
|
logger.Logger.Warn("UpdateCrystalBalance non-zero code", zap.Int32("code", int32(resp.Base.Code)))
|
|
return 0, err
|
|
}
|
|
return resp.NewBalance, nil
|
|
}
|
|
|
|
func (c *userServiceClient) AddExperience(ctx context.Context, userID, starID int64, delta int64) (int64, error) {
|
|
logger.Logger.Debug("Calling UserService.AddExperience",
|
|
zap.Int64("user_id", userID), zap.Int64("star_id", starID), zap.Int64("delta", delta))
|
|
resp, err := c.client.AddExperience(ctx, &pbUser.AddExperienceRequest{
|
|
UserId: userID, StarId: starID, Delta: delta,
|
|
})
|
|
if err != nil {
|
|
logger.Logger.Error("UserService.AddExperience failed", zap.Error(err))
|
|
return 0, err
|
|
}
|
|
if resp.Base.Code != 0 {
|
|
logger.Logger.Warn("AddExperience non-zero code", zap.Int32("code", int32(resp.Base.Code)))
|
|
return 0, err
|
|
}
|
|
return resp.NewExperience, nil
|
|
}
|