init: initial commit
This commit is contained in:
@@ -0,0 +1,11 @@
|
||||
package system
|
||||
|
||||
type ServiceGroup struct {
|
||||
JwtService
|
||||
UserService
|
||||
ClientService
|
||||
RoleService
|
||||
MenuService
|
||||
OperationRecordService
|
||||
OssService
|
||||
}
|
||||
@@ -0,0 +1,135 @@
|
||||
package system
|
||||
|
||||
import (
|
||||
"crypto/md5"
|
||||
"errors"
|
||||
"fmt"
|
||||
"image"
|
||||
_ "image/jpeg"
|
||||
_ "image/png"
|
||||
"io"
|
||||
"mime/multipart"
|
||||
"strings"
|
||||
"sundynix-go/global"
|
||||
common "sundynix-go/model/commom/request"
|
||||
"sundynix-go/model/system"
|
||||
sysReq "sundynix-go/model/system/request"
|
||||
"sundynix-go/utils/upload"
|
||||
|
||||
"go.uber.org/zap"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type OssService struct {
|
||||
}
|
||||
|
||||
var OssServiceApp = new(OssService)
|
||||
|
||||
func (o *OssService) Save(file system.Oss) error {
|
||||
return global.DB.Create(&file).Error
|
||||
}
|
||||
|
||||
func (o *OssService) Upload(multipartFile multipart.File, header *multipart.FileHeader) (file system.Oss, err error) {
|
||||
//1.检查是否已有此文件
|
||||
temp, err := header.Open()
|
||||
if err != nil {
|
||||
return file, err
|
||||
}
|
||||
defer temp.Close()
|
||||
|
||||
hasher := md5.New()
|
||||
if _, copyErr := io.Copy(hasher, temp); copyErr != nil {
|
||||
return file, copyErr
|
||||
}
|
||||
// 步骤3: 计算哈希值并转换为十六进制字符串
|
||||
hashBytes := hasher.Sum(nil)
|
||||
hashString := fmt.Sprintf("%x", hashBytes)
|
||||
var exist system.Oss
|
||||
findErr := global.DB.Where("md5 = ?", hashString).First(&exist).Error
|
||||
if findErr == nil && exist.Id != "" {
|
||||
return exist, nil
|
||||
}
|
||||
if errors.Is(findErr, gorm.ErrRecordNotFound) {
|
||||
//不存在的时候保存
|
||||
instance := upload.OssInstance()
|
||||
filepath, key, uploadErr := instance.UploadFile(header)
|
||||
if uploadErr != nil {
|
||||
return file, uploadErr
|
||||
}
|
||||
//文件后缀
|
||||
s := strings.Split(header.Filename, ".")
|
||||
height := 0
|
||||
width := 0
|
||||
//mime类型
|
||||
contentType := header.Header.Get("Content-Type")
|
||||
allowedImageTypes := map[string]bool{
|
||||
"image/jpeg": true,
|
||||
"image/png": true,
|
||||
}
|
||||
isPossibleImage := allowedImageTypes[contentType]
|
||||
//仅当可能是图片时 才计算图片宽高
|
||||
if isPossibleImage {
|
||||
img, _, err1 := image.Decode(multipartFile)
|
||||
if err1 != nil {
|
||||
return file, err
|
||||
}
|
||||
height = img.Bounds().Max.Y
|
||||
width = img.Bounds().Max.X
|
||||
}
|
||||
f := system.Oss{
|
||||
Key: key, // uploads/2025-09-17/
|
||||
Name: header.Filename,
|
||||
Suffix: s[len(s)-1],
|
||||
Tag: s[len(s)-1],
|
||||
Url: filepath, // http://127.0.0.1:9000/planting-fun/uploads/2025-09-17/211476f3837fc7acbaebf0f901c1bd68.png
|
||||
MD5: hashString,
|
||||
Height: height,
|
||||
Width: width,
|
||||
}
|
||||
return f, global.DB.Create(&f).Error
|
||||
}
|
||||
return file, err
|
||||
}
|
||||
|
||||
func (o *OssService) DeleteFileByIds(ids common.IdsReq) error {
|
||||
//循环删除
|
||||
instance := upload.OssInstance()
|
||||
for _, id := range ids.Ids {
|
||||
file, err := o.GetById(id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if err = instance.DeleteFile(file.Key); err != nil {
|
||||
global.Logger.Error("删除文件失败!", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
}
|
||||
err := global.DB.Where("id IN (?)", ids.Ids).Delete(&system.Oss{}).Error
|
||||
return err
|
||||
}
|
||||
|
||||
func (o *OssService) GetById(id string) (system.Oss, error) {
|
||||
var file system.Oss
|
||||
err := global.DB.Where("id = ?", id).First(&file).Error
|
||||
//不存在的时候不要返回错误,而是返回nil
|
||||
if err != nil {
|
||||
return file, nil
|
||||
}
|
||||
return file, err
|
||||
}
|
||||
|
||||
func (o *OssService) GetFileList(info sysReq.GetOssFileList) (list interface{}, total int64, err error) {
|
||||
limit := info.PageSize
|
||||
offset := info.PageSize * (info.Current - 1)
|
||||
db := global.DB.Model(&system.Oss{})
|
||||
var files []system.Oss
|
||||
if info.Name != "" {
|
||||
db = db.Where("name LIKE ?", "%"+info.Name+"%")
|
||||
}
|
||||
err = db.Count(&total).Error
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
err = db.Limit(limit).Offset(offset).Order("created_at desc").Find(&files).Error
|
||||
return files, total, err
|
||||
}
|
||||
@@ -0,0 +1,61 @@
|
||||
package system
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"sundynix-go/global"
|
||||
common "sundynix-go/model/commom/request"
|
||||
"sundynix-go/model/system"
|
||||
systemReq "sundynix-go/model/system/request"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type ClientService struct{}
|
||||
|
||||
var ClientServiceApp = new(ClientService)
|
||||
|
||||
func (s *ClientService) SaveClient(client system.Client) error {
|
||||
if !errors.Is(global.DB.Where("client_id = ?", client.ClientId).First(&system.Client{}).Error, gorm.ErrRecordNotFound) {
|
||||
return errors.New("存在重复clientId,请修改clientId")
|
||||
}
|
||||
return global.DB.Create(&client).Error
|
||||
}
|
||||
|
||||
func (s *ClientService) UpdateClient(client system.Client) error {
|
||||
return global.DB.Model(&client).Where("id = ?", client.Id).Updates(&client).Error
|
||||
}
|
||||
|
||||
func (s *ClientService) GetClientList(info systemReq.GetClientList) (list interface{}, total int64, err error) {
|
||||
limit := info.PageSize
|
||||
offset := info.PageSize * (info.Current - 1)
|
||||
db := global.DB.Model(&system.Client{})
|
||||
var clientList []system.Client
|
||||
if info.ClientId != "" {
|
||||
db = db.Where("client_id = ?", info.ClientId)
|
||||
}
|
||||
if info.Name != "" {
|
||||
db = db.Where("name LIKE ?", "%"+info.Name+"%")
|
||||
}
|
||||
err = db.Count(&total).Error
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
err = db.Limit(limit).Offset(offset).Find(&clientList).Error
|
||||
return clientList, total, err
|
||||
}
|
||||
|
||||
func (s *ClientService) DeleteClientByIds(ids common.IdsReq) (err error) {
|
||||
return global.DB.Where("id IN (?)", ids.Ids).Delete(&system.Client{}).Error
|
||||
}
|
||||
|
||||
func (s *ClientService) GetClientById(id string) (client system.Client, err error) {
|
||||
var c system.Client
|
||||
err = global.DB.Where("id = ?", id).First(&c).Error
|
||||
return c, err
|
||||
}
|
||||
|
||||
func (s *ClientService) GetClientByClientId(clientId string) (client *system.Client, err error) {
|
||||
var c system.Client
|
||||
err = global.DB.Where("client_id = ?", clientId).First(&c).Error
|
||||
return &c, err
|
||||
}
|
||||
@@ -0,0 +1,26 @@
|
||||
package system
|
||||
|
||||
import (
|
||||
"context"
|
||||
"sundynix-go/global"
|
||||
"sundynix-go/utils"
|
||||
)
|
||||
|
||||
type JwtService struct{}
|
||||
|
||||
var JwtServiceApp = new(JwtService)
|
||||
|
||||
// 登出,禁用jwt
|
||||
func (s *JwtService) PutBlacklist(userId string, token string) (err error) {
|
||||
expire, err := utils.ParseDuration(global.Config.JWT.ExpiresTime)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
err = global.Redis.Set(context.Background(), userId, token, expire).Err()
|
||||
return err
|
||||
}
|
||||
|
||||
func (s *JwtService) IsInBlacklist(userId string, token string) bool {
|
||||
val, err := global.Redis.Get(context.Background(), userId).Result()
|
||||
return err == nil && val == token
|
||||
}
|
||||
@@ -0,0 +1,126 @@
|
||||
package system
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"sundynix-go/global"
|
||||
"sundynix-go/model/system"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type MenuService struct{}
|
||||
|
||||
var MenuServiceApp = new(MenuService)
|
||||
|
||||
func (s *MenuService) SaveMenu(menu system.Menu) error {
|
||||
//1.根据code和name查询是否存在重名
|
||||
if err := global.DB.Where("code = ? or name = ?", menu.Code, menu.Name).First(&system.Menu{}).Error; err == nil {
|
||||
return errors.New("菜单已存在")
|
||||
}
|
||||
return global.DB.Create(&menu).Error
|
||||
}
|
||||
|
||||
func (s *MenuService) UpdateMenu(menu *system.Menu) (err error) {
|
||||
var sysMenu system.Menu
|
||||
menuMap := map[string]interface{}{
|
||||
"Category": menu.Category,
|
||||
"Name": menu.Name,
|
||||
"Title": menu.Title,
|
||||
"Code": menu.Code,
|
||||
"Permission": menu.Permission,
|
||||
"Locale": menu.Locale,
|
||||
"Icon": menu.Icon,
|
||||
"Sort": menu.Sort,
|
||||
}
|
||||
err = global.DB.Where("id = ?", menu.Id).First(&sysMenu).Error
|
||||
if err != nil {
|
||||
global.Logger.Debug(err.Error())
|
||||
return errors.New("查询菜单失败")
|
||||
}
|
||||
err = global.DB.Model(&sysMenu).Updates(menuMap).Error
|
||||
return err
|
||||
}
|
||||
|
||||
func (s *MenuService) DeleteMenu(id string) (err error) {
|
||||
err = global.DB.First(&system.Menu{}, "parent_id = ?", id).Error
|
||||
if err == nil {
|
||||
return errors.New("请先删除子菜单")
|
||||
}
|
||||
var menu system.Menu
|
||||
err = global.DB.Where("id = ?", id).First(&menu).Error
|
||||
if err != nil {
|
||||
return errors.New("菜单记录不存在")
|
||||
}
|
||||
// 同步删除menu表和role-menu表数据
|
||||
return global.DB.Transaction(func(tx *gorm.DB) error {
|
||||
if err = tx.Where("id = ?", id).Delete(&system.Menu{}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
if err = tx.Where("menu_id = ?", id).Delete(&system.RoleMenu{}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
})
|
||||
}
|
||||
|
||||
func (s *MenuService) GetMenuById(id string) (menu system.Menu, err error) {
|
||||
var m system.Menu
|
||||
err = global.DB.Where("id = ?", id).First(&m).Error
|
||||
return m, err
|
||||
}
|
||||
|
||||
func (s *MenuService) GetAllMenuTree(category int, parentId string) (menus []*system.Menu, err error) {
|
||||
//1,先根据category和parentId获取所有菜单 category默认为0,parentId默认为0
|
||||
//2.讲查询出的列表构建为树结构
|
||||
var menuList []*system.Menu
|
||||
db := global.DB.Model(&system.Menu{})
|
||||
if category != 0 {
|
||||
db.Where("category = ?", category)
|
||||
}
|
||||
if parentId != "0" {
|
||||
db.Where("parent_id = ?", parentId)
|
||||
}
|
||||
err = db.Order("sort asc").Find(&menuList).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
tree := buildMenuTree(menuList)
|
||||
return tree, nil
|
||||
}
|
||||
|
||||
func (s *MenuService) GetUserRoutes(userId string) (menus []*system.Menu, err error) {
|
||||
//1.根据userId 查询角色 根据角色查询菜单 去重
|
||||
//2.构建树结构
|
||||
var roleIds []string
|
||||
err = global.DB.Model(&system.UserRole{}).Where("user_id = ?", userId).Pluck("role_id", &roleIds).Error
|
||||
var menuIds []string
|
||||
err = global.DB.Model(&system.RoleMenu{}).Where("role_id in ?", roleIds).Pluck("menu_id", &menuIds).Error
|
||||
var menuList []*system.Menu
|
||||
err = global.DB.Model(&system.Menu{}).Where("id in ?", menuIds).Order("sort asc").Find(&menuList).Error
|
||||
return buildMenuTree(menuList), nil
|
||||
}
|
||||
|
||||
func buildMenuTree(list []*system.Menu) []*system.Menu {
|
||||
//1.定义一个map
|
||||
menuMap := make(map[string]*system.Menu)
|
||||
for _, item := range list {
|
||||
menuMap[item.Id] = item
|
||||
}
|
||||
//构建树结构
|
||||
var treeList []*system.Menu
|
||||
for _, item := range list {
|
||||
if item.ParentId == "0" {
|
||||
// 如果没有父节点,直接添加到树中
|
||||
treeList = append(treeList, item)
|
||||
} else {
|
||||
if parent, exists := menuMap[item.ParentId]; exists {
|
||||
// 如果有父节点,将当前节点添加到父节点的Children中
|
||||
parent.Children = append(parent.Children, item)
|
||||
} else {
|
||||
// 如果没有父节点,将当前节点添加到树中
|
||||
treeList = append(treeList, item)
|
||||
}
|
||||
}
|
||||
}
|
||||
return treeList
|
||||
}
|
||||
@@ -0,0 +1,57 @@
|
||||
package system
|
||||
|
||||
import (
|
||||
"sundynix-go/global"
|
||||
common "sundynix-go/model/commom/request"
|
||||
"sundynix-go/model/system"
|
||||
systemReq "sundynix-go/model/system/request"
|
||||
)
|
||||
|
||||
type OperationRecordService struct{}
|
||||
|
||||
var OperationRecordServiceApp = new(OperationRecordService)
|
||||
|
||||
func (o *OperationRecordService) CreateOperationRecord(operationRecord system.SysOperationRecord) (err error) {
|
||||
return global.DB.Create(&operationRecord).Error
|
||||
}
|
||||
|
||||
func (o *OperationRecordService) GetRecordList(info systemReq.GetOperationRecordList) (list interface{}, total int64, err error) {
|
||||
limit := info.PageSize
|
||||
offset := info.PageSize * (info.Current - 1)
|
||||
db := global.DB.Model(&system.SysOperationRecord{})
|
||||
var operationRecordList []system.SysOperationRecord
|
||||
|
||||
if info.Ip != "" {
|
||||
db = db.Where("ip = ?", info.Method)
|
||||
}
|
||||
if info.Method != "" {
|
||||
db = db.Where("method = ?", info.Method)
|
||||
}
|
||||
if info.Path != "" {
|
||||
db = db.Where("path = ?", info.Path)
|
||||
}
|
||||
if info.UserId != "" {
|
||||
db = db.Where("status = ?", info.UserId)
|
||||
}
|
||||
if info.Status != 0 {
|
||||
db = db.Where("status = ?", info.Status)
|
||||
}
|
||||
err = db.Count(&total).Error
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
err = db.Limit(limit).Offset(offset).Find(&operationRecordList).Error
|
||||
return operationRecordList, total, err
|
||||
}
|
||||
|
||||
func (o *OperationRecordService) GetRecordById(id string) (record system.SysOperationRecord, err error) {
|
||||
var r system.SysOperationRecord
|
||||
err = global.DB.Where("id = ?", id).First(&r).Error
|
||||
return r, err
|
||||
}
|
||||
|
||||
func (o *OperationRecordService) DeleteRecordsByIds(ids common.IdsReq) (err error) {
|
||||
// Unscoped()禁用软删除 --> 永久物理删除
|
||||
err = global.DB.Where("id in ?", ids.Ids).Unscoped().Delete(&system.SysOperationRecord{}).Error
|
||||
return err
|
||||
}
|
||||
@@ -0,0 +1,89 @@
|
||||
package system
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"sundynix-go/global"
|
||||
common "sundynix-go/model/commom/request"
|
||||
"sundynix-go/model/system"
|
||||
systemReq "sundynix-go/model/system/request"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type RoleService struct {
|
||||
}
|
||||
|
||||
var RoleServiceApp = new(RoleService)
|
||||
|
||||
func (s *RoleService) SaveRole(role system.Role) error {
|
||||
if !errors.Is(global.DB.Where("code = ?", role.Code).First(&system.Role{}).Error, gorm.ErrRecordNotFound) {
|
||||
return errors.New("存在重复角色")
|
||||
}
|
||||
return global.DB.Create(&role).Error
|
||||
}
|
||||
|
||||
func (s *RoleService) UpdateRole(role system.Role) error {
|
||||
return global.DB.Model(&role).Where("id = ?", role.Id).Updates(&role).Error
|
||||
}
|
||||
|
||||
func (s *RoleService) GetRoleList(info systemReq.GetRoleList) (list interface{}, total int64, err error) {
|
||||
limit := info.PageSize
|
||||
offset := info.PageSize * (info.Current - 1)
|
||||
db := global.DB.Model(&system.Role{})
|
||||
var roleList []system.Role
|
||||
if info.Code != "" {
|
||||
db = db.Where("code = ?", info.Code)
|
||||
}
|
||||
if info.Name != "" {
|
||||
db = db.Where("name LIKE ?", "%"+info.Name+"%")
|
||||
}
|
||||
err = db.Count(&total).Error
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
err = db.Limit(limit).Offset(offset).Find(&roleList).Error
|
||||
return roleList, total, err
|
||||
}
|
||||
|
||||
func (s *RoleService) DeleteRoleByIds(ids common.IdsReq) error {
|
||||
return global.DB.Where("id in ?", ids.Ids).Delete(&system.Role{}).Error
|
||||
}
|
||||
|
||||
func (s *RoleService) GetRoleById(id string) (role system.Role, err error) {
|
||||
var r system.Role
|
||||
err = global.DB.Where("id = ?", id).First(&r).Error
|
||||
return r, err
|
||||
}
|
||||
|
||||
func (s *RoleService) GrantRole(userId string, roleIds []string) error {
|
||||
//1. 检查是否存在userid的授权记录 存在就删除 不存在就插入
|
||||
//2. 插入新的数据
|
||||
return global.DB.Transaction(func(tx *gorm.DB) error {
|
||||
if err := tx.Where("user_id = ?", userId).Delete(&system.UserRole{}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
for _, roleId := range roleIds {
|
||||
if err := tx.Create(&system.UserRole{UserId: userId, RoleId: roleId}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
})
|
||||
|
||||
}
|
||||
|
||||
func (s *RoleService) GrantMenu(roleId string, menuIds []string) error {
|
||||
//1. 检查是否存在userid的授权记录 存在就删除 不存在就插入
|
||||
//2. 插入新的数据
|
||||
return global.DB.Transaction(func(tx *gorm.DB) error {
|
||||
if err := tx.Where("role_id = ?", roleId).Delete(&system.RoleMenu{}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
for _, menuId := range menuIds {
|
||||
if err := tx.Create(&system.RoleMenu{RoleId: roleId, MenuId: menuId}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
})
|
||||
}
|
||||
@@ -0,0 +1,269 @@
|
||||
package system
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"log"
|
||||
"net/http"
|
||||
url2 "net/url"
|
||||
"strconv"
|
||||
"sundynix-go/global"
|
||||
common "sundynix-go/model/commom/request"
|
||||
"sundynix-go/model/system"
|
||||
systemReq "sundynix-go/model/system/request"
|
||||
systemResp "sundynix-go/model/system/response"
|
||||
"sundynix-go/pkg/httpclient"
|
||||
"sundynix-go/utils"
|
||||
location "sundynix-go/utils/location"
|
||||
"sundynix-go/utils/uniqueid"
|
||||
"sundynix-go/utils/wechat"
|
||||
|
||||
"go.uber.org/zap"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type UserService struct{}
|
||||
|
||||
var UserServiceApp = new(UserService)
|
||||
|
||||
func (userService *UserService) Login(u *system.User) (userInfo *system.User, err error) {
|
||||
var user system.User
|
||||
// 查询出用户信息的同时查询出角色信息
|
||||
err = global.DB.Model(&system.User{}).Where("account = ?", u.Account).First(&user).Error
|
||||
if err == nil {
|
||||
if ok := utils.BcryptCheck(u.Password, user.Password); !ok {
|
||||
return nil, errors.New("密码错误")
|
||||
}
|
||||
}
|
||||
return &user, err
|
||||
}
|
||||
|
||||
func (userService *UserService) SaveUser(user system.User) error {
|
||||
if !errors.Is(global.DB.Where("account = ?", user.Account).First(&system.User{}).Error, gorm.ErrRecordNotFound) {
|
||||
return errors.New("存在重复Account,请修改Account")
|
||||
}
|
||||
user.Password = utils.BcryptHash(user.Password)
|
||||
return global.DB.Create(&user).Error
|
||||
}
|
||||
|
||||
func (userService *UserService) UpdateUser(user *system.User) (err error) {
|
||||
var sysUser system.User
|
||||
userMap := map[string]interface{}{
|
||||
"account": user.Account,
|
||||
"phone": user.Phone,
|
||||
"name": user.Name,
|
||||
"avatar_id": user.AvatarId,
|
||||
}
|
||||
err = global.DB.Where("id = ?", user.Id).First(&sysUser).Error
|
||||
if err != nil {
|
||||
global.Logger.Debug(err.Error())
|
||||
return errors.New("查询用户失败")
|
||||
}
|
||||
err = global.DB.Model(&sysUser).Updates(userMap).Error
|
||||
return err
|
||||
}
|
||||
|
||||
func (userService *UserService) GetUserList(info systemReq.GetUserList) (list interface{}, total int64, err error) {
|
||||
limit := info.PageSize
|
||||
offset := info.PageSize * (info.Current - 1)
|
||||
db := global.DB.Model(&system.User{})
|
||||
var userList []system.User
|
||||
|
||||
if info.Account != "" {
|
||||
db = db.Where("account LIKE ?", "%"+info.Account+"%")
|
||||
}
|
||||
if info.Phone != "" {
|
||||
db = db.Where("phone LIKE ?", "%"+info.Phone+"%")
|
||||
}
|
||||
err = db.Count(&total).Error
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
err = db.Limit(limit).Offset(offset).Find(&userList).Error
|
||||
return userList, total, err
|
||||
}
|
||||
|
||||
func (userService *UserService) DeleteUserByIds(ids common.IdsReq) error {
|
||||
return global.DB.Where("id IN (?)", ids.Ids).Delete(&system.User{}).Error
|
||||
}
|
||||
|
||||
func (userService *UserService) GetUserById(id string) (user *system.User, err error) {
|
||||
var u system.User
|
||||
err = global.DB.Where("id = ?", id).Preload("Avatar").First(&u).Error
|
||||
return &u, err
|
||||
}
|
||||
|
||||
func (userService *UserService) ChangePassword(id string, pwd string) (err error) {
|
||||
return global.DB.Model(&system.User{}).Where("id = ?", id).Update("password", utils.BcryptHash(pwd)).Error
|
||||
}
|
||||
|
||||
func (userService *UserService) MiniLogin(code string) (result *system.User, err error) {
|
||||
//构建参数
|
||||
params := url2.Values{}
|
||||
params.Set("appid", global.Config.MiniProgram.AppId)
|
||||
params.Set("secret", global.Config.MiniProgram.AppSecret)
|
||||
params.Set("js_code", code)
|
||||
params.Set("grant_type", "authorization_code")
|
||||
fullURL := "https://api.weixin.qq.com/sns/jscode2session?" + params.Encode()
|
||||
|
||||
//1. 获取全局 HTTP Client(复用连接池)
|
||||
myHttpClient := httpclient.GetClient()
|
||||
//2.发起请求
|
||||
resp, err := myHttpClient.Get(fullURL)
|
||||
if err != nil {
|
||||
global.Logger.Error("微信登录接口请求失败", zap.Error(err))
|
||||
return nil, fmt.Errorf("微信登录接口请求失败: %w", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
//3.读取响应
|
||||
body, err := io.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
global.Logger.Error("读取微信接口响应失败", zap.Error(err))
|
||||
return nil, fmt.Errorf("读取微信登录接口响应失败: %w", err)
|
||||
}
|
||||
// 4. 解析JSON(用结构体替代map,提升效率+类型安全)
|
||||
var wxResp systemResp.WxCode2SessionResp
|
||||
if err = json.Unmarshal(body, &wxResp); err != nil {
|
||||
global.Logger.Error("解析微信接口响应失败", zap.Error(err))
|
||||
return nil, fmt.Errorf("解析微信登录接口响应失败: %w", err)
|
||||
}
|
||||
// 5. 检查微信接口错误码(关键:原代码未处理errcode,导致无法定位真实错误)
|
||||
if wxResp.Errcode != 0 {
|
||||
errMsg := fmt.Sprintf("微信接口返回错误: errcode=%d, errmsg=%s", wxResp.Errcode, wxResp.Errmsg)
|
||||
global.Logger.Error(errMsg)
|
||||
return nil, errors.New(errMsg)
|
||||
}
|
||||
// 6. 校验openid(空值直接返回)
|
||||
if wxResp.Openid == "" {
|
||||
global.Logger.Error("微信接口返回openid为空")
|
||||
return nil, errors.New("openid为空")
|
||||
}
|
||||
|
||||
// 7. 根据openid查询用户 存在--> 更新session_key 返回数据
|
||||
var user system.User
|
||||
err = global.DB.Where("mini_open_id = ?", wxResp.Openid).Preload("Avatar").First(&user).Error
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
// 8. 使用 Transaction 闭包管理事务
|
||||
err = global.DB.Transaction(func(tx *gorm.DB) error {
|
||||
// 创建新用户
|
||||
newUser := system.User{
|
||||
Name: uniqueid.GenerateName(),
|
||||
MiniOpenId: wxResp.Openid,
|
||||
SessionKey: wxResp.SessionKey,
|
||||
}
|
||||
if err := tx.Create(&newUser).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
//// 归一化到当天的0点(本地时区)
|
||||
//now := time.Now()
|
||||
//today := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, time.Local)
|
||||
//personal := plant.Personal{
|
||||
// UserId: newUser.Id,
|
||||
// JoinDate: today,
|
||||
// PlantCount: 0,
|
||||
// CareCount: 0,
|
||||
// BadgeCount: 0,
|
||||
//}
|
||||
//if err := tx.Create(&personal).Error; err != nil {
|
||||
// return err
|
||||
//}
|
||||
// 赋值给外部变量以便返回
|
||||
user = newUser
|
||||
return nil
|
||||
})
|
||||
if err != nil {
|
||||
global.Logger.Error("创建用户失败", zap.Error(err))
|
||||
return nil, fmt.Errorf("登录失败: %w", err)
|
||||
}
|
||||
return &user, nil
|
||||
}
|
||||
if err == nil && user.Id != "" {
|
||||
// UpdateColumn:只更新字段,不触发模型钩子,比Update更高效
|
||||
if err = global.DB.Model(&user).UpdateColumn("session_key", wxResp.SessionKey).Error; err != nil {
|
||||
global.Logger.Error("更新session_key失败", zap.Error(err))
|
||||
return nil, fmt.Errorf("更新session_key失败: %w", err)
|
||||
}
|
||||
return &user, nil
|
||||
}
|
||||
return nil, errors.New("登录失败")
|
||||
}
|
||||
|
||||
func (userService *UserService) LoginByPhone(code string, openId string) (result *system.User, err error) {
|
||||
token := wechat.GetMiniAccessToken()
|
||||
url := "https://api.weixin.qq.com/wxa/business/getuserphonenumber?access_token=" + token
|
||||
data := map[string]interface{}{
|
||||
"code": code,
|
||||
}
|
||||
jsonData, _ := json.Marshal(data)
|
||||
resp, err := http.Post(url, "application/json", bytes.NewBuffer(jsonData))
|
||||
if err != nil {
|
||||
log.Fatalf("Error making POST request: %s", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
body, err := io.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
log.Fatalf("Error reading response body: %s", err)
|
||||
}
|
||||
var dataMap map[string]interface{}
|
||||
err = json.Unmarshal([]byte(body), &dataMap)
|
||||
if err != nil {
|
||||
log.Fatalf("Error unmarshalling JSON: %s", err)
|
||||
}
|
||||
//用户已经存在 --> 如果有手机号直接返回user,没有则更新手机号并返回user
|
||||
if openId != "" {
|
||||
var user system.User
|
||||
err = global.DB.Where("mini_open_id = ?", openId).First(&user).Error
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return nil, errors.New("用户不存在")
|
||||
}
|
||||
if err == nil && user.Id != "" {
|
||||
if user.Phone != "" {
|
||||
return &user, nil
|
||||
} else {
|
||||
user.Phone = dataMap["phone_info"].(map[string]interface{})["phoneNumber"].(string)
|
||||
return &user, global.DB.Save(&user).Error
|
||||
}
|
||||
}
|
||||
}
|
||||
return nil, errors.New("登录失败")
|
||||
}
|
||||
|
||||
// GetLocation 获取位置信息
|
||||
func (userService *UserService) GetLocation(longitude, latitude string) (res map[string]interface{}, err error) {
|
||||
long, err := strconv.ParseFloat(longitude, 32)
|
||||
if err != nil {
|
||||
return res, err
|
||||
}
|
||||
lati, err := strconv.ParseFloat(latitude, 32)
|
||||
if err != nil {
|
||||
return res, err
|
||||
}
|
||||
|
||||
entity, err := location.Point2code(float32(long), float32(lati))
|
||||
result := map[string]interface{}{
|
||||
"city": entity.AddressComponent.City.String(),
|
||||
"adcode": entity.AddressComponent.Adcode,
|
||||
}
|
||||
return result, err
|
||||
}
|
||||
|
||||
// GetWeather 获取天气信息 adcode 行政区划代码
|
||||
func (userService *UserService) GetWeather(adcode string) (res map[string]interface{}, err error) {
|
||||
weatherResp, err := location.GetWeather(adcode, "base")
|
||||
live := weatherResp.Lives[0] // 实时天气数组仅1条数据
|
||||
result := map[string]interface{}{
|
||||
"province": live.Province,
|
||||
"city": live.City,
|
||||
"adcode": live.Adcode,
|
||||
"weather": live.Weather,
|
||||
"temperature": live.Temperature,
|
||||
"windPower": live.WindPower,
|
||||
"windDirection": live.WindDirection,
|
||||
"humidity": live.Humidity,
|
||||
"reportTime": live.ReportTime,
|
||||
}
|
||||
return result, err
|
||||
}
|
||||
Reference in New Issue
Block a user