yml2213
b377f8350b
完成模块： - auth: 3 个 Repository 方法 + Service + Handler + Middleware - wallet: 已有 context 支持，修复依赖调用 - payment: 已有 context 支持，修复 wallet 调用 - adminaudit: 1 个方法 - notification: 2 个方法 - realname: 2 个方法 - systemconfig: 4 个方法 - adminauth: 7 个方法 - adminuser: 6 个方法 所有数据库调用已改为 r.db.WithContext(ctx)，完整传递 context 链路。 待完成模块: order, listing, chat 等 12 个模块（约 157 个方法）
76 lines
2.1 KiB
Go
76 lines
2.1 KiB
Go
package realname
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"hfb_sys/backend/internal/model"
|
|
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/clause"
|
|
)
|
|
|
|
type Repository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewRepository(db *gorm.DB) *Repository {
|
|
return &Repository{db: db}
|
|
}
|
|
|
|
func (r *Repository) FindByUserID(ctx context.Context, userID uint64) (*model.UserRealname, error) {
|
|
var record model.UserRealname
|
|
if err := r.db.WithContext(ctx).Where("user_id = ?", userID).First(&record).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &record, nil
|
|
}
|
|
|
|
func (r *Repository) SaveResult(ctx context.Context, userID uint64, provider string, result ProviderResult) (*model.UserRealname, error) {
|
|
record := model.UserRealname{
|
|
UserID: userID,
|
|
Provider: provider,
|
|
ProviderOrderNo: result.ProviderOrderNo,
|
|
Status: result.Status,
|
|
MaskedName: result.MaskedName,
|
|
EncryptedName: result.EncryptedName,
|
|
MaskedIDNo: result.MaskedIDNo,
|
|
EncryptedIDNo: result.EncryptedIDNo,
|
|
VerifiedAt: result.VerifiedAt,
|
|
FailReason: result.FailReason,
|
|
}
|
|
|
|
err := r.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
|
|
if err := tx.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "user_id"}},
|
|
DoUpdates: clause.Assignments(map[string]any{
|
|
"provider": record.Provider,
|
|
"provider_order_no": record.ProviderOrderNo,
|
|
"status": record.Status,
|
|
"masked_name": record.MaskedName,
|
|
"encrypted_name": record.EncryptedName,
|
|
"masked_id_no": record.MaskedIDNo,
|
|
"encrypted_id_no": record.EncryptedIDNo,
|
|
"verified_at": record.VerifiedAt,
|
|
"fail_reason": record.FailReason,
|
|
"updated_at": time.Now(),
|
|
}),
|
|
}).Create(&record).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
userStatus := StatusRejected
|
|
if result.Status == StatusVerified {
|
|
userStatus = "verified"
|
|
}
|
|
if result.Status == StatusPending {
|
|
userStatus = "pending"
|
|
}
|
|
return tx.Model(&model.User{}).Where("id = ?", userID).Update("realname_status", userStatus).Error
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return r.FindByUserID(ctx, userID)
|
|
}
|