team/tianyuan-api-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

temp

Browse Source
This commit is contained in:
liangzai
2024-10-12 20:41:55 +08:00
parent 8c09120db6
commit 597e4f1b89
75 changed files with 5009 additions and 823 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
apps/user/internal/logic/apirequestservice/addapirequestlogic.go Normal file
View File

@@ -0,0 +1,52 @@
package apirequestservicelogic
import (
"context"
"tianyuan-api/apps/user/internal/model"
"tianyuan-api/apps/user/internal/svc"
"tianyuan-api/apps/user/user"
"tianyuan-api/pkg/sqlutil"
"time"
"github.com/zeromicro/go-zero/core/logx"
)
type AddApiRequestLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewAddApiRequestLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AddApiRequestLogic {
return &AddApiRequestLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
// 添加API请求记录
func (l *AddApiRequestLogic) AddApiRequest(in *user.AddApiRequestRequest) (*user.AddApiRequestResponse, error) {
chargesValue := int64(0)
if in.Charges {
chargesValue = 1
}
parsedTime, err := time.Parse(time.RFC3339Nano, in.Timestamp)
if err != nil {
// 错误处理，比如日志输出或返回错误
parsedTime = time.Now()
}
_, err = l.svcCtx.ApiRequestsModel.Insert(l.ctx, &model.ApiRequests{
TransactionId: in.TransactionId,
UserId: in.UserId,
ProductCode: in.ProductCode,
Status: in.Status,
Charges: chargesValue,
Remark: sqlutil.StringToNullString(in.Remark),
Timestamp: parsedTime,
})
if err != nil {
return nil, err
}
return &user.AddApiRequestResponse{}, nil
}

38
apps/user/internal/logic/apirequestservice/getapirequestbytransactionidlogic.go Normal file
View File

@@ -0,0 +1,38 @@
package apirequestservicelogic
import (
"context"
"tianyuan-api/apps/user/internal/svc"
"tianyuan-api/apps/user/user"
"github.com/zeromicro/go-zero/core/logx"
)
type GetApiRequestByTransactionIdLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewGetApiRequestByTransactionIdLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetApiRequestByTransactionIdLogic {
return &GetApiRequestByTransactionIdLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
// 查询API请求记录ByTransactionId
func (l *GetApiRequestByTransactionIdLogic) GetApiRequestByTransactionId(in *user.GetApiRequestByTransactionIdRequest) (*user.GetApiRequestByTransactionIdResponse, error) {
apiRequest, err := l.svcCtx.ApiRequestsModel.FindOneByTransactionId(l.ctx, in.TransactionId)
if err != nil {
return nil, err
}
return &user.GetApiRequestByTransactionIdResponse{
Id: apiRequest.Id,
TransactionId: apiRequest.TransactionId,
UserId: apiRequest.UserId,
}, nil
}

31
apps/user/internal/logic/apirequestservice/getapirequestslogic.go Normal file
View File

@@ -0,0 +1,31 @@
package apirequestservicelogic
import (
"context"
"tianyuan-api/apps/user/internal/svc"
"tianyuan-api/apps/user/user"
"github.com/zeromicro/go-zero/core/logx"
)
type GetApiRequestsLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewGetApiRequestsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetApiRequestsLogic {
return &GetApiRequestsLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
// 查询API请求记录
func (l *GetApiRequestsLogic) GetApiRequests(in *user.GetApiRequestsRequest) (*user.GetApiRequestsResponse, error) {
// todo: add your logic here and delete this line
return &user.GetApiRequestsResponse{}, nil
}

5
apps/user/internal/logic/enterprise/reviewenterpriselogic.go
View File

@@ -72,7 +72,10 @@ func (l *ReviewEnterpriseLogic) ReviewEnterprise(in *user.ReviewEnterpriseReq) (
if insertEnterpriseErr != nil {
return insertEnterpriseErr
}
_, InsertWalletsTransErr := l.svcCtx.WalletsModel.InsertWalletsTrans(l.ctx, &model.Wallets{UserId: enterpriseAuth.UserId}, session)
if InsertWalletsTransErr != nil {
return InsertWalletsTransErr
}
_, createSecretErr := l.svcCtx.SecretRpc.CreateSecret(l.ctx, &secret.CreateSecretRequest{
UserId: enterpriseAuth.UserId,
})

41
apps/user/internal/logic/user/getuserinfologic.go Normal file
View File

@@ -0,0 +1,41 @@
package userlogic
import (
"context"
"errors"
"tianyuan-api/apps/user/internal/svc"
"tianyuan-api/apps/user/user"
"github.com/zeromicro/go-zero/core/logx"
)
type GetUserInfoLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewGetUserInfoLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetUserInfoLogic {
return &GetUserInfoLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
func (l *GetUserInfoLogic) GetUserInfo(in *user.UserInfoReq) (*user.GetUserInfoResp, error) {
// 查询用户信息
users, err := l.svcCtx.UserModel.FindOne(l.ctx, in.UserId)
if err != nil {
return nil, errors.New("用户不存在")
}
// 正常返回用户和企业信息
return &user.GetUserInfoResp{
Username: users.Username,
Phone: users.Phone,
Disable: users.Disable,
QuotaExceeded: users.QuotaExceeded,
}, nil
}

38
apps/user/internal/logic/walletservice/getdeductionbytransactionidlogic.go Normal file
View File

@@ -0,0 +1,38 @@
package walletservicelogic
import (
"context"
"tianyuan-api/apps/user/internal/svc"
"tianyuan-api/apps/user/user"
"github.com/zeromicro/go-zero/core/logx"
)
type GetDeductionByTransactionIdLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewGetDeductionByTransactionIdLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetDeductionByTransactionIdLogic {
return &GetDeductionByTransactionIdLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
func (l *GetDeductionByTransactionIdLogic) GetDeductionByTransactionId(in *user.GetDeductionByTransactionIdRequest) (*user.GetDeductionByTransactionIdResponse, error) {
deduction, err := l.svcCtx.DeductionsModel.FindOneByTransactionId(l.ctx, in.TransactionId)
if err != nil {
return nil, err
}
return &user.GetDeductionByTransactionIdResponse{
Id: deduction.Id,
UserId: deduction.UserId,
Amount: deduction.Amount,
TransactionId: deduction.TransactionId,
CreatedAt: deduction.CreatedAt.Format("2006-01-02 15:04:05"),
}, nil
}

31
apps/user/internal/logic/walletservice/getdeductionslogic.go Normal file
View File

@@ -0,0 +1,31 @@
package walletservicelogic
import (
"context"
"tianyuan-api/apps/user/internal/svc"
"tianyuan-api/apps/user/user"
"github.com/zeromicro/go-zero/core/logx"
)
type GetDeductionsLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewGetDeductionsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetDeductionsLogic {
return &GetDeductionsLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
// 查询扣款记录
func (l *GetDeductionsLogic) GetDeductions(in *user.GetDeductionsRequest) (*user.GetDeductionsResponse, error) {
// todo: add your logic here and delete this line
return &user.GetDeductionsResponse{}, nil
}

31
apps/user/internal/logic/walletservice/getwalletlogic.go Normal file
View File

@@ -0,0 +1,31 @@
package walletservicelogic
import (
"context"
"tianyuan-api/apps/user/internal/svc"
"tianyuan-api/apps/user/user"
"github.com/zeromicro/go-zero/core/logx"
)
type GetWalletLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewGetWalletLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetWalletLogic {
return &GetWalletLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
// 查询钱包信息
func (l *GetWalletLogic) GetWallet(in *user.GetWalletRequest) (*user.GetWalletResponse, error) {
// todo: add your logic here and delete this line
return &user.GetWalletResponse{}, nil
}

114
apps/user/internal/logic/walletservice/updatewalletlogic.go Normal file
View File

@@ -0,0 +1,114 @@
package walletservicelogic
import (
"context"
"database/sql"
"errors"
"github.com/zeromicro/go-zero/core/stores/sqlx"
"tianyuan-api/apps/sentinel/client/product"
"tianyuan-api/apps/user/internal/model"
"time"
"tianyuan-api/apps/user/internal/svc"
"tianyuan-api/apps/user/user"
"github.com/zeromicro/go-zero/core/logx"
)
type UpdateWalletLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewUpdateWalletLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdateWalletLogic {
return &UpdateWalletLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
const maxRetries = 5 // 最大重试次数
// 修改钱包余额
func (l *UpdateWalletLogic) UpdateWallet(in *user.UpdateWalletRequest) (*user.UpdateWalletResponse, error) {
if !in.Charge {
return nil, errors.New("不需要扣款")
}
// 获取产品信息
consumeProduct, getProductByCodeErr := l.svcCtx.ProductRpc.GetProductByCode(l.ctx, &product.GetRecordByCodeRequest{Code: in.ProductCode})
if getProductByCodeErr != nil {
return nil, getProductByCodeErr
}
// 检查是否已经扣款
deduction, FindOneByTransactionIdErr := l.svcCtx.DeductionsModel.FindOneByTransactionId(l.ctx, in.TransactionId)
if FindOneByTransactionIdErr != nil {
if FindOneByTransactionIdErr == sql.ErrNoRows {
} else {
// 其他错误，直接返回
return nil, FindOneByTransactionIdErr
}
} else if deduction != nil {
// 找到扣款记录，不能重复扣款
return nil, errors.New("该订单已扣款")
}
// 启动事务
err := l.svcCtx.WalletsModel.TransCtx(l.ctx, func(ctx context.Context, session sqlx.Session) error {
var err error
// 尝试多次更新余额（用于乐观锁）
for i := 0; i < maxRetries; i++ {
err = l.svcCtx.WalletsModel.UpdateBalance(session, l.ctx, in.UserId, -consumeProduct.ProductPrice)
if err == nil {
// 成功，退出循环
break
}
if errors.Is(err, model.ErrVersionMismatch) {
// 版本号不匹配，重试
time.Sleep(100 * time.Millisecond) // 重试前的延迟
continue
} else {
// 其他错误，直接返回
return err
}
}
if err != nil {
return err
}
// 更新余额成功后，插入扣款记录
_, err = l.svcCtx.DeductionsModel.InsertDeductionsTrans(ctx, &model.Deductions{
TransactionId: in.TransactionId,
UserId: in.UserId,
Amount: consumeProduct.ProductPrice,
}, session)
if err != nil {
return err
}
return nil
})
if err != nil {
return nil, err
}
// 获取最新的用户余额
wallet, err := l.svcCtx.WalletsModel.FindOneByUserId(l.ctx, in.UserId)
if err != nil {
return nil, err
}
// 如果余额小于 0，更新 QuotaExceeded 字段为 1
if wallet.Balance < 0 {
users, findOneErr := l.svcCtx.UserModel.FindOne(l.ctx, in.UserId)
if findOneErr != nil {
return nil, findOneErr
}
users.QuotaExceeded = 1
updateErr := l.svcCtx.UserModel.Update(l.ctx, users)
if updateErr != nil {
return nil, updateErr
}
}
return &user.UpdateWalletResponse{}, nil
}

27
apps/user/internal/model/apirequestsmodel.go Normal file
View File

@@ -0,0 +1,27 @@
package model
import (
"github.com/zeromicro/go-zero/core/stores/cache"
"github.com/zeromicro/go-zero/core/stores/sqlx"
)
var _ ApiRequestsModel = (*customApiRequestsModel)(nil)
type (
// ApiRequestsModel is an interface to be customized, add more methods here,
// and implement the added methods in customApiRequestsModel.
ApiRequestsModel interface {
apiRequestsModel
}
customApiRequestsModel struct {
*defaultApiRequestsModel
}
)
// NewApiRequestsModel returns a model for the database table.
func NewApiRequestsModel(conn sqlx.SqlConn, c cache.CacheConf, opts ...cache.Option) ApiRequestsModel {
return &customApiRequestsModel{
defaultApiRequestsModel: newApiRequestsModel(conn, c, opts...),
}
}

152
apps/user/internal/model/apirequestsmodel_gen.go Normal file
View File

@@ -0,0 +1,152 @@
// Code generated by goctl. DO NOT EDIT.
// versions:
// goctl version: 1.7.2
package model
import (
"context"
"database/sql"
"fmt"
"strings"
"time"
"github.com/zeromicro/go-zero/core/stores/builder"
"github.com/zeromicro/go-zero/core/stores/cache"
"github.com/zeromicro/go-zero/core/stores/sqlc"
"github.com/zeromicro/go-zero/core/stores/sqlx"
"github.com/zeromicro/go-zero/core/stringx"
)
var (
apiRequestsFieldNames = builder.RawFieldNames(&ApiRequests{})
apiRequestsRows = strings.Join(apiRequestsFieldNames, ",")
apiRequestsRowsExpectAutoSet = strings.Join(stringx.Remove(apiRequestsFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), ",")
apiRequestsRowsWithPlaceHolder = strings.Join(stringx.Remove(apiRequestsFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), "=?,") + "=?"
cacheApiRequestsIdPrefix = "cache:apiRequests:id:"
cacheApiRequestsTransactionIdPrefix = "cache:apiRequests:transactionId:"
)
type (
apiRequestsModel interface {
Insert(ctx context.Context, data *ApiRequests) (sql.Result, error)
FindOne(ctx context.Context, id int64) (*ApiRequests, error)
FindOneByTransactionId(ctx context.Context, transactionId string) (*ApiRequests, error)
Update(ctx context.Context, data *ApiRequests) error
Delete(ctx context.Context, id int64) error
}
defaultApiRequestsModel struct {
sqlc.CachedConn
table string
}
ApiRequests struct {
Id int64 `db:"id"` // 请求记录ID
TransactionId string `db:"transaction_id"` // 交易ID
UserId int64 `db:"user_id"` // 用户ID
ProductCode string `db:"product_code"` // 产品编码
Status string `db:"status"` // 请求状态：success=成功，failed=失败
Charges int64 `db:"charges"` // 是否需要付费
Remark sql.NullString `db:"remark"` // 备注
Timestamp time.Time `db:"timestamp"` // 请求时间
}
)
func newApiRequestsModel(conn sqlx.SqlConn, c cache.CacheConf, opts ...cache.Option) *defaultApiRequestsModel {
return &defaultApiRequestsModel{
CachedConn: sqlc.NewConn(conn, c, opts...),
table: "`api_requests`",
}
}
func (m *defaultApiRequestsModel) Delete(ctx context.Context, id int64) error {
data, err := m.FindOne(ctx, id)
if err != nil {
return err
}
apiRequestsIdKey := fmt.Sprintf("%s%v", cacheApiRequestsIdPrefix, id)
apiRequestsTransactionIdKey := fmt.Sprintf("%s%v", cacheApiRequestsTransactionIdPrefix, data.TransactionId)
_, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) {
query := fmt.Sprintf("delete from %s where `id` = ?", m.table)
return conn.ExecCtx(ctx, query, id)
}, apiRequestsIdKey, apiRequestsTransactionIdKey)
return err
}
func (m *defaultApiRequestsModel) FindOne(ctx context.Context, id int64) (*ApiRequests, error) {
apiRequestsIdKey := fmt.Sprintf("%s%v", cacheApiRequestsIdPrefix, id)
var resp ApiRequests
err := m.QueryRowCtx(ctx, &resp, apiRequestsIdKey, func(ctx context.Context, conn sqlx.SqlConn, v any) error {
query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", apiRequestsRows, m.table)
return conn.QueryRowCtx(ctx, v, query, id)
})
switch err {
case nil:
return &resp, nil
case sqlc.ErrNotFound:
return nil, ErrNotFound
default:
return nil, err
}
}
func (m *defaultApiRequestsModel) FindOneByTransactionId(ctx context.Context, transactionId string) (*ApiRequests, error) {
apiRequestsTransactionIdKey := fmt.Sprintf("%s%v", cacheApiRequestsTransactionIdPrefix, transactionId)
var resp ApiRequests
err := m.QueryRowIndexCtx(ctx, &resp, apiRequestsTransactionIdKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v any) (i any, e error) {
query := fmt.Sprintf("select %s from %s where `transaction_id` = ? limit 1", apiRequestsRows, m.table)
if err := conn.QueryRowCtx(ctx, &resp, query, transactionId); err != nil {
return nil, err
}
return resp.Id, nil
}, m.queryPrimary)
switch err {
case nil:
return &resp, nil
case sqlc.ErrNotFound:
return nil, ErrNotFound
default:
return nil, err
}
}
func (m *defaultApiRequestsModel) Insert(ctx context.Context, data *ApiRequests) (sql.Result, error) {
apiRequestsIdKey := fmt.Sprintf("%s%v", cacheApiRequestsIdPrefix, data.Id)
apiRequestsTransactionIdKey := fmt.Sprintf("%s%v", cacheApiRequestsTransactionIdPrefix, data.TransactionId)
ret, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) {
query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?)", m.table, apiRequestsRowsExpectAutoSet)
return conn.ExecCtx(ctx, query, data.TransactionId, data.UserId, data.ProductCode, data.Status, data.Charges, data.Remark, data.Timestamp)
}, apiRequestsIdKey, apiRequestsTransactionIdKey)
return ret, err
}
func (m *defaultApiRequestsModel) Update(ctx context.Context, newData *ApiRequests) error {
data, err := m.FindOne(ctx, newData.Id)
if err != nil {
return err
}
apiRequestsIdKey := fmt.Sprintf("%s%v", cacheApiRequestsIdPrefix, data.Id)
apiRequestsTransactionIdKey := fmt.Sprintf("%s%v", cacheApiRequestsTransactionIdPrefix, data.TransactionId)
_, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) {
query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, apiRequestsRowsWithPlaceHolder)
return conn.ExecCtx(ctx, query, newData.TransactionId, newData.UserId, newData.ProductCode, newData.Status, newData.Charges, newData.Remark, newData.Timestamp, newData.Id)
}, apiRequestsIdKey, apiRequestsTransactionIdKey)
return err
}
func (m *defaultApiRequestsModel) formatPrimary(primary any) string {
return fmt.Sprintf("%s%v", cacheApiRequestsIdPrefix, primary)
}
func (m *defaultApiRequestsModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary any) error {
query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", apiRequestsRows, m.table)
return conn.QueryRowCtx(ctx, v, query, primary)
}
func (m *defaultApiRequestsModel) tableName() string {
return m.table
}

52
apps/user/internal/model/deductionsmodel.go Normal file
View File

@@ -0,0 +1,52 @@
package model
import (
"context"
"database/sql"
"fmt"
"github.com/zeromicro/go-zero/core/stores/cache"
"github.com/zeromicro/go-zero/core/stores/sqlx"
)
var _ DeductionsModel = (*customDeductionsModel)(nil)
type (
// DeductionsModel is an interface to be customized, add more methods here,
// and implement the added methods in customDeductionsModel.
DeductionsModel interface {
deductionsModel
InsertDeductionsTrans(ctx context.Context, deductions *Deductions, session sqlx.Session) (sql.Result, error)
}
customDeductionsModel struct {
*defaultDeductionsModel
}
)
// NewDeductionsModel returns a model for the database table.
func NewDeductionsModel(conn sqlx.SqlConn, c cache.CacheConf, opts ...cache.Option) DeductionsModel {
return &customDeductionsModel{
defaultDeductionsModel: newDeductionsModel(conn, c, opts...),
}
}
func (m *customDeductionsModel) InsertDeductionsTrans(ctx context.Context, deductions *Deductions, session sqlx.Session) (sql.Result, error) {
query := fmt.Sprintf("INSERT INTO %s (%s) VALUES (?, ?, ?)", m.table, deductionsRowsExpectAutoSet)
ret, err := session.ExecCtx(ctx, query, deductions.UserId, deductions.Amount, deductions.TransactionId)
if err != nil {
return nil, err
}
deductionsTransactionIdKey := fmt.Sprintf("%s%v", cacheDeductionsTransactionIdPrefix, deductions.TransactionId)
// 2. 更新缓存，保证所有缓存操作成功
cacheKeys := []string{deductionsTransactionIdKey}
cacheErrors := make([]error, len(cacheKeys))
cacheErrors[0] = m.DelCacheCtx(ctx, deductionsTransactionIdKey)
// 3. 检查缓存操作是否全部成功
for _, cacheErr := range cacheErrors {
if cacheErr != nil {
return nil, cacheErr // 返回第一个缓存更新失败的错误
}
}
return ret, err
}

149
apps/user/internal/model/deductionsmodel_gen.go Normal file
View File

@@ -0,0 +1,149 @@
// Code generated by goctl. DO NOT EDIT.
// versions:
// goctl version: 1.7.2
package model
import (
"context"
"database/sql"
"fmt"
"strings"
"time"
"github.com/zeromicro/go-zero/core/stores/builder"
"github.com/zeromicro/go-zero/core/stores/cache"
"github.com/zeromicro/go-zero/core/stores/sqlc"
"github.com/zeromicro/go-zero/core/stores/sqlx"
"github.com/zeromicro/go-zero/core/stringx"
)
var (
deductionsFieldNames = builder.RawFieldNames(&Deductions{})
deductionsRows = strings.Join(deductionsFieldNames, ",")
deductionsRowsExpectAutoSet = strings.Join(stringx.Remove(deductionsFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), ",")
deductionsRowsWithPlaceHolder = strings.Join(stringx.Remove(deductionsFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), "=?,") + "=?"
cacheDeductionsIdPrefix = "cache:deductions:id:"
cacheDeductionsTransactionIdPrefix = "cache:deductions:transactionId:"
)
type (
deductionsModel interface {
Insert(ctx context.Context, data *Deductions) (sql.Result, error)
FindOne(ctx context.Context, id int64) (*Deductions, error)
FindOneByTransactionId(ctx context.Context, transactionId string) (*Deductions, error)
Update(ctx context.Context, data *Deductions) error
Delete(ctx context.Context, id int64) error
}
defaultDeductionsModel struct {
sqlc.CachedConn
table string
}
Deductions struct {
Id int64 `db:"id"` // 扣款记录ID
UserId int64 `db:"user_id"` // 用户ID
Amount float64 `db:"amount"` // 扣款金额
TransactionId string `db:"transaction_id"` // 交易流水号
CreatedAt time.Time `db:"created_at"` // 扣款时间
}
)
func newDeductionsModel(conn sqlx.SqlConn, c cache.CacheConf, opts ...cache.Option) *defaultDeductionsModel {
return &defaultDeductionsModel{
CachedConn: sqlc.NewConn(conn, c, opts...),
table: "`deductions`",
}
}
func (m *defaultDeductionsModel) Delete(ctx context.Context, id int64) error {
data, err := m.FindOne(ctx, id)
if err != nil {
return err
}
deductionsIdKey := fmt.Sprintf("%s%v", cacheDeductionsIdPrefix, id)
deductionsTransactionIdKey := fmt.Sprintf("%s%v", cacheDeductionsTransactionIdPrefix, data.TransactionId)
_, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) {
query := fmt.Sprintf("delete from %s where `id` = ?", m.table)
return conn.ExecCtx(ctx, query, id)
}, deductionsIdKey, deductionsTransactionIdKey)
return err
}
func (m *defaultDeductionsModel) FindOne(ctx context.Context, id int64) (*Deductions, error) {
deductionsIdKey := fmt.Sprintf("%s%v", cacheDeductionsIdPrefix, id)
var resp Deductions
err := m.QueryRowCtx(ctx, &resp, deductionsIdKey, func(ctx context.Context, conn sqlx.SqlConn, v any) error {
query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", deductionsRows, m.table)
return conn.QueryRowCtx(ctx, v, query, id)
})
switch err {
case nil:
return &resp, nil
case sqlc.ErrNotFound:
return nil, ErrNotFound
default:
return nil, err
}
}
func (m *defaultDeductionsModel) FindOneByTransactionId(ctx context.Context, transactionId string) (*Deductions, error) {
deductionsTransactionIdKey := fmt.Sprintf("%s%v", cacheDeductionsTransactionIdPrefix, transactionId)
var resp Deductions
err := m.QueryRowIndexCtx(ctx, &resp, deductionsTransactionIdKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v any) (i any, e error) {
query := fmt.Sprintf("select %s from %s where `transaction_id` = ? limit 1", deductionsRows, m.table)
if err := conn.QueryRowCtx(ctx, &resp, query, transactionId); err != nil {
return nil, err
}
return resp.Id, nil
}, m.queryPrimary)
switch err {
case nil:
return &resp, nil
case sqlc.ErrNotFound:
return nil, ErrNotFound
default:
return nil, err
}
}
func (m *defaultDeductionsModel) Insert(ctx context.Context, data *Deductions) (sql.Result, error) {
deductionsIdKey := fmt.Sprintf("%s%v", cacheDeductionsIdPrefix, data.Id)
deductionsTransactionIdKey := fmt.Sprintf("%s%v", cacheDeductionsTransactionIdPrefix, data.TransactionId)
ret, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) {
query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?)", m.table, deductionsRowsExpectAutoSet)
return conn.ExecCtx(ctx, query, data.UserId, data.Amount, data.TransactionId)
}, deductionsIdKey, deductionsTransactionIdKey)
return ret, err
}
func (m *defaultDeductionsModel) Update(ctx context.Context, newData *Deductions) error {
data, err := m.FindOne(ctx, newData.Id)
if err != nil {
return err
}
deductionsIdKey := fmt.Sprintf("%s%v", cacheDeductionsIdPrefix, data.Id)
deductionsTransactionIdKey := fmt.Sprintf("%s%v", cacheDeductionsTransactionIdPrefix, data.TransactionId)
_, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) {
query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, deductionsRowsWithPlaceHolder)
return conn.ExecCtx(ctx, query, newData.UserId, newData.Amount, newData.TransactionId, newData.Id)
}, deductionsIdKey, deductionsTransactionIdKey)
return err
}
func (m *defaultDeductionsModel) formatPrimary(primary any) string {
return fmt.Sprintf("%s%v", cacheDeductionsIdPrefix, primary)
}
func (m *defaultDeductionsModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary any) error {
query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", deductionsRows, m.table)
return conn.QueryRowCtx(ctx, v, query, primary)
}
func (m *defaultDeductionsModel) tableName() string {
return m.table
}

23
apps/user/internal/model/usersmodel_gen.go
View File

@@ -45,13 +45,16 @@ type (
}
Users struct {
Id int64 `db:"id"` // 用户ID
Username string `db:"username"` // 用户名
Password string `db:"password"` // 用户密码
Phone string `db:"phone"` // 用户手机号
AuthStatus string `db:"auth_status"` // 认证状态：unverified=未提交，pending=待审核，approved=审核通过，rejected=审核拒绝
CreatedAt time.Time `db:"created_at"` // 用户创建时间
UpdatedAt time.Time `db:"updated_at"` // 用户更新时间
Id int64 `db:"id"` // 用户ID
Username string `db:"username"` // 用户名
Password string `db:"password"` // 用户密码
Phone string `db:"phone"` // 用户手机号
AuthStatus string `db:"auth_status"` // 认证状态：unverified=未提交，pending=待审核，approved=审核通过，rejected=审核拒绝
Disable int64 `db:"disable"` // 是否禁用，0=未禁用，1=禁用
Internal int64 `db:"internal"` // 是否内部人员，0=否，1=是
QuotaExceeded int64 `db:"quota_exceeded"` // 是否额度用完，0=否，1=是
CreatedAt time.Time `db:"created_at"` // 用户创建时间
UpdatedAt time.Time `db:"updated_at"` // 用户更新时间
}
)
@@ -140,8 +143,8 @@ func (m *defaultUsersModel) Insert(ctx context.Context, data *Users) (sql.Result
usersPhoneKey := fmt.Sprintf("%s%v", cacheUsersPhonePrefix, data.Phone)
usersUsernameKey := fmt.Sprintf("%s%v", cacheUsersUsernamePrefix, data.Username)
ret, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) {
query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?)", m.table, usersRowsExpectAutoSet)
return conn.ExecCtx(ctx, query, data.Username, data.Password, data.Phone, data.AuthStatus)
query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?)", m.table, usersRowsExpectAutoSet)
return conn.ExecCtx(ctx, query, data.Username, data.Password, data.Phone, data.AuthStatus, data.Disable, data.Internal, data.QuotaExceeded)
}, usersIdKey, usersPhoneKey, usersUsernameKey)
return ret, err
}
@@ -157,7 +160,7 @@ func (m *defaultUsersModel) Update(ctx context.Context, newData *Users) error {
usersUsernameKey := fmt.Sprintf("%s%v", cacheUsersUsernamePrefix, data.Username)
_, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) {
query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, usersRowsWithPlaceHolder)
return conn.ExecCtx(ctx, query, newData.Username, newData.Password, newData.Phone, newData.AuthStatus, newData.Id)
return conn.ExecCtx(ctx, query, newData.Username, newData.Password, newData.Phone, newData.AuthStatus, newData.Disable, newData.Internal, newData.QuotaExceeded, newData.Id)
}, usersIdKey, usersPhoneKey, usersUsernameKey)
return err
}

89
apps/user/internal/model/walletsmodel.go Normal file
View File

@@ -0,0 +1,89 @@
package model
import (
"context"
"database/sql"
"errors"
"fmt"
"github.com/zeromicro/go-zero/core/stores/cache"
"github.com/zeromicro/go-zero/core/stores/sqlx"
)
var _ WalletsModel = (*customWalletsModel)(nil)
type (
// WalletsModel is an interface to be customized, add more methods here,
// and implement the added methods in customWalletsModel.
WalletsModel interface {
walletsModel
InsertWalletsTrans(ctx context.Context, wallets *Wallets, session sqlx.Session) (sql.Result, error)
UpdateBalance(session sqlx.Session, ctx context.Context, userId int64, amount float64) error
TransCtx(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error
}
customWalletsModel struct {
*defaultWalletsModel
}
)
var ErrBalanceNotEnough = errors.New("余额不足")
var ErrVersionMismatch = errors.New("版本号不匹配，请重试")
// NewWalletsModel returns a model for the database table.
func NewWalletsModel(conn sqlx.SqlConn, c cache.CacheConf, opts ...cache.Option) WalletsModel {
return &customWalletsModel{
defaultWalletsModel: newWalletsModel(conn, c, opts...),
}
}
func (m *customWalletsModel) TransCtx(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error {
// 使用带 ctx 的事务处理
err := m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error {
return fn(ctx, session)
})
return err
}
// 更新余额的方法
func (m *customWalletsModel) UpdateBalance(session sqlx.Session, ctx context.Context, userId int64, amount float64) error {
wallet, err := m.FindOneByUserId(ctx, userId)
if err != nil {
return err
}
// 检查余额是否足够
if wallet.Balance+amount < 0 {
return ErrBalanceNotEnough
}
// 使用乐观锁更新余额
result, err := session.Exec("UPDATE wallets SET balance = balance + ?, version = version + 1 WHERE user_id = ? AND version = ?", amount, userId, wallet.Version)
if err != nil {
return err
}
// 检查影响的行数，确保更新成功
rowsAffected, err := result.RowsAffected()
if err != nil {
return err
}
if rowsAffected == 0 {
return ErrVersionMismatch
}
walletsUserIdKey := fmt.Sprintf("%s%v", cacheWalletsUserIdPrefix, userId)
cacheErrors := m.DelCacheCtx(ctx, walletsUserIdKey)
if cacheErrors != nil {
return cacheErrors
}
return nil
}
func (m *customWalletsModel) InsertWalletsTrans(ctx context.Context, wallets *Wallets, session sqlx.Session) (sql.Result, error) {
query := fmt.Sprintf("INSERT INTO %s (%s) VALUES (?, ?, ?)", m.table, walletsRowsExpectAutoSet)
ret, err := session.ExecCtx(ctx, query, wallets.UserId, wallets.Balance, wallets.Version)
if err != nil {
return nil, err
}
return ret, err
}

150
apps/user/internal/model/walletsmodel_gen.go Normal file
View File

@@ -0,0 +1,150 @@
// Code generated by goctl. DO NOT EDIT.
// versions:
// goctl version: 1.7.2
package model
import (
"context"
"database/sql"
"fmt"
"strings"
"time"
"github.com/zeromicro/go-zero/core/stores/builder"
"github.com/zeromicro/go-zero/core/stores/cache"
"github.com/zeromicro/go-zero/core/stores/sqlc"
"github.com/zeromicro/go-zero/core/stores/sqlx"
"github.com/zeromicro/go-zero/core/stringx"
)
var (
walletsFieldNames = builder.RawFieldNames(&Wallets{})
walletsRows = strings.Join(walletsFieldNames, ",")
walletsRowsExpectAutoSet = strings.Join(stringx.Remove(walletsFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), ",")
walletsRowsWithPlaceHolder = strings.Join(stringx.Remove(walletsFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), "=?,") + "=?"
cacheWalletsIdPrefix = "cache:wallets:id:"
cacheWalletsUserIdPrefix = "cache:wallets:userId:"
)
type (
walletsModel interface {
Insert(ctx context.Context, data *Wallets) (sql.Result, error)
FindOne(ctx context.Context, id int64) (*Wallets, error)
FindOneByUserId(ctx context.Context, userId int64) (*Wallets, error)
Update(ctx context.Context, data *Wallets) error
Delete(ctx context.Context, id int64) error
}
defaultWalletsModel struct {
sqlc.CachedConn
table string
}
Wallets struct {
Id int64 `db:"id"` // 钱包ID
UserId int64 `db:"user_id"` // 用户ID
Balance float64 `db:"balance"` // 钱包余额
Version int64 `db:"version"` // 乐观锁版本号
CreatedAt time.Time `db:"created_at"` // 创建时间
UpdatedAt time.Time `db:"updated_at"` // 更新时间
}
)
func newWalletsModel(conn sqlx.SqlConn, c cache.CacheConf, opts ...cache.Option) *defaultWalletsModel {
return &defaultWalletsModel{
CachedConn: sqlc.NewConn(conn, c, opts...),
table: "`wallets`",
}
}
func (m *defaultWalletsModel) Delete(ctx context.Context, id int64) error {
data, err := m.FindOne(ctx, id)
if err != nil {
return err
}
walletsIdKey := fmt.Sprintf("%s%v", cacheWalletsIdPrefix, id)
walletsUserIdKey := fmt.Sprintf("%s%v", cacheWalletsUserIdPrefix, data.UserId)
_, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) {
query := fmt.Sprintf("delete from %s where `id` = ?", m.table)
return conn.ExecCtx(ctx, query, id)
}, walletsIdKey, walletsUserIdKey)
return err
}
func (m *defaultWalletsModel) FindOne(ctx context.Context, id int64) (*Wallets, error) {
walletsIdKey := fmt.Sprintf("%s%v", cacheWalletsIdPrefix, id)
var resp Wallets
err := m.QueryRowCtx(ctx, &resp, walletsIdKey, func(ctx context.Context, conn sqlx.SqlConn, v any) error {
query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", walletsRows, m.table)
return conn.QueryRowCtx(ctx, v, query, id)
})
switch err {
case nil:
return &resp, nil
case sqlc.ErrNotFound:
return nil, ErrNotFound
default:
return nil, err
}
}
func (m *defaultWalletsModel) FindOneByUserId(ctx context.Context, userId int64) (*Wallets, error) {
walletsUserIdKey := fmt.Sprintf("%s%v", cacheWalletsUserIdPrefix, userId)
var resp Wallets
err := m.QueryRowIndexCtx(ctx, &resp, walletsUserIdKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v any) (i any, e error) {
query := fmt.Sprintf("select %s from %s where `user_id` = ? limit 1", walletsRows, m.table)
if err := conn.QueryRowCtx(ctx, &resp, query, userId); err != nil {
return nil, err
}
return resp.Id, nil
}, m.queryPrimary)
switch err {
case nil:
return &resp, nil
case sqlc.ErrNotFound:
return nil, ErrNotFound
default:
return nil, err
}
}
func (m *defaultWalletsModel) Insert(ctx context.Context, data *Wallets) (sql.Result, error) {
walletsIdKey := fmt.Sprintf("%s%v", cacheWalletsIdPrefix, data.Id)
walletsUserIdKey := fmt.Sprintf("%s%v", cacheWalletsUserIdPrefix, data.UserId)
ret, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) {
query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?)", m.table, walletsRowsExpectAutoSet)
return conn.ExecCtx(ctx, query, data.UserId, data.Balance, data.Version)
}, walletsIdKey, walletsUserIdKey)
return ret, err
}
func (m *defaultWalletsModel) Update(ctx context.Context, newData *Wallets) error {
data, err := m.FindOne(ctx, newData.Id)
if err != nil {
return err
}
walletsIdKey := fmt.Sprintf("%s%v", cacheWalletsIdPrefix, data.Id)
walletsUserIdKey := fmt.Sprintf("%s%v", cacheWalletsUserIdPrefix, data.UserId)
_, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) {
query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, walletsRowsWithPlaceHolder)
return conn.ExecCtx(ctx, query, newData.UserId, newData.Balance, newData.Version, newData.Id)
}, walletsIdKey, walletsUserIdKey)
return err
}
func (m *defaultWalletsModel) formatPrimary(primary any) string {
return fmt.Sprintf("%s%v", cacheWalletsIdPrefix, primary)
}
func (m *defaultWalletsModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary any) error {
query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", walletsRows, m.table)
return conn.QueryRowCtx(ctx, v, query, primary)
}
func (m *defaultWalletsModel) tableName() string {
return m.table
}

42
apps/user/internal/server/apirequestservice/apirequestserviceserver.go Normal file
View File

@@ -0,0 +1,42 @@
// Code generated by goctl. DO NOT EDIT.
// goctl 1.7.2
// Source: user.proto
package server
import (
"context"
"tianyuan-api/apps/user/internal/logic/apirequestservice"
"tianyuan-api/apps/user/internal/svc"
"tianyuan-api/apps/user/user"
)
type ApiRequestServiceServer struct {
svcCtx *svc.ServiceContext
user.UnimplementedApiRequestServiceServer
}
func NewApiRequestServiceServer(svcCtx *svc.ServiceContext) *ApiRequestServiceServer {
return &ApiRequestServiceServer{
svcCtx: svcCtx,
}
}
// 添加API请求记录
func (s *ApiRequestServiceServer) AddApiRequest(ctx context.Context, in *user.AddApiRequestRequest) (*user.AddApiRequestResponse, error) {
l := apirequestservicelogic.NewAddApiRequestLogic(ctx, s.svcCtx)
return l.AddApiRequest(in)
}
// 查询API请求记录
func (s *ApiRequestServiceServer) GetApiRequests(ctx context.Context, in *user.GetApiRequestsRequest) (*user.GetApiRequestsResponse, error) {
l := apirequestservicelogic.NewGetApiRequestsLogic(ctx, s.svcCtx)
return l.GetApiRequests(in)
}
// 查询API请求记录ByTransactionId
func (s *ApiRequestServiceServer) GetApiRequestByTransactionId(ctx context.Context, in *user.GetApiRequestByTransactionIdRequest) (*user.GetApiRequestByTransactionIdResponse, error) {
l := apirequestservicelogic.NewGetApiRequestByTransactionIdLogic(ctx, s.svcCtx)
return l.GetApiRequestByTransactionId(in)
}

5
apps/user/internal/server/user/userserver.go
View File

@@ -29,6 +29,11 @@ func (s *UserServer) UserInfo(ctx context.Context, in *user.UserInfoReq) (*user.
return l.UserInfo(in)
}
func (s *UserServer) GetUserInfo(ctx context.Context, in *user.UserInfoReq) (*user.GetUserInfoResp, error) {
l := userlogic.NewGetUserInfoLogic(ctx, s.svcCtx)
return l.GetUserInfo(in)
}
func (s *UserServer) GetEnterpriseAuthStatus(ctx context.Context, in *user.GetEnterpriseAuthStatusReq) (*user.GetEnterpriseAuthStatusResp, error) {
l := userlogic.NewGetEnterpriseAuthStatusLogic(ctx, s.svcCtx)
return l.GetEnterpriseAuthStatus(in)

47
apps/user/internal/server/walletservice/walletserviceserver.go Normal file
View File

@@ -0,0 +1,47 @@
// Code generated by goctl. DO NOT EDIT.
// goctl 1.7.2
// Source: user.proto
package server
import (
"context"
"tianyuan-api/apps/user/internal/logic/walletservice"
"tianyuan-api/apps/user/internal/svc"
"tianyuan-api/apps/user/user"
)
type WalletServiceServer struct {
svcCtx *svc.ServiceContext
user.UnimplementedWalletServiceServer
}
func NewWalletServiceServer(svcCtx *svc.ServiceContext) *WalletServiceServer {
return &WalletServiceServer{
svcCtx: svcCtx,
}
}
// 修改钱包余额
func (s *WalletServiceServer) UpdateWallet(ctx context.Context, in *user.UpdateWalletRequest) (*user.UpdateWalletResponse, error) {
l := walletservicelogic.NewUpdateWalletLogic(ctx, s.svcCtx)
return l.UpdateWallet(in)
}
// 查询钱包信息
func (s *WalletServiceServer) GetWallet(ctx context.Context, in *user.GetWalletRequest) (*user.GetWalletResponse, error) {
l := walletservicelogic.NewGetWalletLogic(ctx, s.svcCtx)
return l.GetWallet(in)
}
// 查询扣款记录
func (s *WalletServiceServer) GetDeductions(ctx context.Context, in *user.GetDeductionsRequest) (*user.GetDeductionsResponse, error) {
l := walletservicelogic.NewGetDeductionsLogic(ctx, s.svcCtx)
return l.GetDeductions(in)
}
func (s *WalletServiceServer) GetDeductionByTransactionId(ctx context.Context, in *user.GetDeductionByTransactionIdRequest) (*user.GetDeductionByTransactionIdResponse, error) {
l := walletservicelogic.NewGetDeductionByTransactionIdLogic(ctx, s.svcCtx)
return l.GetDeductionByTransactionId(in)
}

8
apps/user/internal/svc/servicecontext.go
View File

@@ -15,7 +15,11 @@ type ServiceContext struct {
UserModel model.UsersModel // 用户表的模型
EnterpriseModel model.EnterpriseInfoModel
EnterpriseAuthModel model.EnterpriseAuthModel
WalletsModel model.WalletsModel
DeductionsModel model.DeductionsModel
ApiRequestsModel model.ApiRequestsModel
SecretRpc sentinel.SecretClient
ProductRpc sentinel.ProductClient
}
func NewServiceContext(c config.Config) *ServiceContext {
@@ -34,6 +38,10 @@ func NewServiceContext(c config.Config) *ServiceContext {
UserModel: model.NewUsersModel(db, c.CacheRedis), // 注入UserModel
EnterpriseModel: model.NewEnterpriseInfoModel(db, c.CacheRedis),
EnterpriseAuthModel: model.NewEnterpriseAuthModel(db, c.CacheRedis),
WalletsModel: model.NewWalletsModel(db, c.CacheRedis),
DeductionsModel: model.NewDeductionsModel(db, c.CacheRedis),
ApiRequestsModel: model.NewApiRequestsModel(db, c.CacheRedis),
SecretRpc: sentinel.NewSecretClient(zrpc.MustNewClient(c.SentinelRpc).Conn()),
ProductRpc: sentinel.NewProductClient(zrpc.MustNewClient(c.SentinelRpc).Conn()),
}
}