Integrate Qiniu upload service and update configuration

- Added Qiniu configuration options to .env.example and config.go for file uploads.
- Refactored main.go to include new Qiniu service and upload handler.
- Updated route registration to accommodate the new upload handler.
- Enhanced documentation to include references for Qiniu upload functionality.
- Removed legacy authentication handler and services to streamline the codebase.
This commit is contained in:
nepiedg
2025-12-31 03:18:03 +00:00
parent 2884f54666
commit cd7ae5ac56
14 changed files with 320 additions and 9 deletions
@@ -0,0 +1,149 @@
package service
import (
"context"
"errors"
"fmt"
"sync"
"wx_service/internal/model"
"gorm.io/gorm"
)
var (
ErrCodeRequired = errors.New("code is required")
ErrMiniProgramRequired = errors.New("mini program id is required")
ErrMiniProgramNotFound = errors.New("mini program not found")
)
type AuthService struct {
db *gorm.DB
miniProgramSvc *MiniProgramService
wechatClientCache map[uint]*WeChatClient
cacheMu sync.RWMutex
}
type LoginRequest struct {
MiniProgramID uint
Code string
NickName string
AvatarURL string
Gender *int
Phone string
}
type LoginResult struct {
User *model.User
SessionKey string
MiniProgram *model.MiniProgram
}
func NewAuthService(db *gorm.DB, miniProgramSvc *MiniProgramService) *AuthService {
return &AuthService{
db: db,
miniProgramSvc: miniProgramSvc,
wechatClientCache: make(map[uint]*WeChatClient),
}
}
func (s *AuthService) LoginWithCode(ctx context.Context, req LoginRequest) (*LoginResult, error) {
if req.MiniProgramID == 0 {
return nil, ErrMiniProgramRequired
}
if req.Code == "" {
return nil, ErrCodeRequired
}
miniProgram, err := s.miniProgramSvc.GetByID(ctx, req.MiniProgramID)
if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return nil, ErrMiniProgramNotFound
}
return nil, fmt.Errorf("load mini program: %w", err)
}
client := s.getWeChatClient(miniProgram)
session, err := client.Code2Session(ctx, req.Code)
if err != nil {
return nil, err
}
if session.OpenID == "" {
return nil, fmt.Errorf("wechat response missing openid")
}
tx := s.db.WithContext(ctx)
var user model.User
err = tx.Where("mini_program_id = ? AND open_id = ?", miniProgram.ID, session.OpenID).First(&user).Error
if errors.Is(err, gorm.ErrRecordNotFound) {
user = model.User{
MiniProgramID: miniProgram.ID,
OpenID: session.OpenID,
UnionID: session.UnionID,
NickName: req.NickName,
AvatarURL: req.AvatarURL,
Phone: req.Phone,
SessionKey: session.SessionKey,
}
if req.Gender != nil {
user.Gender = *req.Gender
}
if err := tx.Create(&user).Error; err != nil {
return nil, fmt.Errorf("create user: %w", err)
}
} else if err != nil {
return nil, fmt.Errorf("query user: %w", err)
} else {
updates := map[string]interface{}{
"session_key": session.SessionKey,
}
if session.UnionID != "" && session.UnionID != user.UnionID {
updates["union_id"] = session.UnionID
user.UnionID = session.UnionID
}
if req.NickName != "" && req.NickName != user.NickName {
updates["nick_name"] = req.NickName
user.NickName = req.NickName
}
if req.AvatarURL != "" && req.AvatarURL != user.AvatarURL {
updates["avatar_url"] = req.AvatarURL
user.AvatarURL = req.AvatarURL
}
if req.Phone != "" && req.Phone != user.Phone {
updates["phone"] = req.Phone
user.Phone = req.Phone
}
if req.Gender != nil && user.Gender != *req.Gender {
updates["gender"] = *req.Gender
user.Gender = *req.Gender
}
if len(updates) > 0 {
if err := tx.Model(&user).Updates(updates).Error; err != nil {
return nil, fmt.Errorf("update user: %w", err)
}
}
user.SessionKey = session.SessionKey
}
result := &LoginResult{
User: &user,
SessionKey: session.SessionKey,
MiniProgram: miniProgram,
}
return result, nil
}
func (s *AuthService) getWeChatClient(mp *model.MiniProgram) *WeChatClient {
s.cacheMu.RLock()
client, ok := s.wechatClientCache[mp.ID]
s.cacheMu.RUnlock()
if ok {
return client
}
newClient := NewWeChatClient(mp.AppID, mp.AppSecret, nil)
s.cacheMu.Lock()
s.wechatClientCache[mp.ID] = newClient
s.cacheMu.Unlock()
return newClient
}