init: initial commit
This commit is contained in:
@@ -0,0 +1,13 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"sundynix-go/service/plant"
|
||||
"sundynix-go/service/system"
|
||||
)
|
||||
|
||||
var GroupApp = new(Group)
|
||||
|
||||
type Group struct {
|
||||
SystemServiceGroup system.ServiceGroup
|
||||
PlantServiceGroup plant.ServiceGroup
|
||||
}
|
||||
@@ -0,0 +1,5 @@
|
||||
package plant
|
||||
|
||||
type ServiceGroup struct {
|
||||
MyPlantService
|
||||
}
|
||||
@@ -0,0 +1,224 @@
|
||||
package plant
|
||||
|
||||
import (
|
||||
"sundynix-go/global"
|
||||
common "sundynix-go/model/commom/request"
|
||||
"sundynix-go/model/plant"
|
||||
plantReq "sundynix-go/model/plant/request"
|
||||
plantRes "sundynix-go/model/plant/response"
|
||||
"sundynix-go/model/system"
|
||||
"sundynix-go/utils/timer"
|
||||
"time"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type MyPlantService struct{}
|
||||
|
||||
var MyPlantServiceApp = new(MyPlantService)
|
||||
|
||||
// AddPlant 添加植物
|
||||
func (s *MyPlantService) AddPlant(req plantReq.CreateMyPlant, id string) error {
|
||||
err := global.DB.Transaction(func(tx *gorm.DB) error {
|
||||
//1. 验证oss是否存在
|
||||
var ossList []*system.Oss
|
||||
err := tx.Where("id in ?", req.OssIds).Find(&ossList).Error
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
bgTime := timer.GetZeroTime()
|
||||
//2. 处理养护计划
|
||||
var carePlans []*plant.CarePlan
|
||||
for _, v := range req.CarePlans {
|
||||
carePlans = append(carePlans, &plant.CarePlan{
|
||||
UserId: id,
|
||||
Name: v.Name,
|
||||
Icon: v.Icon,
|
||||
Period: v.Period,
|
||||
})
|
||||
}
|
||||
//3.保存数据
|
||||
myPlant := plant.MyPlant{
|
||||
UserId: id,
|
||||
Name: req.Name,
|
||||
PlantTime: bgTime,
|
||||
Status: 1,
|
||||
Placement: req.Placement,
|
||||
PotMaterial: req.PotMaterial,
|
||||
PotSize: req.PotSize,
|
||||
Sunlight: req.Sunlight,
|
||||
PlantingMaterial: req.PlantingMaterial,
|
||||
CarePlans: carePlans,
|
||||
}
|
||||
err = tx.Create(&myPlant).Error
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
//4.处理图片关系
|
||||
if len(ossList) > 0 {
|
||||
var relations []map[string]interface{}
|
||||
for _, oss := range ossList {
|
||||
relations = append(relations, map[string]interface{}{
|
||||
"my_plant_id": myPlant.Id,
|
||||
"oss_id": oss.Id,
|
||||
})
|
||||
}
|
||||
err = tx.Table("sundynix_my_plant_oss").Create(relations).Error
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
})
|
||||
return err
|
||||
|
||||
}
|
||||
|
||||
// PlantPage 植物列表
|
||||
func (s *MyPlantService) PlantPage(req common.PageInfo, userId string) (list interface{}, total int64, err error) {
|
||||
limit := req.PageSize
|
||||
offset := req.PageSize * (req.Current - 1)
|
||||
db := global.DB.Model(&plant.MyPlant{}).Preload("ImgList", func(db *gorm.DB) *gorm.DB {
|
||||
return db.Order("created_at desc")
|
||||
})
|
||||
var myPlants []*plant.MyPlant
|
||||
db = db.Where("user_id = ?", userId)
|
||||
err = db.Count(&total).Error
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
err = db.Limit(limit).Offset(offset).Order("created_at desc").Find(&myPlants).Error
|
||||
return myPlants, total, err
|
||||
}
|
||||
|
||||
// PlantDetail 植物详情
|
||||
func (s *MyPlantService) PlantDetail(id string) (p plant.MyPlant, err error) {
|
||||
var res plant.MyPlant
|
||||
err = global.DB.Where("id = ?", id).
|
||||
Preload("ImgList", func(db *gorm.DB) *gorm.DB {
|
||||
return db.Order("created_at desc")
|
||||
}).
|
||||
Preload("CarePlans").
|
||||
Preload("CareRecords").
|
||||
First(&res).Error
|
||||
|
||||
//不存在的时候不要返回错误,而是返回nil
|
||||
if err != nil {
|
||||
return res, err
|
||||
}
|
||||
return res, nil
|
||||
}
|
||||
|
||||
func (s *MyPlantService) UpdatePlant(req plantReq.UpdateMyPlant) error {
|
||||
// 以map形式更新 先构建map
|
||||
updateMap := map[string]interface{}{
|
||||
"name": req.Name,
|
||||
"placement": req.Placement,
|
||||
"planting_material": req.PlantingMaterial,
|
||||
"pot_material": req.PotMaterial,
|
||||
"pot_size": req.PotSize,
|
||||
"sunlight": req.Sunlight,
|
||||
}
|
||||
return global.DB.Model(&plant.MyPlant{}).Where("id = ?", req.Id).Updates(updateMap).Error
|
||||
}
|
||||
|
||||
// TodayTask 今日任务
|
||||
func (s *MyPlantService) TodayTask(userId string) ([]plantRes.PlantTaskVO, error) {
|
||||
today := timer.GetZeroTime()
|
||||
endOfToday := today.Add(24 * time.Hour)
|
||||
|
||||
var tasks []*plant.CareTask
|
||||
// 查询条件:1.未完成的任务(包含今天和以前逾期的) 2.今天已经完成的任务
|
||||
err := global.DB.Where("user_id = ? AND ("+
|
||||
"(status = 1 AND due_date < ?) OR "+
|
||||
"(status = 2 AND completed_at >= ? AND completed_at < ?)"+
|
||||
")", userId, endOfToday, today, endOfToday).
|
||||
Order("due_date ASC").
|
||||
Find(&tasks).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
// 2.内存处理:聚合数据
|
||||
plantTaskMap := make(map[string][]*plant.CareTask)
|
||||
expiredMap := make(map[string]bool)
|
||||
var plantIds []string
|
||||
for _, t := range tasks {
|
||||
plantIds = append(plantIds, t.PlantId)
|
||||
plantTaskMap[t.PlantId] = append(plantTaskMap[t.PlantId], t)
|
||||
// 判定右上角红标:状态为待办 且 应做时间早于今天
|
||||
if t.Status == 1 && t.DueDate.Before(today) {
|
||||
expiredMap[t.PlantId] = true
|
||||
}
|
||||
}
|
||||
//3.批量查询植物
|
||||
var myPlants []*plant.MyPlant
|
||||
err = global.DB.Where("id in ?", plantIds).Preload("ImgList", func(db *gorm.DB) *gorm.DB {
|
||||
return db.Order("created_at desc")
|
||||
}).Find(&myPlants).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
//4.组装结果
|
||||
var res []plantRes.PlantTaskVO
|
||||
for _, p := range myPlants {
|
||||
plantTaskVO := plantRes.PlantTaskVO{
|
||||
MyPlant: p,
|
||||
HasExpired: expiredMap[p.Id],
|
||||
Tasks: plantTaskMap[p.Id],
|
||||
}
|
||||
res = append(res, plantTaskVO)
|
||||
}
|
||||
return res, nil
|
||||
}
|
||||
|
||||
// CompleteTask 完成任务
|
||||
func (s *MyPlantService) CompleteTask(req plantReq.CompleteTask) error {
|
||||
return global.DB.Transaction(func(tx *gorm.DB) error {
|
||||
var task plant.CareTask
|
||||
if err := tx.Where("id = ?", req.TaskId).First(&task).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
//1.更新当前任务为完成
|
||||
updateData := map[string]interface{}{
|
||||
"status": 2,
|
||||
"completed_at": time.Now(),
|
||||
}
|
||||
if err := tx.Model(&task).Where("id = ?", req.TaskId).Updates(updateData).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
//2.获取计划模版
|
||||
var plan plant.CarePlan
|
||||
if err := tx.Where("id = ?", task.PlanId).First(&plan).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
//3.生成下个周期的任务
|
||||
today := timer.GetZeroTime()
|
||||
nextDueDate := today.AddDate(0, 0, plan.Period)
|
||||
newTask := plant.CareTask{
|
||||
UserId: plan.UserId,
|
||||
PlantId: plan.PlantId,
|
||||
PlanId: plan.Id,
|
||||
Name: plan.Name,
|
||||
Icon: plan.Icon,
|
||||
DueDate: nextDueDate,
|
||||
Status: 1,
|
||||
}
|
||||
if err := tx.Create(&newTask).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
//4.保存养护记录
|
||||
record := plant.CareRecord{
|
||||
UserId: plan.UserId,
|
||||
PlantId: plan.PlantId,
|
||||
PlanId: plan.Id,
|
||||
Name: plan.Name,
|
||||
Icon: plan.Icon,
|
||||
Remark: req.Remark,
|
||||
}
|
||||
if err := tx.Create(&record).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
})
|
||||
}
|
||||
@@ -0,0 +1,206 @@
|
||||
package plant
|
||||
|
||||
//
|
||||
//import (
|
||||
// "context"
|
||||
// "errors"
|
||||
// "net/http"
|
||||
// "sundynix-go/global"
|
||||
// "sundynix-go/model/plant"
|
||||
// "sundynix-go/model/system"
|
||||
// "sundynix-go/utils/wechat"
|
||||
//
|
||||
// "github.com/gin-gonic/gin"
|
||||
// "github.com/wechatpay-apiv3/wechatpay-go/core"
|
||||
// "github.com/wechatpay-apiv3/wechatpay-go/core/auth/verifiers"
|
||||
// "github.com/wechatpay-apiv3/wechatpay-go/core/notify"
|
||||
// "github.com/wechatpay-apiv3/wechatpay-go/services/payments"
|
||||
// "github.com/wechatpay-apiv3/wechatpay-go/services/payments/jsapi"
|
||||
// "github.com/wechatpay-apiv3/wechatpay-go/utils"
|
||||
// "go.uber.org/zap"
|
||||
// "gorm.io/gorm"
|
||||
//)
|
||||
//
|
||||
//type PayService struct{}
|
||||
//
|
||||
//var PayServiceApp = new(PayService)
|
||||
//
|
||||
//// PrePay 预支付
|
||||
//func (s *PayService) PrePay(orderId, userId string) (resp *jsapi.PrepayWithRequestPaymentResponse, err error) {
|
||||
// //1.查询订单和 用户
|
||||
// var order plant.Order
|
||||
// err = global.DB.Where("id = ?", orderId).First(&order).Error
|
||||
// if err != nil {
|
||||
// return nil, err
|
||||
// }
|
||||
// var user system.User
|
||||
// err = global.DB.Where("id = ?", userId).First(&user).Error
|
||||
// if err != nil {
|
||||
// return nil, err
|
||||
// }
|
||||
// payClient, err := wechat.GetWxPayClient()
|
||||
// if err != nil {
|
||||
// return nil, err
|
||||
// }
|
||||
//
|
||||
// svc := jsapi.JsapiApiService{Client: payClient}
|
||||
// result, _, err := svc.PrepayWithRequestPayment(context.Background(),
|
||||
// jsapi.PrepayRequest{
|
||||
// Appid: core.String(global.Config.MiniProgram.AppId),
|
||||
// Mchid: core.String(global.Config.WechatPay.MchId),
|
||||
// Description: core.String(order.Name),
|
||||
// OutTradeNo: core.String(order.OutTradeNo),
|
||||
// //TimeExpire: core.Time(time.Now()), //选填
|
||||
// //Attach: core.String("自定义数据说明"), //选填
|
||||
// NotifyUrl: core.String(global.Config.WechatPay.NotifyUrl),
|
||||
// //GoodsTag: core.String("WXG"), //选填
|
||||
// //SupportFapiao: core.Bool(false), //选填
|
||||
// Amount: &jsapi.Amount{
|
||||
// Currency: core.String("CNY"),
|
||||
// Total: core.Int64(int64(order.Amount)),
|
||||
// },
|
||||
// Payer: &jsapi.Payer{
|
||||
// Openid: core.String(user.MiniOpenId),
|
||||
// },
|
||||
// //Detail: &jsapi.Detail{
|
||||
// // CostPrice: core.Int64(608800),
|
||||
// // GoodsDetail: []jsapi.GoodsDetail{jsapi.GoodsDetail{
|
||||
// // GoodsName: core.String("iPhoneX 256G"),
|
||||
// // MerchantGoodsId: core.String("ABC"),
|
||||
// // Quantity: core.Int64(1),
|
||||
// // UnitPrice: core.Int64(828800),
|
||||
// // WechatpayGoodsId: core.String("1001"),
|
||||
// // }},
|
||||
// // InvoiceId: core.String("wx123"),
|
||||
// //}, //选填
|
||||
// //SceneInfo: &jsapi.SceneInfo{
|
||||
// // DeviceId: core.String("013467007045764"),
|
||||
// // PayerClientIp: core.String("14.23.150.211"),
|
||||
// // StoreInfo: &jsapi.StoreInfo{
|
||||
// // Address: core.String("广东省深圳市南山区科技中一道10000号"),
|
||||
// // AreaCode: core.String("440305"),
|
||||
// // Id: core.String("0001"),
|
||||
// // Name: core.String("腾讯大厦分店"),
|
||||
// // },
|
||||
// //},
|
||||
// //SettleInfo: &jsapi.SettleInfo{
|
||||
// // ProfitSharing: core.Bool(false),
|
||||
// //}, //选填
|
||||
// })
|
||||
// if err != nil {
|
||||
// return nil, err
|
||||
// }
|
||||
//
|
||||
// return result, nil
|
||||
//
|
||||
//}
|
||||
//
|
||||
//// PayCallback 支付回调
|
||||
//func (s *PayService) PayCallback(c *gin.Context) error {
|
||||
// //1.加载共钥
|
||||
// mchPublicKeyPath := global.Config.WechatPay.PublicKeyPath
|
||||
// mchPublicKey, err := utils.LoadPublicKeyWithPath(mchPublicKeyPath)
|
||||
// if err != nil {
|
||||
// return err
|
||||
// }
|
||||
// ctx := context.Background()
|
||||
// //2.创建客户端
|
||||
// handler, err := notify.NewRSANotifyHandler(global.Config.WechatPay.MchAPIv3Key,
|
||||
// verifiers.NewSHA256WithRSAPubkeyVerifier(global.Config.WechatPay.PublicKeyId, *mchPublicKey))
|
||||
// if err != nil {
|
||||
// return err
|
||||
// }
|
||||
// //3.验签 解密
|
||||
// //将支付回调通知中的内容,解析为 payments.Transaction。
|
||||
// transaction := new(payments.Transaction)
|
||||
// notifyReq, err := handler.ParseNotifyRequest(ctx, c.Request, transaction)
|
||||
// // 4.如果验签未通过,或者解密失败
|
||||
// if err != nil {
|
||||
// //应答微信
|
||||
// c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{
|
||||
// "code": "FAIL",
|
||||
// "message": "失败",
|
||||
// })
|
||||
// global.Logger.Error("wxPay回调-验签或解密:", zap.Error(err))
|
||||
// return err
|
||||
// }
|
||||
// //5.应答微信
|
||||
// c.Status(http.StatusOK)
|
||||
// // 6.处理通知内容
|
||||
// global.Logger.Info("wxPay回调-成功:", zap.Any("notifyReq", notifyReq.Summary))
|
||||
// //7. 异步处理数据
|
||||
// go func() {
|
||||
// payNotify := plant.PayNotify{
|
||||
// Amount: *transaction.Amount.Total,
|
||||
// Currency: *transaction.Amount.Currency,
|
||||
// PayerCurrency: *transaction.Amount.PayerCurrency,
|
||||
// PayerTotal: *transaction.Amount.PayerTotal,
|
||||
// Appid: *transaction.Appid,
|
||||
// Mchid: *transaction.Mchid,
|
||||
// OutTradeNo: *transaction.OutTradeNo,
|
||||
// Attach: *transaction.Attach,
|
||||
// BankType: *transaction.BankType,
|
||||
// Payer: *transaction.Payer.Openid,
|
||||
// SuccessTime: *transaction.SuccessTime,
|
||||
// TradeState: *transaction.TradeState,
|
||||
// TradeStateDesc: *transaction.TradeStateDesc,
|
||||
// TradeType: *transaction.TradeType,
|
||||
// TransactionId: *transaction.TransactionId,
|
||||
// }
|
||||
// //7.1 回调记录
|
||||
// err = global.DB.Create(&payNotify).Error
|
||||
// if err != nil {
|
||||
// global.Logger.Error("wxPay回调-存储数据异常:", zap.Error(err))
|
||||
// }
|
||||
// if payNotify.TradeState == "SUCCESS" {
|
||||
// //7.2 扣商品库存
|
||||
// var order plant.Order
|
||||
// err = global.DB.Where("out_trade_no = ?", *transaction.OutTradeNo).First(&order).Error
|
||||
// if err != nil {
|
||||
// if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
// global.Logger.Error("wxPay回调-订单不存在:", zap.Error(err))
|
||||
// }
|
||||
// }
|
||||
// //根据订单找到领取记录
|
||||
// var record plant.ClaimPlantRecord
|
||||
// err = global.DB.Where("order_id = ?", order.Id).First(&record).Error
|
||||
// if err != nil {
|
||||
// if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
// global.Logger.Error("wxPay回调-领取记录不存在:", zap.Error(err))
|
||||
// }
|
||||
// }
|
||||
// //根据领取记录找到植物
|
||||
// var claimPlant plant.ClaimPlant
|
||||
// err = global.DB.Where("id = ?", record.ClaimPlantId).First(&claimPlant).Error
|
||||
// if err != nil {
|
||||
// if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
// global.Logger.Error("wxPay回调-植物不存在:", zap.Error(err))
|
||||
// }
|
||||
// }
|
||||
// err = global.DB.Transaction(func(tx *gorm.DB) error {
|
||||
// //扣库存
|
||||
// if err = tx.Model(&claimPlant).Update("stock", gorm.Expr("stock - ?", 1)).Error; err != nil {
|
||||
// return err
|
||||
// }
|
||||
// //订单状态
|
||||
// if err = tx.Model(&order).Update("pay_status", "SUCCESS").Error; err != nil {
|
||||
// return err
|
||||
// }
|
||||
// //修改领取记录为成功
|
||||
// if err = tx.Model(&record).Update("status", 1).Error; err != nil {
|
||||
// return err
|
||||
// }
|
||||
// //减用户积分
|
||||
// if err = tx.Model(&plant.Personal{}).Where("user_id = ?", order.UserId).Update("points_count", gorm.Expr("points_count - ?", claimPlant.Points)).Error; err != nil {
|
||||
// return err
|
||||
// }
|
||||
// return nil
|
||||
// })
|
||||
// }
|
||||
// if err != nil {
|
||||
// return
|
||||
// }
|
||||
// }()
|
||||
// return nil
|
||||
//
|
||||
//}
|
||||
@@ -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