diff --git a/.gitignore b/.gitignore index f4d432a..f8b0bcb 100644 --- a/.gitignore +++ b/.gitignore @@ -6,12 +6,13 @@ *.so *.dylib -# Test binary, built with `go test -c` -*.test - # Output of the go coverage tool, specifically when used with LiteIDE *.out +*.log # Dependency directories (remove the comment below to include it) # vendor/ - +.idea/ +.git/ +.DS_Store/ +.tmp/ \ No newline at end of file diff --git a/api/amqp/consumer/base.go b/api/amqp/consumer/base.go new file mode 100644 index 0000000..1931fdc --- /dev/null +++ b/api/amqp/consumer/base.go @@ -0,0 +1,5 @@ +package consumer + +func checkHandleNumber(key string) { + +} diff --git a/api/controller/Article.go b/api/controller/Article.go new file mode 100644 index 0000000..c18b8c6 --- /dev/null +++ b/api/controller/Article.go @@ -0,0 +1,431 @@ +package controller + +import ( + "fmt" + "github.com/gin-gonic/gin" + "strconv" + "vote-admin-video-api/api/dao" + "vote-admin-video-api/api/dto" + "vote-admin-video-api/api/model" + "vote-admin-video-api/api/requests" + "vote-admin-video-api/api/responses" + "vote-admin-video-api/global" + "vote-admin-video-api/utils" +) + +type Article struct{} + +// GetArticlePage 获取图文列表-分页 +func (r *Article) GetArticlePage(c *gin.Context) { + articleRequest := requests.ArticleRequest{} + req := articleRequest.GetArticlePage + if err := c.ShouldBindJSON(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 参数验证 + if err := global.Validate.Struct(req); err != nil { + responses.FailWithMessage(utils.Translate(err), c) + return + } + + if req.Page == 0 { + req.Page = 1 + } + + if req.PageSize == 0 { + req.PageSize = 20 + } + + if req.Order != nil { + if req.Order.VoteNum != "" { + if req.Order.VoteNum != "desc" && req.Order.VoteNum != "asc" { + responses.FailWithMessage("排序字段错误", c) + return + } + } + } + + // 获取数据 + articleDao := dao.ArticleDao{} + articles, total, err := articleDao.GetArticlePageSearch(req, req.Page, req.PageSize) + if err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 处理返回值 + g := dto.GetArticleListDto(articles) + + result := make(map[string]interface{}) + result["page"] = req.Page + result["page_size"] = req.PageSize + result["total"] = total + result["data"] = g + responses.OkWithData(result, c) +} + +// GetArticle 获取图文详情 +func (r *Article) GetArticle(c *gin.Context) { + id := c.Param("article_id") + if id == "" { + responses.FailWithMessage("缺少参数", c) + return + } + + // 将 id 转换为 int64 类型 + articleId, err := strconv.ParseInt(id, 10, 64) + if err != nil { + responses.Fail(c) + return + } + + // 获取数据 + articleDao := dao.ArticleDao{} + article, err := articleDao.GetArticleById(articleId) + if err != nil { + responses.FailWithMessage("文章错误", c) + return + } + + // 获取作者数据 + articleAuthorDao := dao.ArticleAuthorDao{} + maps := make(map[string]interface{}) + maps["article_id"] = articleId + articleAuthors, err := articleAuthorDao.GetArticleAuthorPreloadList(maps) + if err != nil { + responses.FailWithMessage("文章错误", c) + return + } + + // 获取排名 + rank, _ := articleDao.GetArticleRank(article.ArticleId) + + // 处理返回值 + g := dto.GetArticleDto(article) + + // 加载数据-作者 + g.LoadArticleAuthor(articleAuthors) + + // 加载数据-作者排名 + g.LoadRank(rank) + + responses.OkWithData(g, c) +} + +// PutArticle 修改图文详情 +func (r *Article) PutArticle(c *gin.Context) { + articleRequest := requests.ArticleRequest{} + req := articleRequest.PutArticle + if err := c.ShouldBindJSON(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 参数验证 + if err := global.Validate.Struct(req); err != nil { + responses.FailWithMessage(utils.Translate(err), c) + return + } + + id := c.Param("article_id") + if id == "" { + responses.FailWithMessage("缺少参数", c) + return + } + + // 将 id 转换为 int64 类型 + articleId, err := strconv.ParseInt(id, 10, 64) + if err != nil { + responses.Fail(c) + return + } + + if req.VoteNum != nil { + if *req.VoteNum < 0 { + responses.FailWithMessage("票数需大于0", c) + return + } + } + + // 获取订单数据 + articleDao := dao.ArticleDao{} + article, err := articleDao.GetArticleById(articleId) + if err != nil { + responses.FailWithMessage("图文异常", c) + return + } + + // 开始事务 + tx := global.Db.Begin() + defer func() { + if r := recover(); r != nil { + tx.Rollback() + fmt.Println(r) + } + }() + + articleData := make(map[string]interface{}) + + // 文章标题 + if article.ArticleTitle != req.ArticleTitle { + articleData["article_title"] = req.ArticleTitle + } + + // 文章状态 + if article.ArticleStatus != req.ArticleStatus { + articleData["article_status"] = req.ArticleStatus + } + + // 文章内容 + if article.ArticleContent != req.ArticleContent { + articleData["article_content"] = req.ArticleContent + } + + // 票数 + if req.VoteNum != nil { + if article.VoteNum != *req.VoteNum { + articleData["vote_num"] = *req.VoteNum + } + } + + // 修改 + if len(articleData) > 0 { + err = articleDao.EditArticleById(tx, articleId, articleData) + if err != nil { + tx.Rollback() + responses.FailWithMessage(err.Error(), c) + return + } + } + + // 删除作者 + articleAuthorDao := dao.ArticleAuthorDao{} + maps := make(map[string]interface{}) + maps["article_id"] = articleId + err = articleAuthorDao.DeleteArticleAuthor(tx, maps) + if err != nil { + tx.Rollback() + responses.FailWithMessage(err.Error(), c) + return + } + + baseHospitalDao := dao.BaseHospitalDao{} + // 新增作者 + for _, author := range req.ArticleAuthor { + if author.HospitalId == "" { + tx.Rollback() + responses.FailWithMessage("请选择作者所属医院", c) + return + } + + if author.AuthorName == "" { + tx.Rollback() + responses.FailWithMessage("请输入作者名称", c) + return + } + + // 检测医院 + hospitalId, err := strconv.ParseInt(author.HospitalId, 10, 64) + if err != nil { + tx.Rollback() + responses.FailWithMessage(err.Error(), c) + return + } + + baseHospital, err := baseHospitalDao.GetBaseHospitalById(hospitalId) + if err != nil || baseHospital == nil { + tx.Rollback() + responses.FailWithMessage("医院错误", c) + return + } + + ArticleAuthor := &model.ArticleAuthor{ + ArticleId: articleId, + AuthorName: author.AuthorName, + HospitalId: hospitalId, + } + ArticleAuthor, err = articleAuthorDao.AddArticleAuthor(tx, ArticleAuthor) + if err != nil { + tx.Rollback() + responses.FailWithMessage(err.Error(), c) + return + } + } + + tx.Commit() + responses.Ok(c) +} + +// AddArticle 新增图文详情 +func (r *Article) AddArticle(c *gin.Context) { + articleRequest := requests.ArticleRequest{} + req := articleRequest.AddArticle + if err := c.ShouldBindJSON(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 参数验证 + if err := global.Validate.Struct(req); err != nil { + responses.FailWithMessage(utils.Translate(err), c) + return + } + + if req.VoteNum != nil { + if *req.VoteNum < 0 { + responses.FailWithMessage("票数需大于0", c) + return + } + } + + // 开始事务 + tx := global.Db.Begin() + defer func() { + if r := recover(); r != nil { + tx.Rollback() + responses.FailWithMessage("操作失败", c) + return + } + }() + + articleDao := dao.ArticleDao{} + + article := &model.Article{ + ArticleTitle: req.ArticleTitle, + ArticleStatus: req.ArticleStatus, + ArticleContent: req.ArticleContent, + } + + if req.VoteNum != nil { + article.VoteNum = *req.VoteNum + } + + // 生成文章编号 + maps := make(map[string]interface{}) + total, err := articleDao.GetArticleCount(maps) + if err != nil { + tx.Rollback() + responses.FailWithMessage(err.Error(), c) + return + } + + article.ArticleNumber = fmt.Sprintf("%d", 1000+total+1) + + article, err = articleDao.AddArticle(tx, article) + if err != nil { + tx.Rollback() + responses.FailWithMessage(err.Error(), c) + return + } + + // 新增作者 + baseHospitalDao := dao.BaseHospitalDao{} + articleAuthorDao := dao.ArticleAuthorDao{} + for _, author := range req.ArticleAuthor { + if author.HospitalId == "" { + tx.Rollback() + responses.FailWithMessage("请选择作者所属医院", c) + return + } + + if author.AuthorName == "" { + tx.Rollback() + responses.FailWithMessage("请输入作者名称", c) + return + } + + // 检测医院 + hospitalId, err := strconv.ParseInt(author.HospitalId, 10, 64) + if err != nil { + tx.Rollback() + responses.FailWithMessage(err.Error(), c) + return + } + + baseHospital, err := baseHospitalDao.GetBaseHospitalById(hospitalId) + if err != nil || baseHospital == nil { + tx.Rollback() + responses.FailWithMessage("医院错误", c) + return + } + + ArticleAuthor := &model.ArticleAuthor{ + ArticleId: article.ArticleId, + AuthorName: author.AuthorName, + HospitalId: hospitalId, + } + ArticleAuthor, err = articleAuthorDao.AddArticleAuthor(tx, ArticleAuthor) + if err != nil { + tx.Rollback() + responses.FailWithMessage(err.Error(), c) + return + } + } + + tx.Commit() + responses.Ok(c) +} + +// PutArticleStatus 操作图文状态 +func (r *Article) PutArticleStatus(c *gin.Context) { + articleRequest := requests.ArticleRequest{} + req := articleRequest.PutArticleStatus + if err := c.ShouldBind(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 参数验证 + if err := global.Validate.Struct(req); err != nil { + responses.FailWithMessage(utils.Translate(err), c) + return + } + + id := c.Param("article_id") + if id == "" { + responses.FailWithMessage("缺少参数", c) + return + } + + // 将 id 转换为 int64 类型 + articleId, err := strconv.ParseInt(id, 10, 64) + if err != nil { + responses.Fail(c) + return + } + + // 获取订单数据 + articleDao := dao.ArticleDao{} + article, err := articleDao.GetArticleById(articleId) + if err != nil { + responses.FailWithMessage("图文异常", c) + return + } + + // 开始事务 + tx := global.Db.Begin() + defer func() { + if r := recover(); r != nil { + tx.Rollback() + } + }() + + articleData := make(map[string]interface{}) + if req.ArticleStatus != article.ArticleStatus { + articleData["article_status"] = req.ArticleStatus + } + + if len(articleData) > 0 { + err = articleDao.EditArticleById(tx, articleId, articleData) + if err != nil { + tx.Rollback() + responses.FailWithMessage("修改失败", c) + return + } + } + + tx.Commit() + responses.Ok(c) +} diff --git a/api/controller/Base.go b/api/controller/Base.go new file mode 100644 index 0000000..0f9d7da --- /dev/null +++ b/api/controller/Base.go @@ -0,0 +1,14 @@ +package controller + +// Api api接口 +type Api struct { + Public // 公共方法 + Article // 图文 + Video // 视频 + BaseAgreement // 基础数据-协议 + User // 用户 + UserVoteDay // 投票记录 + System // 配置 + BaseHospital // 基础数据-医院 + Editor // 配置-编辑器 +} diff --git a/api/controller/BaseAgreement.go b/api/controller/BaseAgreement.go new file mode 100644 index 0000000..6b763dc --- /dev/null +++ b/api/controller/BaseAgreement.go @@ -0,0 +1,198 @@ +package controller + +import ( + "github.com/gin-gonic/gin" + "strconv" + "vote-admin-video-api/api/dao" + "vote-admin-video-api/api/dto" + "vote-admin-video-api/api/model" + "vote-admin-video-api/api/requests" + "vote-admin-video-api/api/responses" + "vote-admin-video-api/global" + "vote-admin-video-api/utils" +) + +type BaseAgreement struct{} + +// GetBaseAgreementPage 获取协议列表-分页 +func (b *BaseAgreement) GetBaseAgreementPage(c *gin.Context) { + baseAgreementRequest := requests.BaseAgreementRequest{} + req := baseAgreementRequest.GetBaseAgreementPage + if err := c.ShouldBind(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 参数验证 + if err := global.Validate.Struct(req); err != nil { + responses.FailWithMessage(utils.Translate(err), c) + return + } + + if req.Page == 0 { + req.Page = 1 + } + + if req.PageSize == 0 { + req.PageSize = 20 + } + + // 获取分类数据 + baseAgreementDao := dao.BaseAgreementDao{} + baseAgreement, total, err := baseAgreementDao.GetBaseAgreementPageSearch(req.Page, req.PageSize) + if err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 处理返回值 + g := dto.GetBaseAgreementListDto(baseAgreement) + + result := make(map[string]interface{}) + result["page"] = req.Page + result["page_size"] = req.PageSize + result["total"] = total + result["data"] = g + responses.OkWithData(result, c) +} + +// GetBaseAgreement 获取协议详情 +func (b *BaseAgreement) GetBaseAgreement(c *gin.Context) { + id := c.Param("agreement_id") + if id == "" { + responses.FailWithMessage("缺少参数", c) + return + } + + // 将 id 转换为 int64 类型 + agreementId, err := strconv.ParseInt(id, 10, 64) + if err != nil { + responses.Fail(c) + return + } + + // 获取协议数据 + baseAgreementDao := dao.BaseAgreementDao{} + baseAgreement, err := baseAgreementDao.GetBaseAgreementById(agreementId) + if err != nil { + responses.FailWithMessage("分类异常", c) + return + } + + // 处理返回值 + g := dto.GetBaseAgreementDto(baseAgreement) + + responses.OkWithData(g, c) +} + +// PutBaseAgreement 修改协议 +func (b *BaseAgreement) PutBaseAgreement(c *gin.Context) { + BaseAgreementRequest := requests.BaseAgreementRequest{} + req := BaseAgreementRequest.PutBaseAgreement + if err := c.ShouldBind(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 参数验证 + if err := global.Validate.Struct(req); err != nil { + responses.FailWithMessage(utils.Translate(err), c) + return + } + + id := c.Param("agreement_id") + if id == "" { + responses.FailWithMessage("缺少参数", c) + return + } + + // 将 id 转换为 int64 类型 + agreementId, err := strconv.ParseInt(id, 10, 64) + if err != nil { + responses.Fail(c) + return + } + + // 获取协议数据 + baseAgreementDao := dao.BaseAgreementDao{} + baseAgreement, err := baseAgreementDao.GetBaseAgreementById(agreementId) + if err != nil { + responses.FailWithMessage("分类异常", c) + return + } + + // 修改值 + baseAgreementData := make(map[string]interface{}) + + // 协议标题 + if req.AgreementTitle != baseAgreement.AgreementTitle { + baseAgreementData["agreement_title"] = req.AgreementTitle + } + + // 协议内容 + if req.AgreementContent != baseAgreement.AgreementContent { + baseAgreementData["agreement_content"] = req.AgreementContent + } + + if len(baseAgreementData) > 0 { + // 开始事务 + tx := global.Db.Begin() + defer func() { + if r := recover(); r != nil { + tx.Rollback() + } + }() + + err = baseAgreementDao.EditBaseAgreementById(tx, agreementId, baseAgreementData) + if err != nil { + tx.Rollback() + responses.FailWithMessage("操作失败", c) + return + } + + tx.Commit() + } + + responses.Ok(c) +} + +// AddBaseAgreement 新增协议 +func (b *BaseAgreement) AddBaseAgreement(c *gin.Context) { + BaseAgreementRequest := requests.BaseAgreementRequest{} + req := BaseAgreementRequest.AddBaseAgreement + if err := c.ShouldBind(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 参数验证 + if err := global.Validate.Struct(req); err != nil { + responses.FailWithMessage(utils.Translate(err), c) + return + } + + // 开始事务 + tx := global.Db.Begin() + defer func() { + if r := recover(); r != nil { + tx.Rollback() + } + }() + + baseAgreement := &model.BaseAgreement{ + AgreementTitle: req.AgreementTitle, + AgreementType: req.AgreementType, + AgreementContent: req.AgreementContent, + } + + baseAgreementDao := dao.BaseAgreementDao{} + baseAgreement, err := baseAgreementDao.AddBaseAgreement(tx, baseAgreement) + if err != nil { + tx.Rollback() + responses.FailWithMessage(err.Error(), c) + return + } + + tx.Commit() + responses.Ok(c) +} diff --git a/api/controller/BaseHospital.go b/api/controller/BaseHospital.go new file mode 100644 index 0000000..3a00c43 --- /dev/null +++ b/api/controller/BaseHospital.go @@ -0,0 +1,41 @@ +// Package controller 科室管理 +package controller + +import ( + "github.com/gin-gonic/gin" + "vote-admin-video-api/api/dao" + "vote-admin-video-api/api/dto" + "vote-admin-video-api/api/requests" + "vote-admin-video-api/api/responses" + "vote-admin-video-api/global" + "vote-admin-video-api/utils" +) + +type BaseHospital struct{} + +// GetHospitalList 获取医院列表 +func (b *BaseHospital) GetHospitalList(c *gin.Context) { + baseHospitalRequest := requests.BaseHospitalRequest{} + req := baseHospitalRequest.GetBaseHospitalList + if err := c.ShouldBind(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 参数验证 + if err := global.Validate.Struct(req); err != nil { + responses.FailWithMessage(utils.Translate(err), c) + return + } + + baseHospitalDao := dao.BaseHospitalDao{} + baseHospitals, err := baseHospitalDao.GetBaseHospitalLimitByMaps(req) + if err != nil { + responses.Ok(c) + return + } + + // 处理返回值 + getHospitalLimitResponse := dto.GetBaseHospitalListDto(baseHospitals) + responses.OkWithData(getHospitalLimitResponse, c) +} diff --git a/api/controller/Editor.go b/api/controller/Editor.go new file mode 100644 index 0000000..3489bbe --- /dev/null +++ b/api/controller/Editor.go @@ -0,0 +1,139 @@ +package controller + +import ( + "fmt" + "github.com/gin-gonic/gin" + "io" + "math/rand" + "mime/multipart" + "net/http" + "time" + "vote-admin-video-api/api/responses" + e "vote-admin-video-api/extend/Editor" + "vote-admin-video-api/extend/aliyun" + "vote-admin-video-api/utils" +) + +type Editor struct{} + +// GetEditorConfig 编辑器-获取配置 +func (b *Editor) GetEditorConfig(c *gin.Context) { + action := c.Query("action") + if action == "" { + responses.FailWithMessage("缺少参数", c) + return + } + + // 获取配置 config + if action == "config" { + config := e.GetConfig() + c.JSON(http.StatusOK, config) + return + } + + // 图片列表 listImage + if action == "listImage" { + responses.Ok(c) + return + } + + // 文件列表 listFile + if action == "listFile" { + responses.Ok(c) + return + } + + responses.Ok(c) + return +} + +// EditorUpload 编辑器-上传 +func (b *Editor) EditorUpload(c *gin.Context) { + action := c.Query("action") + if action == "" { + responses.FailWithMessage("缺少参数", c) + return + } + + file, err := c.FormFile("file") + if err != nil { + responses.FailWithMessage("文件错误", c) + return + } + + f, err := file.Open() + if err != nil { + responses.FailWithMessage("文件错误", c) + return + } + defer func(f multipart.File) { + err := f.Close() + if err != nil { + fmt.Println(err) + } + }(f) + + // 读取文件内容到字节切片 + fileBytes, err := io.ReadAll(f) + if err != nil { + responses.FailWithMessage("文件错误", c) + return + } + + // 添加图片水印 + fileBytes, err = utils.AddWatermarkToImage(fileBytes, "./resource/3061726102564.png") + if err != nil { + return + } + + fileType := "jpg" + if file.Filename != "" { + fileType = utils.GetExtension(file.Filename) + if fileType == "" { + fileType = "jpg" + } + } + + now := time.Now() + dateTimeString := now.Format("20060102150405") // 当前时间字符串 + rand.New(rand.NewSource(time.Now().UnixNano())) // 设置随机数 + + var ossPath string + + // 上传图片 image + if action == "image" { + ossPath = "static/images/" + fmt.Sprintf("%d", rand.Intn(9000)+1000) + dateTimeString + "." + fileType + } + + // 上传视频 video + if action == "video" { + ossPath = "static/video/" + fmt.Sprintf("%d", rand.Intn(9000)+1000) + dateTimeString + "." + fileType + } + + // 上传文件 file + if action == "file" { + ossPath = "static/file/" + fmt.Sprintf("%d", rand.Intn(9000)+1000) + dateTimeString + "." + fileType + } + + if ossPath == "" { + responses.FailWithMessage("上传失败", c) + return + } + + // 上传oss + _, err = aliyun.PutObjectByte(ossPath, fileBytes) + if err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + var g e.UploadDto + g.Url = utils.AddOssDomain("/" + ossPath) + g.State = "SUCCESS" + g.Title = dateTimeString + "." + fileType + g.Original = dateTimeString + "." + fileType + g.Type = fileType + + c.JSON(http.StatusOK, g) + return +} diff --git a/api/controller/Public.go b/api/controller/Public.go new file mode 100644 index 0000000..df66eb9 --- /dev/null +++ b/api/controller/Public.go @@ -0,0 +1,226 @@ +package controller + +import ( + "crypto/md5" + "encoding/hex" + "fmt" + "github.com/gin-gonic/gin" + "time" + "vote-admin-video-api/api/dao" + "vote-admin-video-api/api/dto" + "vote-admin-video-api/api/requests" + "vote-admin-video-api/api/responses" + "vote-admin-video-api/config" + "vote-admin-video-api/global" + "vote-admin-video-api/utils" +) + +type Public struct{} + +// Login 登陆 +func (b *Public) Login(c *gin.Context) { + publicRequest := requests.PublicRequest{} + req := publicRequest.Login + if err := c.ShouldBind(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 参数验证 + if err := global.Validate.Struct(req); err != nil { + responses.FailWithMessage(utils.Translate(err), c) + return + } + + if err := c.ShouldBind(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 验证验证码 + if config.C.Env == "prod" { + isValid := utils.VerifyCaptcha(req.CaptchaId, req.Captcha) + if !isValid { + // 验证码错误 + responses.FailWithMessage("验证码错误", c) + return + } + } + + // 获取用户信息 + AdminUserDao := dao.AdminUserDao{} + maps := make(map[string]interface{}) + maps["access"] = req.Access + adminUser, err := AdminUserDao.GetAdminUser(maps) + if err != nil || adminUser == nil { + responses.FailWithMessage("用户名或密码错误", c) + return + } + + // 检测用户密码 + password := md5.Sum([]byte(req.Password + adminUser.Salt)) + // 将哈希值转换为16进制字符串 + passwordString := hex.EncodeToString(password[:]) + + fmt.Println(passwordString) + if passwordString != adminUser.Password { + responses.FailWithMessage("用户名或密码错误", c) + return + } + + // 检测用户状态 + if adminUser.IsDeleted == 1 { + responses.FailWithMessage("非法用户", c) + return + } + + if adminUser.IsDisabled == 1 { + responses.FailWithMessage("您的账号已被禁用，请联系管理员处理", c) + return + } + + // 下发token + token := &utils.Token{ + UserId: fmt.Sprintf("%d", adminUser.UserId), + } + + // 生成jwt + jwt, err := token.NewJWT() + if err != nil || jwt == "" { + responses.FailWithMessage("登陆失败", c) + return + } + + g := dto.AdminLoginDto(adminUser) + + g.LoadToken(jwt) + + responses.OkWithData(g, c) +} + +// GetCaptcha 获取验证码 +func (b *Public) GetCaptcha(c *gin.Context) { + id, b64s, err := utils.GenerateCaptcha() + if err != nil { + responses.FailWithMessage("验证码获取失败", c) + return + } + + responses.OkWithData(gin.H{ + "id": id, + "b64s": b64s, + }, c) +} + +// GetIndex 首页 +func (b *Public) GetIndex(c *gin.Context) { + dataDao := dao.DataDao{} + data, err := dataDao.GetDataById(1) + if err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + g := dto.IndexDto{ + ViewNum: data.ViewNum, + VoteNum: data.VoteNum, + } + + responses.OkWithData(g, c) +} + +// GetIndexData 首页动态统计数据 +func (b *Public) GetIndexData(c *gin.Context) { + publicRequest := requests.PublicRequest{} + req := publicRequest.GetIndexData + if err := c.ShouldBind(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 参数验证 + if err := global.Validate.Struct(req); err != nil { + responses.FailWithMessage(utils.Translate(err), c) + return + } + + if err := c.ShouldBind(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 返回值 + var g []*dto.IndexDataDto + results := make(map[string]int64) + + // 分类（1:新增用户数 2:图文投票数 3:视频投票数） + if req.Type == 1 { + // 新增用户数 + endTime, _ := time.Parse("2006-01-02", req.EndTime) + req.EndTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second).Format("2006-01-02 15:04:05") + + userDao := dao.UserDao{} + maps := make(map[string]interface{}) + users, err := userDao.GetUserListByTime(maps, req.StartTime, req.EndTime, "created_at") + if err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + for _, user := range users { + date := time.Time(user.CreatedAt).Format("2006-01-02") + results[date]++ + } + } + + // 图文投票数 + if req.Type == 2 { + endTime, _ := time.Parse("2006-01-02", req.EndTime) + req.EndTime = endTime.Format("2006-01-02") + + articleVoteDayDao := dao.ArticleVoteDayDao{} + maps := make(map[string]interface{}) + articleVoteDays, err := articleVoteDayDao.GetArticleVoteDayListByTime(maps, req.StartTime, req.EndTime, "voted_at") + if err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + for _, articleVoteDay := range articleVoteDays { + votedAt := time.Time(*articleVoteDay.VotedAt) + date := votedAt.Format("2006-01-02") + results[date]++ + } + } + + // 视频投票数 + if req.Type == 3 { + endTime, _ := time.Parse("2006-01-02", req.EndTime) + req.EndTime = endTime.Format("2006-01-02") + + videoVoteDayDao := dao.VideoVoteDayDao{} + maps := make(map[string]interface{}) + videoVoteDays, err := videoVoteDayDao.GetVideoVoteDayListByTime(maps, req.StartTime, req.EndTime, "voted_at") + if err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + for _, videoVoteDay := range videoVoteDays { + votedAt := time.Time(*videoVoteDay.VotedAt) + date := votedAt.Format("2006-01-02") + results[date]++ + } + } + + for k, v := range results { + response := &dto.IndexDataDto{ + Date: k, + Count: v, + } + + g = append(g, response) + } + + responses.OkWithData(g, c) +} diff --git a/api/controller/System.go b/api/controller/System.go new file mode 100644 index 0000000..826d92a --- /dev/null +++ b/api/controller/System.go @@ -0,0 +1,153 @@ +package controller + +import ( + "github.com/gin-gonic/gin" + "strconv" + "time" + "vote-admin-video-api/api/dao" + "vote-admin-video-api/api/dto" + "vote-admin-video-api/api/model" + "vote-admin-video-api/api/requests" + "vote-admin-video-api/api/responses" + "vote-admin-video-api/global" + "vote-admin-video-api/utils" +) + +type System struct{} + +// GetSystemTime 获取投票时间详情 +func (r *System) GetSystemTime(c *gin.Context) { + id := c.Param("system_time_id") + if id == "" { + responses.FailWithMessage("缺少参数", c) + return + } + + // 将 id 转换为 int64 类型 + systemTimeId, err := strconv.ParseInt(id, 10, 64) + if err != nil { + responses.Fail(c) + return + } + + // 获取数据 + systemTimeDao := dao.SystemTimeDao{} + systemTime, err := systemTimeDao.GetSystemTimeById(systemTimeId) + if err != nil { + responses.FailWithMessage("数据异常", c) + return + } + + // 处理返回值 + g := dto.GetSystemTimeDto(systemTime) + + responses.OkWithData(g, c) +} + +// PutSystemTime 修改投票时间 +func (b *System) PutSystemTime(c *gin.Context) { + systemRequest := requests.SystemRequest{} + req := systemRequest.PutSystemTime + if err := c.ShouldBind(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 参数验证 + if err := global.Validate.Struct(req); err != nil { + responses.FailWithMessage(utils.Translate(err), c) + return + } + + id := c.Param("system_time_id") + if id == "" { + responses.FailWithMessage("缺少参数", c) + return + } + + // 将 id 转换为 int64 类型 + systemTimeId, err := strconv.ParseInt(id, 10, 64) + if err != nil { + responses.Fail(c) + return + } + + // 获取数据 + systemTimeDao := dao.SystemTimeDao{} + systemTime, err := systemTimeDao.GetSystemTimeById(systemTimeId) + if err != nil { + responses.FailWithMessage("数据异常", c) + return + } + + // 修改值 + systemTimeData := make(map[string]interface{}) + + // 获取本地时区 + location, err := time.LoadLocation("Local") + if err != nil { + responses.FailWithMessage("修改失败", c) + return + } + + startTime, err := time.ParseInLocation("2006-01-02 15:04:05", req.StartTime, location) + if err != nil { + responses.FailWithMessage("修改失败", c) + return + } + + if model.LocalTime(startTime) != *systemTime.StartTime { + systemTimeData["start_time"] = req.StartTime + } + + endTime, err := time.ParseInLocation("2006-01-02 15:04:05", req.EndTime, location) + if err != nil { + responses.FailWithMessage("修改失败", c) + return + } + + if model.LocalTime(endTime) != *systemTime.EndTime { + systemTimeData["end_time"] = req.EndTime + } + + if len(systemTimeData) > 0 { + // 开始事务 + tx := global.Db.Begin() + defer func() { + if r := recover(); r != nil { + tx.Rollback() + } + }() + + err = systemTimeDao.EditSystemTimeById(tx, systemTimeId, systemTimeData) + if err != nil { + tx.Rollback() + responses.FailWithMessage("操作失败", c) + return + } + + // 增加过期时间缓存 + redisKey := "VoteSystemTime" + // 当前时间 + now := time.Now() + + duration := endTime.Sub(now) + if duration < 0 { + tx.Rollback() + responses.FailWithMessage("结束时间错误", c) + return + } + + // 添加缓存 + _, err = global.Redis.Set(c, redisKey, "1", duration).Result() + if err != nil { + tx.Rollback() + responses.FailWithMessage("修改失败", c) + return + } + + tx.Commit() + } + + responses.Ok(c) +} diff --git a/api/controller/User.go b/api/controller/User.go new file mode 100644 index 0000000..ec79e3e --- /dev/null +++ b/api/controller/User.go @@ -0,0 +1,147 @@ +package controller + +import ( + "github.com/gin-gonic/gin" + "strconv" + "vote-admin-video-api/api/dao" + "vote-admin-video-api/api/dto" + "vote-admin-video-api/api/requests" + "vote-admin-video-api/api/responses" + "vote-admin-video-api/global" + "vote-admin-video-api/utils" +) + +type User struct{} + +// GetUserPage 获取用户列表-分页 +func (r *User) GetUserPage(c *gin.Context) { + userRequest := requests.UserRequest{} + req := userRequest.GetUserPage + if err := c.ShouldBind(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 参数验证 + if err := global.Validate.Struct(req); err != nil { + responses.FailWithMessage(utils.Translate(err), c) + return + } + + if req.Page == 0 { + req.Page = 1 + } + + if req.PageSize == 0 { + req.PageSize = 20 + } + + // 获取数据 + userDao := dao.UserDao{} + user, total, err := userDao.GetUserPageSearch(req, req.Page, req.PageSize) + if err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 处理返回值 + g := dto.GetUserListDto(user) + + result := make(map[string]interface{}) + result["page"] = req.Page + result["page_size"] = req.PageSize + result["total"] = total + result["data"] = g + responses.OkWithData(result, c) +} + +// GetUser 获取用户详情 +func (r *User) GetUser(c *gin.Context) { + id := c.Param("user_id") + if id == "" { + responses.FailWithMessage("缺少参数", c) + return + } + + // 将 id 转换为 int64 类型 + userId, err := strconv.ParseInt(id, 10, 64) + if err != nil { + responses.Fail(c) + return + } + + // 获取数据 + userDao := dao.UserDao{} + user, err := userDao.GetUserById(userId) + if err != nil { + responses.FailWithMessage("用户异常", c) + return + } + + // 处理返回值 + g := dto.GetUserDto(user) + + responses.OkWithData(g, c) +} + +// PutUserStatus 操作用户状态 +func (r *User) PutUserStatus(c *gin.Context) { + userRequest := requests.UserRequest{} + req := userRequest.PutUserStatus + if err := c.ShouldBind(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 参数验证 + if err := global.Validate.Struct(req); err != nil { + responses.FailWithMessage(utils.Translate(err), c) + return + } + + id := c.Param("user_id") + if id == "" { + responses.FailWithMessage("缺少参数", c) + return + } + + // 将 id 转换为 int64 类型 + userId, err := strconv.ParseInt(id, 10, 64) + if err != nil { + responses.Fail(c) + return + } + + // 获取数据 + userDao := dao.UserDao{} + user, err := userDao.GetUserById(userId) + if err != nil { + responses.FailWithMessage("用户异常", c) + return + } + + if req.UserStatus == user.UserStatus { + responses.Ok(c) + return + } + + // 开始事务 + tx := global.Db.Begin() + defer func() { + if r := recover(); r != nil { + tx.Rollback() + } + }() + + userData := make(map[string]interface{}) + userData["user_status"] = req.UserStatus + err = userDao.EditUserById(tx, userId, userData) + if err != nil { + tx.Rollback() + responses.FailWithMessage("操作失败", c) + return + } + + tx.Commit() + responses.Ok(c) +} diff --git a/api/controller/UserVoteDay.go b/api/controller/UserVoteDay.go new file mode 100644 index 0000000..fd22f29 --- /dev/null +++ b/api/controller/UserVoteDay.go @@ -0,0 +1,262 @@ +package controller + +import ( + "fmt" + "github.com/gin-gonic/gin" + "strconv" + "time" + "vote-admin-video-api/api/dao" + "vote-admin-video-api/api/dto" + "vote-admin-video-api/api/requests" + "vote-admin-video-api/api/responses" + "vote-admin-video-api/global" + "vote-admin-video-api/utils" +) + +type UserVoteDay struct{} + +// GetUserArticleVotePage 用户投票记录列表-图文-分页 +func (r *UserVoteDay) GetUserArticleVotePage(c *gin.Context) { + userVoteDayRequest := requests.UserVoteDayRequest{} + req := userVoteDayRequest.GetArticleUserVoteDayPage + if err := c.ShouldBind(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 参数验证 + if err := global.Validate.Struct(req); err != nil { + responses.FailWithMessage(utils.Translate(err), c) + return + } + + if req.Page == 0 { + req.Page = 1 + } + + if req.PageSize == 0 { + req.PageSize = 20 + } + + userId, err := strconv.ParseInt(req.UserId, 10, 64) + if err != nil { + responses.Fail(c) + return + } + + var total int64 + + articleVoteDayDao := dao.ArticleVoteDayDao{} + articleVoteDay, total, err := articleVoteDayDao.GetArticleVoteDayByUserIdPageSearch(userId, req.Page, req.PageSize) + if err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 处理返回值 + g := make([]*dto.UserVoteDto, len(articleVoteDay)) + for i, v := range articleVoteDay { + createdAt := time.Time(v.CreatedAt).Format("2006-01-02 15:04:05") + + response := &dto.UserVoteDto{ + Id: fmt.Sprintf("%d", v.ArticleId), + Title: v.Article.ArticleTitle, + VotedAt: createdAt, + } + + // 将转换后的结构体添加到新切片中 + g[i] = response + } + + result := make(map[string]interface{}) + result["page"] = req.Page + result["page_size"] = req.PageSize + result["total"] = total + result["data"] = g + responses.OkWithData(result, c) +} + +// GetUserVideoVotePage 用户投票记录列表-视频-分页 +func (r *UserVoteDay) GetUserVideoVotePage(c *gin.Context) { + userVoteDayRequest := requests.UserVoteDayRequest{} + req := userVoteDayRequest.GetVideoUserVoteDayPage + if err := c.ShouldBind(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 参数验证 + if err := global.Validate.Struct(req); err != nil { + responses.FailWithMessage(utils.Translate(err), c) + return + } + + if req.Page == 0 { + req.Page = 1 + } + + if req.PageSize == 0 { + req.PageSize = 20 + } + + userId, err := strconv.ParseInt(req.UserId, 10, 64) + if err != nil { + responses.Fail(c) + return + } + + var total int64 + + videoVoteDayDao := dao.VideoVoteDayDao{} + videoVoteDay, total, err := videoVoteDayDao.GetVideoVoteDayByUserIdPageSearch(userId, req.Page, req.PageSize) + if err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 处理返回值 + g := make([]*dto.UserVoteDto, len(videoVoteDay)) + for i, v := range videoVoteDay { + createdAt := time.Time(v.CreatedAt).Format("2006-01-02 15:04:05") + + response := &dto.UserVoteDto{ + Id: fmt.Sprintf("%d", v.VideoId), + Title: v.Video.VideoTitle, + VotedAt: createdAt, + } + + // 将转换后的结构体添加到新切片中 + g[i] = response + } + + result := make(map[string]interface{}) + result["page"] = req.Page + result["page_size"] = req.PageSize + result["total"] = total + result["data"] = g + responses.OkWithData(result, c) +} + +// GetArticleVotePage 投票记录列表-图文-分页 +func (r *UserVoteDay) GetArticleVotePage(c *gin.Context) { + userVoteDayRequest := requests.UserVoteDayRequest{} + req := userVoteDayRequest.GetArticleVoteDayPage + if err := c.ShouldBind(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 参数验证 + if err := global.Validate.Struct(req); err != nil { + responses.FailWithMessage(utils.Translate(err), c) + return + } + + if req.Page == 0 { + req.Page = 1 + } + + if req.PageSize == 0 { + req.PageSize = 20 + } + + articleId, err := strconv.ParseInt(req.ArticleId, 10, 64) + if err != nil { + responses.Fail(c) + return + } + + var total int64 + + articleVoteDayDao := dao.ArticleVoteDayDao{} + articleVoteDay, total, err := articleVoteDayDao.GetArticleVoteDayByArticleIdPageSearch(articleId, req.Page, req.PageSize) + if err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 处理返回值 + g := make([]*dto.UserVoteDto, len(articleVoteDay)) + for i, v := range articleVoteDay { + createdAt := time.Time(v.CreatedAt).Format("2006-01-02 15:04:05") + + response := &dto.UserVoteDto{ + Id: fmt.Sprintf("%d", v.ArticleId), + AppIden: v.User.AppIden, + OpenId: v.User.OpenId, + VotedAt: createdAt, + } + + // 将转换后的结构体添加到新切片中 + g[i] = response + } + + result := make(map[string]interface{}) + result["page"] = req.Page + result["page_size"] = req.PageSize + result["total"] = total + result["data"] = g + responses.OkWithData(result, c) +} + +// GetVideoVotePage 投票记录列表-视频-分页 +func (r *UserVoteDay) GetVideoVotePage(c *gin.Context) { + userVoteDayRequest := requests.UserVoteDayRequest{} + req := userVoteDayRequest.GetVideoVoteDayPage + if err := c.ShouldBind(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 参数验证 + if err := global.Validate.Struct(req); err != nil { + responses.FailWithMessage(utils.Translate(err), c) + return + } + + if req.Page == 0 { + req.Page = 1 + } + + if req.PageSize == 0 { + req.PageSize = 20 + } + + videoId, err := strconv.ParseInt(req.VideoId, 10, 64) + if err != nil { + responses.Fail(c) + return + } + + var total int64 + + videoVoteDayDao := dao.VideoVoteDayDao{} + videoVoteDay, total, err := videoVoteDayDao.GetVideoVoteDayByVideoIdPageSearch(videoId, req.Page, req.PageSize) + if err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 处理返回值 + g := make([]*dto.UserVoteDto, len(videoVoteDay)) + for i, v := range videoVoteDay { + createdAt := time.Time(v.CreatedAt).Format("2006-01-02 15:04:05") + + response := &dto.UserVoteDto{ + Id: fmt.Sprintf("%d", v.VideoId), + AppIden: v.User.AppIden, + OpenId: v.User.OpenId, + VotedAt: createdAt, + } + + // 将转换后的结构体添加到新切片中 + g[i] = response + } + + result := make(map[string]interface{}) + result["page"] = req.Page + result["page_size"] = req.PageSize + result["total"] = total + result["data"] = g + responses.OkWithData(result, c) +} diff --git a/api/controller/Video.go b/api/controller/Video.go new file mode 100644 index 0000000..ea8ea4b --- /dev/null +++ b/api/controller/Video.go @@ -0,0 +1,437 @@ +package controller + +import ( + "fmt" + "github.com/gin-gonic/gin" + "strconv" + "vote-admin-video-api/api/dao" + "vote-admin-video-api/api/dto" + "vote-admin-video-api/api/model" + "vote-admin-video-api/api/requests" + "vote-admin-video-api/api/responses" + "vote-admin-video-api/global" + "vote-admin-video-api/utils" +) + +type Video struct{} + +// GetVideoPage 获取视频列表-分页 +func (r *Video) GetVideoPage(c *gin.Context) { + VideoRequest := requests.VideoRequest{} + req := VideoRequest.GetVideoPage + if err := c.ShouldBindJSON(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 参数验证 + if err := global.Validate.Struct(req); err != nil { + responses.FailWithMessage(utils.Translate(err), c) + return + } + + if req.Page == 0 { + req.Page = 1 + } + + if req.PageSize == 0 { + req.PageSize = 20 + } + + if req.Order != nil { + if req.Order.VoteNum != "" { + if req.Order.VoteNum != "desc" && req.Order.VoteNum != "asc" { + responses.FailWithMessage("排序字段错误", c) + return + } + } + } + + // 获取数据 + VideoDao := dao.VideoDao{} + Videos, total, err := VideoDao.GetVideoPageSearch(req, req.Page, req.PageSize) + if err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 处理返回值 + g := dto.GetVideoListDto(Videos) + + result := make(map[string]interface{}) + result["page"] = req.Page + result["page_size"] = req.PageSize + result["total"] = total + result["data"] = g + responses.OkWithData(result, c) +} + +// GetVideo 获取视频详情 +func (r *Video) GetVideo(c *gin.Context) { + id := c.Param("video_id") + if id == "" { + responses.FailWithMessage("缺少参数", c) + return + } + + // 将 id 转换为 int64 类型 + videoId, err := strconv.ParseInt(id, 10, 64) + if err != nil { + responses.Fail(c) + return + } + + // 获取数据 + VideoDao := dao.VideoDao{} + video, err := VideoDao.GetVideoById(videoId) + if err != nil { + responses.FailWithMessage("视频错误", c) + return + } + + // 获取作者数据 + VideoAuthorDao := dao.VideoAuthorDao{} + maps := make(map[string]interface{}) + maps["video_id"] = videoId + VideoAuthors, err := VideoAuthorDao.GetVideoAuthorPreloadList(maps) + if err != nil { + responses.FailWithMessage("视频错误", c) + return + } + + // 获取排名 + rank, _ := VideoDao.GetVideoRank(video.VideoId) + + // 处理返回值 + g := dto.GetVideoDto(video) + + // 加载数据-作者 + g.LoadVideoAuthor(VideoAuthors) + + // 加载数据-作者排名 + g.LoadRank(rank) + + responses.OkWithData(g, c) +} + +// PutVideo 修改视频详情 +func (r *Video) PutVideo(c *gin.Context) { + VideoRequest := requests.VideoRequest{} + req := VideoRequest.PutVideo + if err := c.ShouldBindJSON(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 参数验证 + if err := global.Validate.Struct(req); err != nil { + responses.FailWithMessage(utils.Translate(err), c) + return + } + + id := c.Param("video_id") + if id == "" { + responses.FailWithMessage("缺少参数", c) + return + } + + // 将 id 转换为 int64 类型 + videoId, err := strconv.ParseInt(id, 10, 64) + if err != nil { + responses.Fail(c) + return + } + + if req.VoteNum != nil { + if *req.VoteNum < 0 { + responses.FailWithMessage("票数需大于0", c) + return + } + } + + // 获取订单数据 + videoDao := dao.VideoDao{} + video, err := videoDao.GetVideoById(videoId) + if err != nil { + responses.FailWithMessage("视频异常", c) + return + } + + // 开始事务 + tx := global.Db.Begin() + defer func() { + if r := recover(); r != nil { + tx.Rollback() + fmt.Println(r) + } + }() + + videoData := make(map[string]interface{}) + + // 视频标题 + if video.VideoTitle != req.VideoTitle { + videoData["video_title"] = req.VideoTitle + } + + // 视频状态 + if video.VideoStatus != req.VideoStatus { + videoData["video_status"] = req.VideoStatus + } + + if req.VoteNum != nil { + if video.VoteNum != *req.VoteNum { + videoData["vote_num"] = *req.VoteNum + } + } + + // 视频编号 + if video.VideoNo != req.VideoNo { + videoData["video_no"] = req.VideoNo + } + + // 视频内容 + if video.VideoContent != req.VideoContent { + videoData["video_content"] = req.VideoContent + } + + // 修改 + if len(videoData) > 0 { + err = videoDao.EditVideoById(tx, videoId, videoData) + if err != nil { + tx.Rollback() + responses.FailWithMessage(err.Error(), c) + return + } + } + + // 删除作者 + VideoAuthorDao := dao.VideoAuthorDao{} + maps := make(map[string]interface{}) + maps["video_id"] = videoId + err = VideoAuthorDao.DeleteVideoAuthor(tx, maps) + if err != nil { + tx.Rollback() + responses.FailWithMessage(err.Error(), c) + return + } + + baseHospitalDao := dao.BaseHospitalDao{} + // 新增作者 + for _, author := range req.VideoAuthor { + if author.HospitalId == "" { + tx.Rollback() + responses.FailWithMessage("请选择作者所属医院", c) + return + } + + if author.AuthorName == "" { + tx.Rollback() + responses.FailWithMessage("请输入作者名称", c) + return + } + + // 检测医院 + hospitalId, err := strconv.ParseInt(author.HospitalId, 10, 64) + if err != nil { + tx.Rollback() + responses.FailWithMessage(err.Error(), c) + return + } + + baseHospital, err := baseHospitalDao.GetBaseHospitalById(hospitalId) + if err != nil || baseHospital == nil { + tx.Rollback() + responses.FailWithMessage("医院错误", c) + return + } + + videoAuthor := &model.VideoAuthor{ + VideoId: videoId, + AuthorName: author.AuthorName, + HospitalId: hospitalId, + } + videoAuthor, err = VideoAuthorDao.AddVideoAuthor(tx, videoAuthor) + if err != nil { + tx.Rollback() + responses.FailWithMessage(err.Error(), c) + return + } + } + + tx.Commit() + responses.Ok(c) +} + +// AddVideo 新增视频详情 +func (r *Video) AddVideo(c *gin.Context) { + videoRequest := requests.VideoRequest{} + req := videoRequest.AddVideo + if err := c.ShouldBindJSON(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 参数验证 + if err := global.Validate.Struct(req); err != nil { + responses.FailWithMessage(utils.Translate(err), c) + return + } + + if req.VoteNum != nil { + if *req.VoteNum < 0 { + responses.FailWithMessage("票数需大于0", c) + return + } + } + + // 开始事务 + tx := global.Db.Begin() + defer func() { + if r := recover(); r != nil { + tx.Rollback() + responses.FailWithMessage("操作失败", c) + return + } + }() + + videoDao := dao.VideoDao{} + + video := &model.Video{ + VideoTitle: req.VideoTitle, + VideoStatus: req.VideoStatus, + VideoNo: req.VideoNo, + VideoContent: req.VideoContent, + } + + if req.VoteNum != nil { + video.VoteNum = *req.VoteNum + } + + // 获取总数量 + maps := make(map[string]interface{}) + total, err := videoDao.GetVideoCount(maps) + if err != nil { + tx.Rollback() + responses.FailWithMessage(err.Error(), c) + return + } + + // 生成视频编号 + video.VideoNumber = fmt.Sprintf("%d", 2000+total+1) + + video, err = videoDao.AddVideo(tx, video) + if err != nil { + tx.Rollback() + responses.FailWithMessage(err.Error(), c) + return + } + + // 新增作者 + baseHospitalDao := dao.BaseHospitalDao{} + videoAuthorDao := dao.VideoAuthorDao{} + for _, author := range req.VideoAuthor { + if author.HospitalId == "" { + tx.Rollback() + responses.FailWithMessage("请选择作者所属医院", c) + return + } + + if author.AuthorName == "" { + tx.Rollback() + responses.FailWithMessage("请输入作者名称", c) + return + } + + // 检测医院 + hospitalId, err := strconv.ParseInt(author.HospitalId, 10, 64) + if err != nil { + tx.Rollback() + responses.FailWithMessage(err.Error(), c) + return + } + + baseHospital, err := baseHospitalDao.GetBaseHospitalById(hospitalId) + if err != nil || baseHospital == nil { + tx.Rollback() + responses.FailWithMessage("医院错误", c) + return + } + + videoAuthor := &model.VideoAuthor{ + VideoId: video.VideoId, + AuthorName: author.AuthorName, + HospitalId: hospitalId, + } + videoAuthor, err = videoAuthorDao.AddVideoAuthor(tx, videoAuthor) + if err != nil { + tx.Rollback() + responses.FailWithMessage(err.Error(), c) + return + } + } + + tx.Commit() + responses.Ok(c) +} + +// PutVideoStatus 操作视频状态 +func (r *Video) PutVideoStatus(c *gin.Context) { + videoRequest := requests.VideoRequest{} + req := videoRequest.PutVideoStatus + if err := c.ShouldBind(&req); err != nil { + responses.FailWithMessage(err.Error(), c) + return + } + + // 参数验证 + if err := global.Validate.Struct(req); err != nil { + responses.FailWithMessage(utils.Translate(err), c) + return + } + + id := c.Param("video_id") + if id == "" { + responses.FailWithMessage("缺少参数", c) + return + } + + // 将 id 转换为 int64 类型 + videoId, err := strconv.ParseInt(id, 10, 64) + if err != nil { + responses.Fail(c) + return + } + + // 获取订单数据 + videoDao := dao.VideoDao{} + video, err := videoDao.GetVideoById(videoId) + if err != nil { + responses.FailWithMessage("视频异常", c) + return + } + + // 开始事务 + tx := global.Db.Begin() + defer func() { + if r := recover(); r != nil { + tx.Rollback() + } + }() + + videoData := make(map[string]interface{}) + if req.VideoStatus != video.VideoStatus { + videoData["video_status"] = req.VideoStatus + } + + if len(videoData) > 0 { + err = videoDao.EditVideoById(tx, videoId, videoData) + if err != nil { + tx.Rollback() + responses.FailWithMessage("修改失败", c) + return + } + } + + tx.Commit() + responses.Ok(c) +} diff --git a/api/dao/AdminUser.go b/api/dao/AdminUser.go new file mode 100644 index 0000000..aba0616 --- /dev/null +++ b/api/dao/AdminUser.go @@ -0,0 +1,108 @@ +package dao + +import ( + "gorm.io/gorm" + "gorm.io/gorm/clause" + "vote-admin-video-api/api/model" + "vote-admin-video-api/global" +) + +type AdminUserDao struct { +} + +// GetAdminUserById 获取数据-id +func (r *AdminUserDao) GetAdminUserById(AdminUserId int64) (m *model.AdminUser, err error) { + err = global.Db.First(&m, AdminUserId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetAdminUserPreloadById 获取数据-加载全部关联-id +func (r *AdminUserDao) GetAdminUserPreloadById(AdminUserId int64) (m *model.AdminUser, err error) { + err = global.Db.Preload(clause.Associations).First(&m, AdminUserId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// DeleteAdminUser 删除 +func (r *AdminUserDao) DeleteAdminUser(tx *gorm.DB, maps interface{}) error { + err := tx.Where(maps).Delete(&model.AdminUser{}).Error + if err != nil { + return err + } + return nil +} + +// DeleteAdminUserById 删除-id +func (r *AdminUserDao) DeleteAdminUserById(tx *gorm.DB, AdminUserId int64) error { + if err := tx.Delete(&model.AdminUser{}, AdminUserId).Error; err != nil { + return err + } + return nil +} + +// EditAdminUser 修改 +func (r *AdminUserDao) EditAdminUser(tx *gorm.DB, maps interface{}, data interface{}) error { + err := tx.Model(&model.AdminUser{}).Where(maps).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// EditAdminUserById 修改-id +func (r *AdminUserDao) EditAdminUserById(tx *gorm.DB, AdminUserId int64, data interface{}) error { + err := tx.Model(&model.AdminUser{}).Where("AdminUser_id = ?", AdminUserId).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// GetAdminUserList 获取列表 +func (r *AdminUserDao) GetAdminUserList(maps interface{}) (m []*model.AdminUser, err error) { + err = global.Db.Where(maps).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetAdminUserCount 获取数量 +func (r *AdminUserDao) GetAdminUserCount(maps interface{}) (total int64, err error) { + err = global.Db.Model(&model.AdminUser{}).Where(maps).Count(&total).Error + if err != nil { + return 0, err + } + return total, nil +} + +// GetAdminUserListRand 获取列表-随机 +func (r *AdminUserDao) GetAdminUserListRand(maps interface{}, limit int) (m []*model.AdminUser, err error) { + err = global.Db.Where(maps).Order("rand()").Limit(limit).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// AddAdminUser 新增 +func (r *AdminUserDao) AddAdminUser(tx *gorm.DB, model *model.AdminUser) (*model.AdminUser, error) { + if err := tx.Create(model).Error; err != nil { + return nil, err + } + return model, nil +} + +// GetAdminUser 获取 +func (r *AdminUserDao) GetAdminUser(maps interface{}) (m *model.AdminUser, err error) { + err = global.Db.Where(maps).First(&m).Error + if err != nil { + return nil, err + } + return m, nil +} diff --git a/api/dao/Article.go b/api/dao/Article.go new file mode 100644 index 0000000..465e1b7 --- /dev/null +++ b/api/dao/Article.go @@ -0,0 +1,228 @@ +package dao + +import ( + "gorm.io/gorm" + "gorm.io/gorm/clause" + "vote-admin-video-api/api/model" + "vote-admin-video-api/api/requests" + "vote-admin-video-api/global" +) + +type ArticleDao struct { +} + +// GetArticleById 获取数据-id +func (r *ArticleDao) GetArticleById(ArticleId int64) (m *model.Article, err error) { + err = global.Db.First(&m, ArticleId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetArticlePreloadById 获取数据-加载全部关联-id +func (r *ArticleDao) GetArticlePreloadById(ArticleId int64) (m *model.Article, err error) { + err = global.Db.Preload(clause.Associations).First(&m, ArticleId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// DeleteArticle 删除 +func (r *ArticleDao) DeleteArticle(tx *gorm.DB, maps interface{}) error { + err := tx.Where(maps).Delete(&model.Article{}).Error + if err != nil { + return err + } + return nil +} + +// DeleteArticleById 删除-id +func (r *ArticleDao) DeleteArticleById(tx *gorm.DB, ArticleId int64) error { + if err := tx.Delete(&model.Article{}, ArticleId).Error; err != nil { + return err + } + return nil +} + +// EditArticle 修改 +func (r *ArticleDao) EditArticle(tx *gorm.DB, maps interface{}, data interface{}) error { + err := tx.Model(&model.Article{}).Where(maps).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// EditArticleById 修改-id +func (r *ArticleDao) EditArticleById(tx *gorm.DB, ArticleId int64, data interface{}) error { + err := tx.Model(&model.Article{}).Where("article_id = ?", ArticleId).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// GetArticleList 获取列表 +func (r *ArticleDao) GetArticleList(maps interface{}) (m []*model.Article, err error) { + err = global.Db.Where(maps).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetArticleCount 获取数量 +func (r *ArticleDao) GetArticleCount(maps interface{}) (total int64, err error) { + err = global.Db.Model(&model.Article{}).Where(maps).Count(&total).Error + if err != nil { + return 0, err + } + return total, nil +} + +// GetArticleListRand 获取列表-随机 +func (r *ArticleDao) GetArticleListRand(maps interface{}, limit int) (m []*model.Article, err error) { + err = global.Db.Where(maps).Order("rand()").Limit(limit).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// AddArticle 新增 +func (r *ArticleDao) AddArticle(tx *gorm.DB, model *model.Article) (*model.Article, error) { + if err := tx.Create(model).Error; err != nil { + return nil, err + } + return model, nil +} + +// GetArticle 获取 +func (r *ArticleDao) GetArticle(maps interface{}) (m *model.Article, err error) { + err = global.Db.Where(maps).First(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// Inc 自增 +func (r *ArticleDao) Inc(tx *gorm.DB, ArticleId int64, field string, numeral int) error { + err := tx.Model(&model.Article{}).Where("article_id = ?", ArticleId).UpdateColumn(field, gorm.Expr(field+" + ?", numeral)).Error + if err != nil { + return err + } + return nil +} + +// Dec 自减 +func (r *ArticleDao) Dec(tx *gorm.DB, ArticleId int64, field string, numeral int) error { + err := tx.Model(&model.Article{}).Where("article_id = ?", ArticleId).UpdateColumn(field, gorm.Expr(field+" - ?", numeral)).Error + if err != nil { + return err + } + return nil +} + +// GetArticlePageSearch 获取图文列表-分页 +func (r *ArticleDao) GetArticlePageSearch(req requests.GetArticlePage, page, pageSize int) (m []*model.Article, total int64, err error) { + var totalRecords int64 + + // 构建查询条件 + query := global.Db.Model(&model.Article{}) + + // 作者 + query = query.Preload("ArticleAuthor") + + // 作者关联医院 + query = query.Preload("ArticleAuthor.BaseHospital") + + // 搜索关键字 + if req.Keyword != "" { + keyword := "%" + req.Keyword + "%" // + + // 标题 + orQuery := global.Db.Model(&model.Article{}).Or("article_title LIKE ?", keyword) + + // 医院名称 + hospitalSubQuery := global.Db.Model(&model.BaseHospital{}). + Select("hospital_id"). + Where("hospital_name LIKE ?", keyword) + + articleAuthorSubQuery := global.Db.Model(&model.ArticleAuthor{}). + Select("article_id"). + Where(gorm.Expr("hospital_id IN (?)", hospitalSubQuery)) + + orQuery = orQuery.Or(gorm.Expr("article_id IN (?)", articleAuthorSubQuery)) + + // 作者姓名 + subQuery := global.Db.Model(&model.ArticleAuthor{}). + Select("article_id"). + Where("author_name LIKE ?", keyword) + + orQuery = orQuery.Or(gorm.Expr("article_id IN (?)", subQuery)) + + // 执行组建 + query = query.Where(orQuery) + } + + // 文章状态 + if req.ArticleStatus != nil { + query = query.Where("article_status = ?", req.ArticleStatus) + } + + // 排序 + if req.Order != nil { + if req.Order.VoteNum != "" { + query = query.Order("vote_num " + req.Order.VoteNum) + } + } + + // 排序 + query = query.Order("created_at desc") + + // 查询总数量 + if err := query.Count(&totalRecords).Error; err != nil { + return nil, 0, err + } + + err = query.Scopes(model.Paginate(page, pageSize)).Find(&m).Error + if err != nil { + return nil, 0, err + } + return m, totalRecords, nil +} + +// GetArticleOrderList 获取列表-排序 +func (r *ArticleDao) GetArticleOrderList(maps interface{}, orderField string, limit int) (m []*model.Article, err error) { + err = global.Db.Where(maps).Preload(clause.Associations).Order(orderField).Limit(limit).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetArticleRank 获取某一条数据的排名 +func (r *ArticleDao) GetArticleRank(articleID int64) (int, error) { + var rank int + + // 定义子查询 + subQuery := global.Db.Model(&model.Article{}). + Select("article_id, vote_num, (@rank := @rank + 1) AS rank"). + Where("article_status = ?", 1). + Order("vote_num DESC"). + Joins(", (SELECT @rank := 0) AS r") + + // 将子查询作为命名子查询的一部分进行查询 + err := global.Db.Table("(?) AS sub", subQuery). + Where("sub.article_id = ?", articleID). + Pluck("sub.rank", &rank).Error + + if err != nil { + return 0, err + } + + return rank, nil +} diff --git a/api/dao/ArticleAuthor.go b/api/dao/ArticleAuthor.go new file mode 100644 index 0000000..6f8a76f --- /dev/null +++ b/api/dao/ArticleAuthor.go @@ -0,0 +1,135 @@ +package dao + +import ( + "gorm.io/gorm" + "gorm.io/gorm/clause" + "vote-admin-video-api/api/model" + "vote-admin-video-api/global" +) + +type ArticleAuthorDao struct { +} + +// GetArticleAuthorById 获取数据-id +func (r *ArticleAuthorDao) GetArticleAuthorById(AuthorId int64) (m *model.ArticleAuthor, err error) { + err = global.Db.First(&m, AuthorId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetArticleAuthorPreloadById 获取数据-加载全部关联-id +func (r *ArticleAuthorDao) GetArticleAuthorPreloadById(AuthorId int64) (m *model.ArticleAuthor, err error) { + err = global.Db.Preload(clause.Associations).First(&m, AuthorId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// DeleteArticleAuthor 删除 +func (r *ArticleAuthorDao) DeleteArticleAuthor(tx *gorm.DB, maps interface{}) error { + err := tx.Where(maps).Delete(&model.ArticleAuthor{}).Error + if err != nil { + return err + } + return nil +} + +// DeleteArticleAuthorById 删除-id +func (r *ArticleAuthorDao) DeleteArticleAuthorById(tx *gorm.DB, ArticleAuthorId int64) error { + if err := tx.Delete(&model.ArticleAuthor{}, ArticleAuthorId).Error; err != nil { + return err + } + return nil +} + +// EditArticleAuthor 修改 +func (r *ArticleAuthorDao) EditArticleAuthor(tx *gorm.DB, maps interface{}, data interface{}) error { + err := tx.Model(&model.ArticleAuthor{}).Where(maps).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// EditArticleAuthorById 修改-id +func (r *ArticleAuthorDao) EditArticleAuthorById(tx *gorm.DB, AuthorId int64, data interface{}) error { + err := tx.Model(&model.ArticleAuthor{}).Where("author_id = ?", AuthorId).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// GetArticleAuthorList 获取列表 +func (r *ArticleAuthorDao) GetArticleAuthorList(maps interface{}) (m []*model.ArticleAuthor, err error) { + err = global.Db.Where(maps).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetArticleAuthorPreloadList 获取列表-加载全部关联 +func (r *ArticleAuthorDao) GetArticleAuthorPreloadList(maps interface{}) (m []*model.ArticleAuthor, err error) { + err = global.Db.Preload(clause.Associations).Where(maps).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetArticleAuthorCount 获取数量 +func (r *ArticleAuthorDao) GetArticleAuthorCount(maps interface{}) (total int64, err error) { + err = global.Db.Model(&model.ArticleAuthor{}).Where(maps).Count(&total).Error + if err != nil { + return 0, err + } + return total, nil +} + +// GetArticleAuthorListRand 获取列表-随机 +func (r *ArticleAuthorDao) GetArticleAuthorListRand(maps interface{}, limit int) (m []*model.ArticleAuthor, err error) { + err = global.Db.Where(maps).Order("rand()").Limit(limit).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// AddArticleAuthor 新增 +func (r *ArticleAuthorDao) AddArticleAuthor(tx *gorm.DB, model *model.ArticleAuthor) (*model.ArticleAuthor, error) { + if err := tx.Create(model).Error; err != nil { + return nil, err + } + return model, nil +} + +// GetArticleAuthor 获取 +func (r *ArticleAuthorDao) GetArticleAuthor(maps interface{}) (m *model.ArticleAuthor, err error) { + err = global.Db.Where(maps).First(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// Inc 自增 +func (r *ArticleAuthorDao) Inc(tx *gorm.DB, AuthorId int64, field string, numeral int) error { + err := tx.Model(&model.ArticleAuthor{}).Where("author_id = ?", AuthorId).UpdateColumn(field, gorm.Expr(field+" + ?", numeral)).Error + if err != nil { + return err + } + return nil +} + +// Dec 自减 +func (r *ArticleAuthorDao) Dec(tx *gorm.DB, AuthorId int64, field string, numeral int) error { + err := tx.Model(&model.ArticleAuthor{}).Where("author_id = ?", AuthorId).UpdateColumn(field, gorm.Expr(field+" - ?", numeral)).Error + if err != nil { + return err + } + return nil +} diff --git a/api/dao/ArticleVoteDay.go b/api/dao/ArticleVoteDay.go new file mode 100644 index 0000000..997c3b3 --- /dev/null +++ b/api/dao/ArticleVoteDay.go @@ -0,0 +1,187 @@ +package dao + +import ( + "gorm.io/gorm" + "gorm.io/gorm/clause" + "vote-admin-video-api/api/model" + "vote-admin-video-api/global" +) + +type ArticleVoteDayDao struct { +} + +// GetArticleVoteDayById 获取数据-id +func (r *ArticleVoteDayDao) GetArticleVoteDayById(voteDayId int64) (m *model.ArticleVoteDay, err error) { + err = global.Db.First(&m, voteDayId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetArticleVoteDayPreloadById 获取数据-加载全部关联-id +func (r *ArticleVoteDayDao) GetArticleVoteDayPreloadById(voteDayId int64) (m *model.ArticleVoteDay, err error) { + err = global.Db.Preload(clause.Associations).First(&m, voteDayId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// DeleteArticleVoteDay 删除 +func (r *ArticleVoteDayDao) DeleteArticleVoteDay(tx *gorm.DB, maps interface{}) error { + err := tx.Where(maps).Delete(&model.ArticleVoteDay{}).Error + if err != nil { + return err + } + return nil +} + +// DeleteArticleVoteDayById 删除-id +func (r *ArticleVoteDayDao) DeleteArticleVoteDayById(tx *gorm.DB, voteDayId int64) error { + if err := tx.Delete(&model.ArticleVoteDay{}, voteDayId).Error; err != nil { + return err + } + return nil +} + +// EditArticleVoteDay 修改 +func (r *ArticleVoteDayDao) EditArticleVoteDay(tx *gorm.DB, maps interface{}, data interface{}) error { + err := tx.Model(&model.ArticleVoteDay{}).Where(maps).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// EditArticleVoteDayById 修改-id +func (r *ArticleVoteDayDao) EditArticleVoteDayById(tx *gorm.DB, voteDayId int64, data interface{}) error { + err := tx.Model(&model.ArticleVoteDay{}).Where("vote_day_id = ?", voteDayId).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// GetArticleVoteDayList 获取列表 +func (r *ArticleVoteDayDao) GetArticleVoteDayList(maps interface{}) (m []*model.ArticleVoteDay, err error) { + err = global.Db.Where(maps).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetArticleVoteDayCount 获取数量 +func (r *ArticleVoteDayDao) GetArticleVoteDayCount(maps interface{}) (total int64, err error) { + err = global.Db.Model(&model.ArticleVoteDay{}).Where(maps).Count(&total).Error + if err != nil { + return 0, err + } + return total, nil +} + +// GetArticleVoteDayListRand 获取列表-随机 +func (r *ArticleVoteDayDao) GetArticleVoteDayListRand(maps interface{}, limit int) (m []*model.ArticleVoteDay, err error) { + err = global.Db.Where(maps).Order("rand()").Limit(limit).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// AddArticleVoteDay 新增 +func (r *ArticleVoteDayDao) AddArticleVoteDay(tx *gorm.DB, model *model.ArticleVoteDay) (*model.ArticleVoteDay, error) { + if err := tx.Create(model).Error; err != nil { + return nil, err + } + return model, nil +} + +// GetArticleVoteDay 获取 +func (r *ArticleVoteDayDao) GetArticleVoteDay(maps interface{}) (m *model.ArticleVoteDay, err error) { + err = global.Db.Where(maps).First(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// Inc 自增 +func (r *ArticleVoteDayDao) Inc(tx *gorm.DB, voteDayId int64, field string, numeral int) error { + err := tx.Model(&model.ArticleVoteDay{}).Where("vote_day_id = ?", voteDayId).UpdateColumn(field, gorm.Expr(field+" + ?", numeral)).Error + if err != nil { + return err + } + return nil +} + +// Dec 自减 +func (r *ArticleVoteDayDao) Dec(tx *gorm.DB, voteDayId int64, field string, numeral int) error { + err := tx.Model(&model.ArticleVoteDay{}).Where("vote_day_id = ?", voteDayId).UpdateColumn(field, gorm.Expr(field+" - ?", numeral)).Error + if err != nil { + return err + } + return nil +} + +// GetArticleVoteDayByUserIdPageSearch 获取列表-分页-用户id +func (r *ArticleVoteDayDao) GetArticleVoteDayByUserIdPageSearch(userId int64, page, pageSize int) (m []*model.ArticleVoteDay, total int64, err error) { + var totalRecords int64 + + // 构建查询条件 + query := global.Db.Model(&model.ArticleVoteDay{}) + + // 图文 + query = query.Preload("Article") + + query = query.Where("user_id = ?", userId) + + query = query.Order("created_at desc") + + // 查询总数量 + if err := query.Count(&totalRecords).Error; err != nil { + return nil, 0, err + } + + err = query.Scopes(model.Paginate(page, pageSize)).Find(&m).Error + if err != nil { + return nil, 0, err + } + return m, totalRecords, nil +} + +// GetArticleVoteDayByArticleIdPageSearch 获取列表-分页-文章id +func (r *ArticleVoteDayDao) GetArticleVoteDayByArticleIdPageSearch(articleId int64, page, pageSize int) (m []*model.ArticleVoteDay, total int64, err error) { + var totalRecords int64 + + // 构建查询条件 + query := global.Db.Model(&model.ArticleVoteDay{}) + + // 用户 + query = query.Preload("User") + + query = query.Where("article_id = ?", articleId) + + query = query.Order("created_at desc") + + // 查询总数量 + if err := query.Count(&totalRecords).Error; err != nil { + return nil, 0, err + } + + err = query.Scopes(model.Paginate(page, pageSize)).Find(&m).Error + if err != nil { + return nil, 0, err + } + return m, totalRecords, nil +} + +// GetArticleVoteDayListByTime 获取列表-开始时间/过期时间 +func (r *ArticleVoteDayDao) GetArticleVoteDayListByTime(maps interface{}, startTime, endTime, field string) (m []*model.ArticleVoteDay, err error) { + err = global.Db.Where(maps).Where(field+" BETWEEN ? AND ?", startTime, endTime).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} diff --git a/api/dao/BaseAgreement.go b/api/dao/BaseAgreement.go new file mode 100644 index 0000000..7d11354 --- /dev/null +++ b/api/dao/BaseAgreement.go @@ -0,0 +1,148 @@ +package dao + +import ( + "gorm.io/gorm" + "gorm.io/gorm/clause" + "vote-admin-video-api/api/model" + "vote-admin-video-api/global" +) + +type BaseAgreementDao struct { +} + +// GetBaseAgreementById 获取数据-id +func (r *BaseAgreementDao) GetBaseAgreementById(AgreementId int64) (m *model.BaseAgreement, err error) { + err = global.Db.First(&m, AgreementId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetBaseAgreementPreloadById 获取数据-加载全部关联-id +func (r *BaseAgreementDao) GetBaseAgreementPreloadById(AgreementId int64) (m *model.BaseAgreement, err error) { + err = global.Db.Preload(clause.Associations).First(&m, AgreementId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// DeleteBaseAgreement 删除 +func (r *BaseAgreementDao) DeleteBaseAgreement(tx *gorm.DB, maps interface{}) error { + err := tx.Where(maps).Delete(&model.BaseAgreement{}).Error + if err != nil { + return err + } + return nil +} + +// DeleteBaseAgreementById 删除-id +func (r *BaseAgreementDao) DeleteBaseAgreementById(tx *gorm.DB, voteDayId int64) error { + if err := tx.Delete(&model.BaseAgreement{}, voteDayId).Error; err != nil { + return err + } + return nil +} + +// EditBaseAgreement 修改 +func (r *BaseAgreementDao) EditBaseAgreement(tx *gorm.DB, maps interface{}, data interface{}) error { + err := tx.Model(&model.BaseAgreement{}).Where(maps).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// EditBaseAgreementById 修改-id +func (r *BaseAgreementDao) EditBaseAgreementById(tx *gorm.DB, AgreementId int64, data interface{}) error { + err := tx.Model(&model.BaseAgreement{}).Where("agreement_id = ?", AgreementId).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// GetBaseAgreementList 获取列表 +func (r *BaseAgreementDao) GetBaseAgreementList(maps interface{}) (m []*model.BaseAgreement, err error) { + err = global.Db.Where(maps).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetBaseAgreementCount 获取数量 +func (r *BaseAgreementDao) GetBaseAgreementCount(maps interface{}) (total int64, err error) { + err = global.Db.Model(&model.BaseAgreement{}).Where(maps).Count(&total).Error + if err != nil { + return 0, err + } + return total, nil +} + +// GetBaseAgreementListRand 获取列表-随机 +func (r *BaseAgreementDao) GetBaseAgreementListRand(maps interface{}, limit int) (m []*model.BaseAgreement, err error) { + err = global.Db.Where(maps).Order("rand()").Limit(limit).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// AddBaseAgreement 新增 +func (r *BaseAgreementDao) AddBaseAgreement(tx *gorm.DB, model *model.BaseAgreement) (*model.BaseAgreement, error) { + if err := tx.Create(model).Error; err != nil { + return nil, err + } + return model, nil +} + +// GetBaseAgreement 获取 +func (r *BaseAgreementDao) GetBaseAgreement(maps interface{}) (m *model.BaseAgreement, err error) { + err = global.Db.Where(maps).First(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// Inc 自增 +func (r *BaseAgreementDao) Inc(tx *gorm.DB, AgreementId int64, field string, numeral int) error { + err := tx.Model(&model.BaseAgreement{}).Where("agreement_id = ?", AgreementId).UpdateColumn(field, gorm.Expr(field+" + ?", numeral)).Error + if err != nil { + return err + } + return nil +} + +// Dec 自减 +func (r *BaseAgreementDao) Dec(tx *gorm.DB, AgreementId int64, field string, numeral int) error { + err := tx.Model(&model.BaseAgreement{}).Where("agreement_id = ?", AgreementId).UpdateColumn(field, gorm.Expr(field+" - ?", numeral)).Error + if err != nil { + return err + } + return nil +} + +// GetBaseAgreementPageSearch 获取列表-分页 +func (r *BaseAgreementDao) GetBaseAgreementPageSearch(page, pageSize int) (m []*model.BaseAgreement, total int64, err error) { + var totalRecords int64 + + // 构建查询条件 + query := global.Db.Model(&model.BaseAgreement{}) + + // 排序 + query = query.Order("created_at desc") + + // 查询总数量 + if err := query.Count(&totalRecords).Error; err != nil { + return nil, 0, err + } + + err = query.Scopes(model.Paginate(page, pageSize)).Find(&m).Error + if err != nil { + return nil, 0, err + } + return m, totalRecords, nil +} diff --git a/api/dao/BaseArea.go b/api/dao/BaseArea.go new file mode 100644 index 0000000..9fafab2 --- /dev/null +++ b/api/dao/BaseArea.go @@ -0,0 +1,72 @@ +package dao + +import ( + "gorm.io/gorm" + "vote-admin-video-api/api/model" + "vote-admin-video-api/global" +) + +type BaseAreaDao struct { +} + +// GetBaseAreaById 获取地区-地区id +func (r *BaseAreaDao) GetBaseAreaById(BaseAreaId int) (m *model.BaseArea, err error) { + err = global.Db.First(&m, BaseAreaId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// DeleteBaseArea 删除地区 +func (r *BaseAreaDao) DeleteBaseArea(tx *gorm.DB, maps interface{}) error { + err := tx.Where(maps).Delete(&model.BaseArea{}).Error + if err != nil { + return err + } + return nil +} + +// EditBaseArea 修改地区 +func (r *BaseAreaDao) EditBaseArea(tx *gorm.DB, maps interface{}, data interface{}) error { + err := tx.Model(&model.BaseArea{}).Where(maps).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// EditBaseAreaById 修改地区-医生id +func (r *BaseAreaDao) EditBaseAreaById(tx *gorm.DB, BaseAreaId int, data interface{}) error { + err := tx.Model(&model.BaseArea{}).Where("BaseArea_id = ?", BaseAreaId).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// GetBaseAreaList 获取地区列表 +func (r *BaseAreaDao) GetBaseAreaList(maps interface{}) (m []*model.BaseArea, err error) { + err = global.Db.Where(maps).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// AddBaseArea 新增地区 +func (r *BaseAreaDao) AddBaseArea(tx *gorm.DB, model *model.BaseArea) (*model.BaseArea, error) { + if err := tx.Create(model).Error; err != nil { + return nil, err + } + return model, nil +} + +// AddBaseAreaByMap 新增地区-map +func (r *BaseAreaDao) AddBaseAreaByMap(tx *gorm.DB, data map[string]interface{}) (*model.BaseArea, error) { + userDoctorInfo := &model.BaseArea{} + if err := tx.Model(&model.BaseArea{}).Create(data).Error; err != nil { + return nil, err + } + return userDoctorInfo, nil +} diff --git a/api/dao/BaseHospital.go b/api/dao/BaseHospital.go new file mode 100644 index 0000000..14cd895 --- /dev/null +++ b/api/dao/BaseHospital.go @@ -0,0 +1,145 @@ +package dao + +import ( + "gorm.io/gorm" + "gorm.io/gorm/clause" + "vote-admin-video-api/api/model" + "vote-admin-video-api/api/requests" + "vote-admin-video-api/global" +) + +type BaseHospitalDao struct { +} + +// GetBaseHospitalById 获取数据-id +func (r *BaseHospitalDao) GetBaseHospitalById(HospitalId int64) (m *model.BaseHospital, err error) { + err = global.Db.First(&m, HospitalId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetBaseHospitalPreloadById 获取数据-加载全部关联-id +func (r *BaseHospitalDao) GetBaseHospitalPreloadById(HospitalId int64) (m *model.BaseHospital, err error) { + err = global.Db.Preload(clause.Associations).First(&m, HospitalId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// DeleteBaseHospital 删除 +func (r *BaseHospitalDao) DeleteBaseHospital(tx *gorm.DB, maps interface{}) error { + err := tx.Where(maps).Delete(&model.BaseHospital{}).Error + if err != nil { + return err + } + return nil +} + +// DeleteBaseHospitalById 删除-id +func (r *BaseHospitalDao) DeleteBaseHospitalById(tx *gorm.DB, HospitalId int64) error { + if err := tx.Delete(&model.BaseHospital{}, HospitalId).Error; err != nil { + return err + } + return nil +} + +// EditBaseHospital 修改 +func (r *BaseHospitalDao) EditBaseHospital(tx *gorm.DB, maps interface{}, data interface{}) error { + err := tx.Model(&model.BaseHospital{}).Where(maps).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// EditBaseHospitalById 修改-id +func (r *BaseHospitalDao) EditBaseHospitalById(tx *gorm.DB, HospitalId int64, data interface{}) error { + err := tx.Model(&model.BaseHospital{}).Where("hospital_id = ?", HospitalId).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// GetBaseHospitalList 获取列表 +func (r *BaseHospitalDao) GetBaseHospitalList(maps interface{}) (m []*model.BaseHospital, err error) { + err = global.Db.Where(maps).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetBaseHospitalCount 获取数量 +func (r *BaseHospitalDao) GetBaseHospitalCount(maps interface{}) (total int64, err error) { + err = global.Db.Model(&model.BaseHospital{}).Where(maps).Count(&total).Error + if err != nil { + return 0, err + } + return total, nil +} + +// GetBaseHospitalListRand 获取列表-随机 +func (r *BaseHospitalDao) GetBaseHospitalListRand(maps interface{}, limit int) (m []*model.BaseHospital, err error) { + err = global.Db.Where(maps).Order("rand()").Limit(limit).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// AddBaseHospital 新增 +func (r *BaseHospitalDao) AddBaseHospital(tx *gorm.DB, model *model.BaseHospital) (*model.BaseHospital, error) { + if err := tx.Create(model).Error; err != nil { + return nil, err + } + return model, nil +} + +// GetBaseHospital 获取 +func (r *BaseHospitalDao) GetBaseHospital(maps interface{}) (m *model.BaseHospital, err error) { + err = global.Db.Where(maps).First(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// Inc 自增 +func (r *BaseHospitalDao) Inc(tx *gorm.DB, HospitalId int64, field string, numeral int) error { + err := tx.Model(&model.BaseHospital{}).Where("hospital_id = ?", HospitalId).UpdateColumn(field, gorm.Expr(field+" + ?", numeral)).Error + if err != nil { + return err + } + return nil +} + +// Dec 自减 +func (r *BaseHospitalDao) Dec(tx *gorm.DB, HospitalId int64, field string, numeral int) error { + err := tx.Model(&model.BaseHospital{}).Where("hospital_id = ?", HospitalId).UpdateColumn(field, gorm.Expr(field+" - ?", numeral)).Error + if err != nil { + return err + } + return nil +} + +// GetBaseHospitalLimitByMaps 获取医院列表-限制条数 +func (r *BaseHospitalDao) GetBaseHospitalLimitByMaps(hospitalRequest requests.GetBaseHospitalList) (m []*model.BaseHospital, err error) { + result := global.Db + if hospitalRequest.HospitalName != "" { + result = result.Where("hospital_name like ?", "%"+hospitalRequest.HospitalName+"%") + } + + if hospitalRequest.HospitalLevelName != "" { + result = result.Where("hospital_level_name like ?", "%"+hospitalRequest.HospitalLevelName+"%") + } + + err = result.Limit(10).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} diff --git a/api/dao/Data.go b/api/dao/Data.go new file mode 100644 index 0000000..4f4c2c2 --- /dev/null +++ b/api/dao/Data.go @@ -0,0 +1,135 @@ +package dao + +import ( + "gorm.io/gorm" + "gorm.io/gorm/clause" + "vote-admin-video-api/api/model" + "vote-admin-video-api/global" +) + +type DataDao struct { +} + +// GetDataById 获取数据-id +func (r *DataDao) GetDataById(DataId int64) (m *model.Data, err error) { + err = global.Db.First(&m, DataId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetDataPreloadById 获取数据-加载全部关联-id +func (r *DataDao) GetDataPreloadById(DataId int64) (m *model.Data, err error) { + err = global.Db.Preload(clause.Associations).First(&m, DataId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// DeleteData 删除 +func (r *DataDao) DeleteData(tx *gorm.DB, maps interface{}) error { + err := tx.Where(maps).Delete(&model.Data{}).Error + if err != nil { + return err + } + return nil +} + +// DeleteDataById 删除-id +func (r *DataDao) DeleteDataById(tx *gorm.DB, DataId int64) error { + if err := tx.Delete(&model.Data{}, DataId).Error; err != nil { + return err + } + return nil +} + +// EditData 修改 +func (r *DataDao) EditData(tx *gorm.DB, maps interface{}, data interface{}) error { + err := tx.Model(&model.Data{}).Where(maps).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// EditDataById 修改-id +func (r *DataDao) EditDataById(tx *gorm.DB, DataId int64, data interface{}) error { + err := tx.Model(&model.Data{}).Where("data_id = ?", DataId).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// GetDataList 获取列表 +func (r *DataDao) GetDataList(maps interface{}) (m []*model.Data, err error) { + err = global.Db.Where(maps).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetDataCount 获取数量 +func (r *DataDao) GetDataCount(maps interface{}) (total int64, err error) { + err = global.Db.Model(&model.Data{}).Where(maps).Count(&total).Error + if err != nil { + return 0, err + } + return total, nil +} + +// GetDataListRand 获取列表-随机 +func (r *DataDao) GetDataListRand(maps interface{}, limit int) (m []*model.Data, err error) { + err = global.Db.Where(maps).Order("rand()").Limit(limit).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// AddData 新增 +func (r *DataDao) AddData(tx *gorm.DB, model *model.Data) (*model.Data, error) { + if err := tx.Create(model).Error; err != nil { + return nil, err + } + return model, nil +} + +// GetData 获取 +func (r *DataDao) GetData(maps interface{}) (m *model.Data, err error) { + err = global.Db.Where(maps).First(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetDataListByMemberValidTime 获取列表-今天开始时间/过期时间 +func (r *DataDao) GetDataListByMemberValidTime(maps interface{}, startTime, endTime string) (m []*model.Data, err error) { + err = global.Db.Where(maps).Where("member_expire_date BETWEEN ? AND ?", startTime, endTime).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// Inc 自增 +func (r *DataDao) Inc(tx *gorm.DB, DataId int64, field string, numeral int) error { + err := tx.Model(&model.Data{}).Where("data_id = ?", DataId).UpdateColumn(field, gorm.Expr(field+" + ?", numeral)).Error + if err != nil { + return err + } + return nil +} + +// Dec 自减 +func (r *DataDao) Dec(tx *gorm.DB, DataId int64, field string, numeral int) error { + err := tx.Model(&model.Data{}).Where("data_id = ?", DataId).UpdateColumn(field, gorm.Expr(field+" - ?", numeral)).Error + if err != nil { + return err + } + return nil +} diff --git a/api/dao/SystemTime.go b/api/dao/SystemTime.go new file mode 100644 index 0000000..43d4a89 --- /dev/null +++ b/api/dao/SystemTime.go @@ -0,0 +1,117 @@ +package dao + +import ( + "gorm.io/gorm" + "gorm.io/gorm/clause" + "vote-admin-video-api/api/model" + "vote-admin-video-api/global" +) + +type SystemTimeDao struct { +} + +// GetSystemTimeById 获取数据-id +func (r *SystemTimeDao) GetSystemTimeById(SystemTimeId int64) (m *model.SystemTime, err error) { + err = global.Db.First(&m, SystemTimeId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetSystemTimePreloadById 获取数据-加载全部关联-id +func (r *SystemTimeDao) GetSystemTimePreloadById(SystemTimeId int64) (m *model.SystemTime, err error) { + err = global.Db.Preload(clause.Associations).First(&m, SystemTimeId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// DeleteSystemTime 删除 +func (r *SystemTimeDao) DeleteSystemTime(tx *gorm.DB, maps interface{}) error { + err := tx.Where(maps).Delete(&model.SystemTime{}).Error + if err != nil { + return err + } + return nil +} + +// DeleteSystemTimeById 删除-id +func (r *SystemTimeDao) DeleteSystemTimeById(tx *gorm.DB, SystemTimeId int64) error { + if err := tx.Delete(&model.SystemTime{}, SystemTimeId).Error; err != nil { + return err + } + return nil +} + +// EditSystemTime 修改 +func (r *SystemTimeDao) EditSystemTime(tx *gorm.DB, maps interface{}, data interface{}) error { + err := tx.Model(&model.SystemTime{}).Where(maps).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// EditSystemTimeById 修改-id +func (r *SystemTimeDao) EditSystemTimeById(tx *gorm.DB, SystemTimeId int64, data interface{}) error { + err := tx.Model(&model.SystemTime{}).Where("system_time_id = ?", SystemTimeId).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// GetSystemTimeList 获取列表 +func (r *SystemTimeDao) GetSystemTimeList(maps interface{}) (m []*model.SystemTime, err error) { + err = global.Db.Where(maps).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetSystemTimeCount 获取数量 +func (r *SystemTimeDao) GetSystemTimeCount(maps interface{}) (total int64, err error) { + err = global.Db.Model(&model.SystemTime{}).Where(maps).Count(&total).Error + if err != nil { + return 0, err + } + return total, nil +} + +// GetSystemTimeListRand 获取列表-随机 +func (r *SystemTimeDao) GetSystemTimeListRand(maps interface{}, limit int) (m []*model.SystemTime, err error) { + err = global.Db.Where(maps).Order("rand()").Limit(limit).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// AddSystemTime 新增 +func (r *SystemTimeDao) AddSystemTime(tx *gorm.DB, model *model.SystemTime) (*model.SystemTime, error) { + if err := tx.Create(model).Error; err != nil { + return nil, err + } + return model, nil +} + +// Inc 自增 +func (r *SystemTimeDao) Inc(tx *gorm.DB, SystemTimeId int64, field string, numeral int) error { + err := tx.Model(&model.SystemTime{}).Where("SystemTime_id = ?", SystemTimeId).UpdateColumn(field, gorm.Expr(field+" + ?", numeral)).Error + if err != nil { + return err + } + return nil +} + +// Dec 自减 +func (r *SystemTimeDao) Dec(tx *gorm.DB, SystemTimeId int64, field string, numeral int) error { + err := tx.Model(&model.SystemTime{}).Where("SystemTime_id = ?", SystemTimeId).UpdateColumn(field, gorm.Expr(field+" - ?", numeral)).Error + if err != nil { + return err + } + return nil +} diff --git a/api/dao/User.go b/api/dao/User.go new file mode 100644 index 0000000..2a8782d --- /dev/null +++ b/api/dao/User.go @@ -0,0 +1,177 @@ +package dao + +import ( + "gorm.io/gorm" + "gorm.io/gorm/clause" + "vote-admin-video-api/api/model" + "vote-admin-video-api/api/requests" + "vote-admin-video-api/global" +) + +type UserDao struct { +} + +// GetUserById 获取数据-id +func (r *UserDao) GetUserById(UserId int64) (m *model.User, err error) { + err = global.Db.First(&m, UserId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetUserPreloadById 获取数据-加载全部关联-id +func (r *UserDao) GetUserPreloadById(UserId int64) (m *model.User, err error) { + err = global.Db.Preload(clause.Associations).First(&m, UserId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// DeleteUser 删除 +func (r *UserDao) DeleteUser(tx *gorm.DB, maps interface{}) error { + err := tx.Where(maps).Delete(&model.User{}).Error + if err != nil { + return err + } + return nil +} + +// DeleteUserById 删除-id +func (r *UserDao) DeleteUserById(tx *gorm.DB, UserId int64) error { + if err := tx.Delete(&model.User{}, UserId).Error; err != nil { + return err + } + return nil +} + +// EditUser 修改 +func (r *UserDao) EditUser(tx *gorm.DB, maps interface{}, data interface{}) error { + err := tx.Model(&model.User{}).Where(maps).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// EditUserById 修改-id +func (r *UserDao) EditUserById(tx *gorm.DB, UserId int64, data interface{}) error { + err := tx.Model(&model.User{}).Where("User_id = ?", UserId).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// GetUserList 获取列表 +func (r *UserDao) GetUserList(maps interface{}) (m []*model.User, err error) { + err = global.Db.Where(maps).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetUserCount 获取数量 +func (r *UserDao) GetUserCount(maps interface{}) (total int64, err error) { + err = global.Db.Model(&model.User{}).Where(maps).Count(&total).Error + if err != nil { + return 0, err + } + return total, nil +} + +// GetUserListRand 获取列表-随机 +func (r *UserDao) GetUserListRand(maps interface{}, limit int) (m []*model.User, err error) { + err = global.Db.Where(maps).Order("rand()").Limit(limit).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// AddUser 新增 +func (r *UserDao) AddUser(tx *gorm.DB, model *model.User) (*model.User, error) { + if err := tx.Create(model).Error; err != nil { + return nil, err + } + return model, nil +} + +// GetUser 获取 +func (r *UserDao) GetUser(maps interface{}) (m *model.User, err error) { + err = global.Db.Where(maps).First(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// Inc 自增 +func (r *UserDao) Inc(tx *gorm.DB, UserId int64, field string, numeral int) error { + err := tx.Model(&model.User{}).Where("User_id = ?", UserId).UpdateColumn(field, gorm.Expr(field+" + ?", numeral)).Error + if err != nil { + return err + } + return nil +} + +// Dec 自减 +func (r *UserDao) Dec(tx *gorm.DB, UserId int64, field string, numeral int) error { + err := tx.Model(&model.User{}).Where("User_id = ?", UserId).UpdateColumn(field, gorm.Expr(field+" - ?", numeral)).Error + if err != nil { + return err + } + return nil +} + +// GetUserPageSearch 获取列表-分页 +func (r *UserDao) GetUserPageSearch(req requests.GetUserPage, page, pageSize int) (m []*model.User, total int64, err error) { + var totalRecords int64 + + // 构建查询条件 + query := global.Db.Model(&model.User{}) + + // 用户id + if req.UserId != "" { + query = query.Where("user_id = ?", req.UserId) + } + + // app唯一标识 + if req.AppIden != "" { + query = query.Where("app_iden = ?", req.AppIden) + } + + // 状态 + if req.UserStatus != 0 { + query = query.Where("user_status = ?", req.UserStatus) + } + + // 用户微信标识 + if req.OpenId != "" { + query = query.Where("open_id LIKE ?", "%"+req.OpenId+"%") + } + + query = query.Order("created_at desc") + + // 查询总数量 + if err := query.Count(&totalRecords).Error; err != nil { + return nil, 0, err + } + + err = query.Scopes(model.Paginate(page, pageSize)).Find(&m).Error + if err != nil { + return nil, 0, err + } + return m, totalRecords, nil +} + +// GetUserListByTime 获取列表-今天开始时间/过期时间 +func (r *UserDao) GetUserListByTime(maps interface{}, startTime, endTime, field string) (m []*model.User, err error) { + err = global.Db.Where(maps).Where(field+" BETWEEN ? AND ?", startTime, endTime).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} diff --git a/api/dao/Video.go b/api/dao/Video.go new file mode 100644 index 0000000..aec6d17 --- /dev/null +++ b/api/dao/Video.go @@ -0,0 +1,233 @@ +package dao + +import ( + "gorm.io/gorm" + "gorm.io/gorm/clause" + "vote-admin-video-api/api/model" + "vote-admin-video-api/api/requests" + "vote-admin-video-api/global" +) + +type VideoDao struct { +} + +// GetVideoById 获取数据-id +func (r *VideoDao) GetVideoById(VideoId int64) (m *model.Video, err error) { + err = global.Db.First(&m, VideoId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetVideoPreloadById 获取数据-加载全部关联-id +func (r *VideoDao) GetVideoPreloadById(VideoId int64) (m *model.Video, err error) { + err = global.Db.Preload(clause.Associations).First(&m, VideoId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// DeleteVideo 删除 +func (r *VideoDao) DeleteVideo(tx *gorm.DB, maps interface{}) error { + err := tx.Where(maps).Delete(&model.Video{}).Error + if err != nil { + return err + } + return nil +} + +// DeleteVideoById 删除-id +func (r *VideoDao) DeleteVideoById(tx *gorm.DB, VideoId int64) error { + if err := tx.Delete(&model.Video{}, VideoId).Error; err != nil { + return err + } + return nil +} + +// EditVideo 修改 +func (r *VideoDao) EditVideo(tx *gorm.DB, maps interface{}, data interface{}) error { + err := tx.Model(&model.Video{}).Where(maps).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// EditVideoById 修改-id +func (r *VideoDao) EditVideoById(tx *gorm.DB, VideoId int64, data interface{}) error { + err := tx.Model(&model.Video{}).Where("video_id = ?", VideoId).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// GetVideoList 获取列表 +func (r *VideoDao) GetVideoList(maps interface{}) (m []*model.Video, err error) { + err = global.Db.Where(maps).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetVideoCount 获取数量 +func (r *VideoDao) GetVideoCount(maps interface{}) (total int64, err error) { + err = global.Db.Model(&model.Video{}).Where(maps).Count(&total).Error + if err != nil { + return 0, err + } + return total, nil +} + +// GetVideoListRand 获取列表-随机 +func (r *VideoDao) GetVideoListRand(maps interface{}, limit int) (m []*model.Video, err error) { + err = global.Db.Where(maps).Order("rand()").Limit(limit).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// AddVideo 新增 +func (r *VideoDao) AddVideo(tx *gorm.DB, model *model.Video) (*model.Video, error) { + if err := tx.Create(model).Error; err != nil { + return nil, err + } + return model, nil +} + +// GetVideo 获取 +func (r *VideoDao) GetVideo(maps interface{}) (m *model.Video, err error) { + err = global.Db.Where(maps).First(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// Inc 自增 +func (r *VideoDao) Inc(tx *gorm.DB, VideoId int64, field string, numeral int) error { + err := tx.Model(&model.Video{}).Where("video_id = ?", VideoId).UpdateColumn(field, gorm.Expr(field+" + ?", numeral)).Error + if err != nil { + return err + } + return nil +} + +// Dec 自减 +func (r *VideoDao) Dec(tx *gorm.DB, VideoId int64, field string, numeral int) error { + err := tx.Model(&model.Video{}).Where("video_id = ?", VideoId).UpdateColumn(field, gorm.Expr(field+" - ?", numeral)).Error + if err != nil { + return err + } + return nil +} + +// GetVideoPageSearch 获取列表-分页 +func (r *VideoDao) GetVideoPageSearch(req requests.GetVideoPage, page, pageSize int) (m []*model.Video, total int64, err error) { + var totalRecords int64 + + // 构建查询条件 + query := global.Db.Model(&model.Video{}) + + // 作者 + query = query.Preload("VideoAuthor") + + // 作者关联医院 + query = query.Preload("VideoAuthor.BaseHospital") + + // 视频编号 + if req.VideoNo != "" { + query = query.Where("video_no LIKE ?", "%"+req.VideoNo+"%") + } + + // 搜索关键字 + if req.Keyword != "" { + keyword := "%" + req.Keyword + "%" // + + // 标题 + orQuery := global.Db.Model(&model.Video{}).Or("video_title LIKE ?", keyword) + + // 医院名称 + hospitalSubQuery := global.Db.Model(&model.BaseHospital{}). + Select("hospital_id"). + Where("hospital_name LIKE ?", keyword) + + articleAuthorSubQuery := global.Db.Model(&model.VideoAuthor{}). + Select("video_id"). + Where(gorm.Expr("hospital_id IN (?)", hospitalSubQuery)) + + orQuery = orQuery.Or(gorm.Expr("video_id IN (?)", articleAuthorSubQuery)) + + // 作者姓名 + subQuery := global.Db.Model(&model.VideoAuthor{}). + Select("video_id"). + Where("author_name LIKE ?", keyword) + + orQuery = orQuery.Or(gorm.Expr("video_id IN (?)", subQuery)) + + // 执行组建 + query = query.Where(orQuery) + } + + // 文章状态 + if req.VideoStatus != nil { + query = query.Where("video_status = ?", req.VideoStatus) + } + + // 排序 + if req.Order != nil { + if req.Order.VoteNum != "" { + query = query.Order("vote_num " + req.Order.VoteNum) + } + } + + // 排序 + query = query.Order("created_at desc") + + // 查询总数量 + if err := query.Count(&totalRecords).Error; err != nil { + return nil, 0, err + } + + err = query.Scopes(model.Paginate(page, pageSize)).Find(&m).Error + if err != nil { + return nil, 0, err + } + return m, totalRecords, nil +} + +// GetVideoOrderList 获取列表-排序 +func (r *VideoDao) GetVideoOrderList(maps interface{}, orderField string, limit int) (m []*model.Video, err error) { + err = global.Db.Where(maps).Preload(clause.Associations).Order(orderField).Limit(limit).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetVideoRank 获取某一条数据的排名 +func (r *VideoDao) GetVideoRank(videoID int64) (int, error) { + var rank int + + // 定义子查询 + subQuery := global.Db.Model(&model.Video{}). + Select("video_id, vote_num, (@rank := @rank + 1) AS rank"). + Where("video_status = ?", 1). + Order("vote_num DESC"). + Joins(", (SELECT @rank := 0) AS r") + + // 将子查询作为命名子查询的一部分进行查询 + err := global.Db.Table("(?) AS sub", subQuery). + Where("sub.video_id = ?", videoID). + Pluck("sub.rank", &rank).Error + + if err != nil { + return 0, err + } + + return rank, nil +} diff --git a/api/dao/VideoAuthor.go b/api/dao/VideoAuthor.go new file mode 100644 index 0000000..811c81f --- /dev/null +++ b/api/dao/VideoAuthor.go @@ -0,0 +1,135 @@ +package dao + +import ( + "gorm.io/gorm" + "gorm.io/gorm/clause" + "vote-admin-video-api/api/model" + "vote-admin-video-api/global" +) + +type VideoAuthorDao struct { +} + +// GetVideoAuthorById 获取数据-id +func (r *VideoAuthorDao) GetVideoAuthorById(authorId int64) (m *model.VideoAuthor, err error) { + err = global.Db.First(&m, authorId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetVideoAuthorPreloadById 获取数据-加载全部关联-id +func (r *VideoAuthorDao) GetVideoAuthorPreloadById(authorId int64) (m *model.VideoAuthor, err error) { + err = global.Db.Preload(clause.Associations).First(&m, authorId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// DeleteVideoAuthor 删除 +func (r *VideoAuthorDao) DeleteVideoAuthor(tx *gorm.DB, maps interface{}) error { + err := tx.Where(maps).Delete(&model.VideoAuthor{}).Error + if err != nil { + return err + } + return nil +} + +// DeleteVideoAuthorById 删除-id +func (r *VideoAuthorDao) DeleteVideoAuthorById(tx *gorm.DB, authorId int64) error { + if err := tx.Delete(&model.VideoAuthor{}, authorId).Error; err != nil { + return err + } + return nil +} + +// EditVideoAuthor 修改 +func (r *VideoAuthorDao) EditVideoAuthor(tx *gorm.DB, maps interface{}, data interface{}) error { + err := tx.Model(&model.VideoAuthor{}).Where(maps).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// EditVideoAuthorById 修改-id +func (r *VideoAuthorDao) EditVideoAuthorById(tx *gorm.DB, authorId int64, data interface{}) error { + err := tx.Model(&model.VideoAuthor{}).Where("authorId = ?", authorId).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// GetVideoAuthorList 获取列表 +func (r *VideoAuthorDao) GetVideoAuthorList(maps interface{}) (m []*model.VideoAuthor, err error) { + err = global.Db.Where(maps).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetVideoAuthorPreloadList 获取列表-加载全部关联 +func (r *VideoAuthorDao) GetVideoAuthorPreloadList(maps interface{}) (m []*model.VideoAuthor, err error) { + err = global.Db.Preload(clause.Associations).Where(maps).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetVideoAuthorCount 获取数量 +func (r *VideoAuthorDao) GetVideoAuthorCount(maps interface{}) (total int64, err error) { + err = global.Db.Model(&model.VideoAuthor{}).Where(maps).Count(&total).Error + if err != nil { + return 0, err + } + return total, nil +} + +// GetVideoAuthorListRand 获取列表-随机 +func (r *VideoAuthorDao) GetVideoAuthorListRand(maps interface{}, limit int) (m []*model.VideoAuthor, err error) { + err = global.Db.Where(maps).Order("rand()").Limit(limit).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// AddVideoAuthor 新增 +func (r *VideoAuthorDao) AddVideoAuthor(tx *gorm.DB, model *model.VideoAuthor) (*model.VideoAuthor, error) { + if err := tx.Create(model).Error; err != nil { + return nil, err + } + return model, nil +} + +// GetVideoAuthor 获取 +func (r *VideoAuthorDao) GetVideoAuthor(maps interface{}) (m *model.VideoAuthor, err error) { + err = global.Db.Where(maps).First(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// Inc 自增 +func (r *VideoAuthorDao) Inc(tx *gorm.DB, authorId int64, field string, numeral int) error { + err := tx.Model(&model.VideoAuthor{}).Where("authorId = ?", authorId).UpdateColumn(field, gorm.Expr(field+" + ?", numeral)).Error + if err != nil { + return err + } + return nil +} + +// Dec 自减 +func (r *VideoAuthorDao) Dec(tx *gorm.DB, authorId int64, field string, numeral int) error { + err := tx.Model(&model.VideoAuthor{}).Where("authorId = ?", authorId).UpdateColumn(field, gorm.Expr(field+" - ?", numeral)).Error + if err != nil { + return err + } + return nil +} diff --git a/api/dao/VideoVoteDay.go b/api/dao/VideoVoteDay.go new file mode 100644 index 0000000..6ab65ec --- /dev/null +++ b/api/dao/VideoVoteDay.go @@ -0,0 +1,186 @@ +package dao + +import ( + "gorm.io/gorm" + "gorm.io/gorm/clause" + "vote-admin-video-api/api/model" + "vote-admin-video-api/global" +) + +type VideoVoteDayDao struct { +} + +// GetVideoVoteDayById 获取数据-id +func (r *VideoVoteDayDao) GetVideoVoteDayById(voteDayId int64) (m *model.VideoVoteDay, err error) { + err = global.Db.First(&m, voteDayId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetVideoVoteDayPreloadById 获取数据-加载全部关联-id +func (r *VideoVoteDayDao) GetVideoVoteDayPreloadById(voteDayId int64) (m *model.VideoVoteDay, err error) { + err = global.Db.Preload(clause.Associations).First(&m, voteDayId).Error + if err != nil { + return nil, err + } + return m, nil +} + +// DeleteVideoVoteDay 删除 +func (r *VideoVoteDayDao) DeleteVideoVoteDay(tx *gorm.DB, maps interface{}) error { + err := tx.Where(maps).Delete(&model.VideoVoteDay{}).Error + if err != nil { + return err + } + return nil +} + +// DeleteVideoVoteDayById 删除-id +func (r *VideoVoteDayDao) DeleteVideoVoteDayById(tx *gorm.DB, authorId int64) error { + if err := tx.Delete(&model.VideoVoteDay{}, authorId).Error; err != nil { + return err + } + return nil +} + +// EditVideoVoteDay 修改 +func (r *VideoVoteDayDao) EditVideoVoteDay(tx *gorm.DB, maps interface{}, data interface{}) error { + err := tx.Model(&model.VideoVoteDay{}).Where(maps).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// EditVideoVoteDayById 修改-id +func (r *VideoVoteDayDao) EditVideoVoteDayById(tx *gorm.DB, voteDayId int64, data interface{}) error { + err := tx.Model(&model.VideoVoteDay{}).Where("vote_day_id = ?", voteDayId).Updates(data).Error + if err != nil { + return err + } + return nil +} + +// GetVideoVoteDayList 获取列表 +func (r *VideoVoteDayDao) GetVideoVoteDayList(maps interface{}) (m []*model.VideoVoteDay, err error) { + err = global.Db.Where(maps).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// GetVideoVoteDayCount 获取数量 +func (r *VideoVoteDayDao) GetVideoVoteDayCount(maps interface{}) (total int64, err error) { + err = global.Db.Model(&model.VideoVoteDay{}).Where(maps).Count(&total).Error + if err != nil { + return 0, err + } + return total, nil +} + +// GetVideoVoteDayListRand 获取列表-随机 +func (r *VideoVoteDayDao) GetVideoVoteDayListRand(maps interface{}, limit int) (m []*model.VideoVoteDay, err error) { + err = global.Db.Where(maps).Order("rand()").Limit(limit).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// AddVideoVoteDay 新增 +func (r *VideoVoteDayDao) AddVideoVoteDay(tx *gorm.DB, model *model.VideoVoteDay) (*model.VideoVoteDay, error) { + if err := tx.Create(model).Error; err != nil { + return nil, err + } + return model, nil +} + +// GetVideoVoteDay 获取 +func (r *VideoVoteDayDao) GetVideoVoteDay(maps interface{}) (m *model.VideoVoteDay, err error) { + err = global.Db.Where(maps).First(&m).Error + if err != nil { + return nil, err + } + return m, nil +} + +// Inc 自增 +func (r *VideoVoteDayDao) Inc(tx *gorm.DB, voteDayId int64, field string, numeral int) error { + err := tx.Model(&model.VideoVoteDay{}).Where("vote_day_id = ?", voteDayId).UpdateColumn(field, gorm.Expr(field+" + ?", numeral)).Error + if err != nil { + return err + } + return nil +} + +// Dec 自减 +func (r *VideoVoteDayDao) Dec(tx *gorm.DB, voteDayId int64, field string, numeral int) error { + err := tx.Model(&model.VideoVoteDay{}).Where("vote_day_id = ?", voteDayId).UpdateColumn(field, gorm.Expr(field+" - ?", numeral)).Error + if err != nil { + return err + } + return nil +} + +// GetVideoVoteDayByUserIdPageSearch 获取列表-分页-用户id +func (r *VideoVoteDayDao) GetVideoVoteDayByUserIdPageSearch(userId int64, page, pageSize int) (m []*model.VideoVoteDay, total int64, err error) { + var totalRecords int64 + + // 构建查询条件 + query := global.Db.Model(&model.VideoVoteDay{}) + + query = query.Preload("video") + + query = query.Where("user_id = ?", userId) + + query = query.Order("created_at desc") + + // 查询总数量 + if err := query.Count(&totalRecords).Error; err != nil { + return nil, 0, err + } + + err = query.Scopes(model.Paginate(page, pageSize)).Find(&m).Error + if err != nil { + return nil, 0, err + } + return m, totalRecords, nil +} + +// GetVideoVoteDayByVideoIdPageSearch 获取列表-分页-视频id +func (r *VideoVoteDayDao) GetVideoVoteDayByVideoIdPageSearch(videoId int64, page, pageSize int) (m []*model.VideoVoteDay, total int64, err error) { + var totalRecords int64 + + // 构建查询条件 + query := global.Db.Model(&model.VideoVoteDay{}) + + // 用户 + query = query.Preload("User") + + query = query.Where("video_id = ?", videoId) + + query = query.Order("created_at desc") + + // 查询总数量 + if err := query.Count(&totalRecords).Error; err != nil { + return nil, 0, err + } + + err = query.Scopes(model.Paginate(page, pageSize)).Find(&m).Error + if err != nil { + return nil, 0, err + } + return m, totalRecords, nil +} + +// GetVideoVoteDayListByTime 获取列表-开始时间/过期时间 +func (r *VideoVoteDayDao) GetVideoVoteDayListByTime(maps interface{}, startTime, endTime, field string) (m []*model.VideoVoteDay, err error) { + err = global.Db.Where(maps).Where(field+" BETWEEN ? AND ?", startTime, endTime).Find(&m).Error + if err != nil { + return nil, err + } + return m, nil +} diff --git a/api/dto/Article.go b/api/dto/Article.go new file mode 100644 index 0000000..b7929cb --- /dev/null +++ b/api/dto/Article.go @@ -0,0 +1,81 @@ +package dto + +import ( + "fmt" + "vote-admin-video-api/api/model" +) + +// ArticleDto 文章表 +type ArticleDto struct { + ArticleId string `json:"article_id"` // 主键id + ArticleTitle string `json:"article_title"` // 文章标题 + ArticleStatus int `json:"article_status"` // 文章状态（1:正常 2:禁用） + ArticleNumber string `json:"article_number"` // 文章编号 + VoteNum uint `json:"vote_num"` // 总票数 + ArticleContent string `json:"article_content"` // 文章内容 + CreatedAt model.LocalTime `json:"created_at"` // 创建时间 + UpdatedAt model.LocalTime `json:"updated_at"` // 更新时间 + ArticleAuthor []*ArticleAuthorDto `json:"article_author"` // 作者 + Rank *int `json:"rank"` // 排名 +} + +// GetArticleListDto 列表-分页 +func GetArticleListDto(m []*model.Article) []*ArticleDto { + // 处理返回值 + responses := make([]*ArticleDto, len(m)) + + if len(m) > 0 { + for i, v := range m { + response := &ArticleDto{ + ArticleId: fmt.Sprintf("%d", v.ArticleId), + ArticleTitle: v.ArticleTitle, + ArticleStatus: v.ArticleStatus, + ArticleNumber: v.ArticleNumber, + VoteNum: v.VoteNum, + ArticleContent: v.ArticleContent, + CreatedAt: v.CreatedAt, + UpdatedAt: v.UpdatedAt, + } + + // 加载数据-作者 + if v.ArticleAuthor != nil { + response = response.LoadArticleAuthor(v.ArticleAuthor) + } + + // 将转换后的结构体添加到新切片中 + responses[i] = response + } + } + + return responses +} + +// GetArticleDto 详情 +func GetArticleDto(m *model.Article) *ArticleDto { + return &ArticleDto{ + ArticleId: fmt.Sprintf("%d", m.ArticleId), + ArticleTitle: m.ArticleTitle, + ArticleStatus: m.ArticleStatus, + ArticleNumber: m.ArticleNumber, + VoteNum: m.VoteNum, + ArticleContent: m.ArticleContent, + CreatedAt: m.CreatedAt, + UpdatedAt: m.UpdatedAt, + } +} + +// LoadArticleAuthor 加载数据-作者 +func (r *ArticleDto) LoadArticleAuthor(m []*model.ArticleAuthor) *ArticleDto { + if len(m) > 0 { + r.ArticleAuthor = GetArticleAuthorListDto(m) + } + return r +} + +// LoadRank 加载数据-排名 +func (r *ArticleDto) LoadRank(m int) *ArticleDto { + if m > 0 { + r.Rank = &m + } + return r +} diff --git a/api/dto/ArticleAuthor.go b/api/dto/ArticleAuthor.go new file mode 100644 index 0000000..2a498e6 --- /dev/null +++ b/api/dto/ArticleAuthor.go @@ -0,0 +1,49 @@ +package dto + +import ( + "fmt" + "vote-admin-video-api/api/model" +) + +type ArticleAuthorDto struct { + AuthorId string `json:"author_id"` // 主键id + ArticleId string `json:"article_id"` // 文章id + AuthorName string `json:"author_name"` // 作者姓名 + HospitalId string `json:"hospital_id"` // 作者所属id + HospitalName string `json:"hospital_name"` // 作者所属医院 +} + +// GetArticleAuthorListDto 列表-分页 +func GetArticleAuthorListDto(m []*model.ArticleAuthor) []*ArticleAuthorDto { + // 处理返回值 + responses := make([]*ArticleAuthorDto, len(m)) + + if len(m) > 0 { + for i, v := range m { + response := &ArticleAuthorDto{ + AuthorId: fmt.Sprintf("%d", v.AuthorId), + ArticleId: fmt.Sprintf("%d", v.ArticleId), + AuthorName: v.AuthorName, + HospitalId: fmt.Sprintf("%d", v.HospitalId), + } + + // 加载数据-医院属性 + if v.BaseHospital != nil { + response = response.LoadBaseHospital(v.BaseHospital) + } + + // 将转换后的结构体添加到新切片中 + responses[i] = response + } + } + + return responses +} + +// LoadBaseHospital 加载数据-医院 +func (r *ArticleAuthorDto) LoadBaseHospital(m *model.BaseHospital) *ArticleAuthorDto { + if m != nil { + r.HospitalName = m.HospitalName + } + return r +} diff --git a/api/dto/BaseAgreement.go b/api/dto/BaseAgreement.go new file mode 100644 index 0000000..d594fce --- /dev/null +++ b/api/dto/BaseAgreement.go @@ -0,0 +1,52 @@ +package dto + +import ( + "fmt" + "vote-admin-video-api/api/model" +) + +// BaseAgreementDto 基础-协议 +type BaseAgreementDto struct { + AgreementId string `json:"agreement_id"` // 主键id + AgreementTitle string `json:"agreement_title"` // 协议名称 + AgreementType int `json:"agreement_type"` // 协议类型（1:大赛介绍 2:投票规则） + AgreementContent string `json:"agreement_content"` // 协议内容 + CreatedAt model.LocalTime `json:"created_at"` // 创建时间 + UpdatedAt model.LocalTime `json:"updated_at"` // 修改时间 +} + +// GetBaseAgreementDto 详情 +func GetBaseAgreementDto(m *model.BaseAgreement) *BaseAgreementDto { + return &BaseAgreementDto{ + AgreementId: fmt.Sprintf("%d", m.AgreementId), + AgreementTitle: m.AgreementTitle, + AgreementType: m.AgreementType, + AgreementContent: m.AgreementContent, + CreatedAt: m.CreatedAt, + UpdatedAt: m.UpdatedAt, + } +} + +// GetBaseAgreementListDto 列表-基础数据-协议 +func GetBaseAgreementListDto(m []*model.BaseAgreement) []*BaseAgreementDto { + // 处理返回值 + responses := make([]*BaseAgreementDto, len(m)) + + if len(m) > 0 { + for i, v := range m { + response := &BaseAgreementDto{ + AgreementId: fmt.Sprintf("%d", v.AgreementId), + AgreementTitle: v.AgreementTitle, + AgreementType: v.AgreementType, + AgreementContent: v.AgreementContent, + CreatedAt: v.CreatedAt, + UpdatedAt: v.UpdatedAt, + } + + // 将转换后的结构体添加到新切片中 + responses[i] = response + } + } + + return responses +} diff --git a/api/dto/BaseHospital.go b/api/dto/BaseHospital.go new file mode 100644 index 0000000..b2c28b5 --- /dev/null +++ b/api/dto/BaseHospital.go @@ -0,0 +1,85 @@ +package dto + +import ( + "fmt" + "vote-admin-video-api/api/model" +) + +type BaseHospitalDto struct { + HospitalID string `json:"hospital_id"` // 主键id + HospitalName string `json:"hospital_name"` // 医院名称 + HospitalStatus int `json:"hospital_status"` // 状态（0:禁用 1:正常 2:删除） + HospitalLevelName string `json:"hospital_level_name"` // 医院等级名称 + PostCode string `json:"post_code"` // 邮政编码 + Telephone string `json:"telephone"` // 电话 + ProvinceID int `json:"province_id"` // 省份id + Province string `json:"province"` // 省份 + CityID int `json:"city_id"` // 城市id + City string `json:"city"` // 城市 + CountyID int `json:"county_id"` // 区县id + County string `json:"county"` // 区县 + Address string `json:"address"` // 地址 + Latitude string `json:"latitude"` // 纬度 + Longitude string `json:"longitude"` // 经度 + Description string `json:"description"` // 简介 + CreatedAt model.LocalTime `json:"created_at"` // 创建时间 + UpdatedAt model.LocalTime `json:"updated_at"` // 修改时间 +} + +func GetBaseHospitalDto(m *model.BaseHospital) *BaseHospitalDto { + return &BaseHospitalDto{ + HospitalID: fmt.Sprintf("%d", m.HospitalId), + HospitalName: m.HospitalName, + HospitalStatus: m.HospitalStatus, + HospitalLevelName: m.HospitalLevelName, + PostCode: m.PostCode, + Telephone: m.HospitalName, + ProvinceID: m.ProvinceId, + Province: m.Province, + CityID: m.CityId, + City: m.City, + CountyID: m.CountyId, + County: m.County, + Address: m.Address, + Latitude: m.Lat, + Longitude: m.HospitalName, + Description: m.Desc, + CreatedAt: m.CreatedAt, + UpdatedAt: m.UpdatedAt, + } +} + +func GetBaseHospitalListDto(m []*model.BaseHospital) []BaseHospitalDto { + // 处理返回值 + responses := make([]BaseHospitalDto, len(m)) + + if len(m) > 0 { + for i, v := range m { + response := BaseHospitalDto{ + HospitalID: fmt.Sprintf("%d", v.HospitalId), + HospitalName: v.HospitalName, + HospitalStatus: v.HospitalStatus, + HospitalLevelName: v.HospitalLevelName, + PostCode: v.PostCode, + Telephone: v.HospitalName, + ProvinceID: v.ProvinceId, + Province: v.Province, + CityID: v.CityId, + City: v.City, + CountyID: v.CountyId, + County: v.County, + Address: v.Address, + Latitude: v.Lat, + Longitude: v.HospitalName, + Description: v.Desc, + CreatedAt: v.CreatedAt, + UpdatedAt: v.UpdatedAt, + } + + // 将转换后的结构体添加到新切片中 + responses[i] = response + } + } + + return responses +} diff --git a/api/dto/Data.go b/api/dto/Data.go new file mode 100644 index 0000000..81a993d --- /dev/null +++ b/api/dto/Data.go @@ -0,0 +1,18 @@ +package dto + +import ( + "vote-admin-video-api/api/model" +) + +type DataDto struct { + ViewNum uint `json:"view_num"` // 浏览数量 + VoteNum uint `json:"vote_num"` // 投票数量 +} + +// GetDataDto 详情 +func GetDataDto(m *model.Data) *DataDto { + return &DataDto{ + ViewNum: m.ViewNum, + VoteNum: m.VoteNum, + } +} diff --git a/api/dto/Public.go b/api/dto/Public.go new file mode 100644 index 0000000..80666ab --- /dev/null +++ b/api/dto/Public.go @@ -0,0 +1,42 @@ +package dto + +import ( + "fmt" + "vote-admin-video-api/api/model" + "vote-admin-video-api/utils" +) + +// LoginDto 登陆 +type LoginDto struct { + UserId string `json:"user_id"` // 用户id + NickName string `json:"nick_name"` // 用户名称 + Avatar string `json:"avatar"` // 头像 + Token string `json:"token"` // token +} + +// IndexDto 首页 +type IndexDto struct { + ViewNum uint `json:"view_num"` // 浏览数量 + VoteNum uint `json:"vote_num"` // 投票数量 +} + +// IndexDataDto 首页动态统计数据 +type IndexDataDto struct { + Date string `json:"date"` // 日期 + Count int64 `json:"count"` // 数量 +} + +// AdminLoginDto 微信登陆 +func AdminLoginDto(m *model.AdminUser) *LoginDto { + return &LoginDto{ + UserId: fmt.Sprintf("%d", m.UserId), + NickName: m.NickName, + Avatar: utils.AddOssDomain(m.Avatar), + } +} + +// LoadToken 加载token +func (r *LoginDto) LoadToken(token string) *LoginDto { + r.Token = token + return r +} diff --git a/api/dto/SystemTime.go b/api/dto/SystemTime.go new file mode 100644 index 0000000..7152c40 --- /dev/null +++ b/api/dto/SystemTime.go @@ -0,0 +1,26 @@ +package dto + +import ( + "fmt" + "vote-admin-video-api/api/model" +) + +// SystemTimeDto 配置-时间 +type SystemTimeDto struct { + SystemTimeId string `json:"system_time_id"` // 主键id + StartTime *model.LocalTime `json:"start_time"` // 开始投票时间 + EndTime *model.LocalTime `json:"end_time"` // 结束投票时间 + CreatedAt model.LocalTime `json:"created_at"` // 创建时间 + UpdatedAt model.LocalTime `json:"updated_at"` // 修改时间 +} + +// GetSystemTimeDto 详情 +func GetSystemTimeDto(m *model.SystemTime) *SystemTimeDto { + return &SystemTimeDto{ + SystemTimeId: fmt.Sprintf("%d", m.SystemTimeId), + StartTime: m.StartTime, + EndTime: m.EndTime, + CreatedAt: m.CreatedAt, + UpdatedAt: m.UpdatedAt, + } +} diff --git a/api/dto/User.go b/api/dto/User.go new file mode 100644 index 0000000..aa06f72 --- /dev/null +++ b/api/dto/User.go @@ -0,0 +1,67 @@ +package dto + +import ( + "fmt" + "vote-admin-video-api/api/model" +) + +// UserDto 用户表 +type UserDto struct { + UserId string `json:"user_id"` // 用户id + UserStatus int `json:"user_status"` // 状态（1:正常 2:禁用） + OpenId string `json:"open_id"` // 用户微信标识 + AppIden string `json:"app_iden"` // app唯一标识 + LoginAt *model.LocalTime `json:"login_at"` // 登陆时间 + LoginIp string `json:"login_ip"` // 登陆ip + CreatedAt model.LocalTime `json:"created_at"` // 创建时间 + UpdatedAt model.LocalTime `json:"updated_at"` // 更新时间 +} + +// UserVoteDto 用户投票记录 +type UserVoteDto struct { + Id string `json:"id"` // 图文/视频标识 + AppIden string `json:"app_iden"` // app唯一标识 + OpenId string `json:"open_id"` // open_id + Title string `json:"title"` // 标题 + VotedAt string `json:"voted_at"` // 投票时间 +} + +// GetUserListDto 列表 +func GetUserListDto(m []*model.User) []*UserDto { + // 处理返回值 + responses := make([]*UserDto, len(m)) + + if len(m) > 0 { + for i, v := range m { + response := &UserDto{ + UserId: fmt.Sprintf("%d", v.UserId), + UserStatus: v.UserStatus, + OpenId: v.OpenId, + AppIden: v.AppIden, + LoginAt: v.LoginAt, + LoginIp: v.LoginIp, + CreatedAt: v.CreatedAt, + UpdatedAt: v.UpdatedAt, + } + + // 将转换后的结构体添加到新切片中 + responses[i] = response + } + } + + return responses +} + +// GetUserDto 详情-问题 +func GetUserDto(m *model.User) *UserDto { + return &UserDto{ + UserId: fmt.Sprintf("%d", m.UserId), + UserStatus: m.UserStatus, + OpenId: m.OpenId, + AppIden: m.AppIden, + LoginAt: m.LoginAt, + LoginIp: m.LoginIp, + CreatedAt: m.CreatedAt, + UpdatedAt: m.UpdatedAt, + } +} diff --git a/api/dto/Video.go b/api/dto/Video.go new file mode 100644 index 0000000..38382aa --- /dev/null +++ b/api/dto/Video.go @@ -0,0 +1,84 @@ +package dto + +import ( + "fmt" + "vote-admin-video-api/api/model" +) + +// VideoDto 视频表 +type VideoDto struct { + VideoId string `json:"video_id"` // 主键id + VideoTitle string `json:"video_title"` // 视频标题 + VideoStatus int `json:"video_status"` // 视频状态（1:正常 2:禁用） + VideoNumber string `json:"video_number"` // 视频编号 + VideoNo string `json:"video_no"` // 视频编号 + VoteNum uint `json:"vote_num"` // 总票数 + VideoContent string `json:"video_content"` // 视频内容 + CreatedAt model.LocalTime `json:"created_at"` // 创建时间 + UpdatedAt model.LocalTime `json:"updated_at"` // 更新时间 + VideoAuthor []*VideoAuthorDto `json:"video_author"` // 作者 + Rank *int `json:"rank"` // 排名 +} + +// GetVideoListDto 列表-分页 +func GetVideoListDto(m []*model.Video) []*VideoDto { + // 处理返回值 + responses := make([]*VideoDto, len(m)) + + if len(m) > 0 { + for i, v := range m { + response := &VideoDto{ + VideoId: fmt.Sprintf("%d", v.VideoId), + VideoTitle: v.VideoTitle, + VideoStatus: v.VideoStatus, + VideoNumber: v.VideoNumber, + VideoNo: v.VideoNo, + VoteNum: v.VoteNum, + VideoContent: v.VideoContent, + CreatedAt: v.CreatedAt, + UpdatedAt: v.UpdatedAt, + } + + // 加载数据-作者 + if v.VideoAuthor != nil { + response = response.LoadVideoAuthor(v.VideoAuthor) + } + + // 将转换后的结构体添加到新切片中 + responses[i] = response + } + } + + return responses +} + +// GetVideoDto 详情 +func GetVideoDto(m *model.Video) *VideoDto { + return &VideoDto{ + VideoId: fmt.Sprintf("%d", m.VideoId), + VideoTitle: m.VideoTitle, + VideoStatus: m.VideoStatus, + VideoNumber: m.VideoNumber, + VideoNo: m.VideoNo, + VoteNum: m.VoteNum, + VideoContent: m.VideoContent, + CreatedAt: m.CreatedAt, + UpdatedAt: m.UpdatedAt, + } +} + +// LoadVideoAuthor 加载数据-作者 +func (r *VideoDto) LoadVideoAuthor(m []*model.VideoAuthor) *VideoDto { + if len(m) > 0 { + r.VideoAuthor = GetVideoAuthorListDto(m) + } + return r +} + +// LoadRank 加载数据-排名 +func (r *VideoDto) LoadRank(m int) *VideoDto { + if m > 0 { + r.Rank = &m + } + return r +} diff --git a/api/dto/VideoAuthor.go b/api/dto/VideoAuthor.go new file mode 100644 index 0000000..cd5c50c --- /dev/null +++ b/api/dto/VideoAuthor.go @@ -0,0 +1,49 @@ +package dto + +import ( + "fmt" + "vote-admin-video-api/api/model" +) + +type VideoAuthorDto struct { + AuthorId string `json:"author_id"` // 主键id + VideoId string `json:"video_id"` // 文章id + HospitalId string `json:"hospital_id"` // 作者所属id + HospitalName string `json:"hospital_name"` // 作者所属医院 + AuthorName string `json:"author_name"` // 作者姓名 +} + +// GetVideoAuthorListDto 列表-分页 +func GetVideoAuthorListDto(m []*model.VideoAuthor) []*VideoAuthorDto { + // 处理返回值 + responses := make([]*VideoAuthorDto, len(m)) + + if len(m) > 0 { + for i, v := range m { + response := &VideoAuthorDto{ + AuthorId: fmt.Sprintf("%d", v.AuthorId), + VideoId: fmt.Sprintf("%d", v.VideoId), + AuthorName: v.AuthorName, + HospitalId: fmt.Sprintf("%d", v.HospitalId), + } + + // 加载数据-医院属性 + if v.BaseHospital != nil { + response = response.LoadBaseHospitalAttr(v.BaseHospital) + } + + // 将转换后的结构体添加到新切片中 + responses[i] = response + } + } + + return responses +} + +// LoadBaseHospitalAttr 加载数据-医院属性 +func (r *VideoAuthorDto) LoadBaseHospitalAttr(m *model.BaseHospital) *VideoAuthorDto { + if m != nil { + r.HospitalName = m.HospitalName + } + return r +} diff --git a/api/exception/exception.go b/api/exception/exception.go new file mode 100644 index 0000000..c2ad971 --- /dev/null +++ b/api/exception/exception.go @@ -0,0 +1,43 @@ +package exception + +import ( + "github.com/gin-gonic/gin" + "log" + "net/http" + "runtime/debug" + "vote-admin-video-api/consts" +) + +// Recover +// @Description: 处理全局异常 +// @return gin.HandlerFunc +func Recover() gin.HandlerFunc { + return func(c *gin.Context) { + defer func() { + if r := recover(); r != nil { + // 打印错误堆栈信息 + log.Printf("panic: %v\n", r) + debug.PrintStack() + c.JSON(http.StatusInternalServerError, gin.H{ + "code": consts.ServerError, + "message": errorToString(r), + "data": "", + }) + // 终止后续接口调用，不加的话recover到异常后，还会继续执行接口里后续代码 + c.Abort() + } + }() + // 加载完 defer recover，继续后续接口调用 + c.Next() + } +} + +// recover错误，转string +func errorToString(r interface{}) string { + switch v := r.(type) { + case error: + return v.Error() + default: + return r.(string) + } +} diff --git a/api/middlewares/auth.go b/api/middlewares/auth.go new file mode 100644 index 0000000..580868a --- /dev/null +++ b/api/middlewares/auth.go @@ -0,0 +1,79 @@ +package middlewares + +import ( + "github.com/gin-gonic/gin" + "net/http" + "vote-admin-video-api/api/dao" + "vote-admin-video-api/api/responses" + "vote-admin-video-api/consts" +) + +// Auth Auth认证 +func Auth() gin.HandlerFunc { + return func(c *gin.Context) { + // 获取用户id + adminUserId := c.GetInt64("AdminUserId") + if adminUserId == 0 { + responses.Fail(c) + c.Abort() + return + } + + // 获取用户数据 + adminUserDao := dao.AdminUserDao{} + adminUser, err := adminUserDao.GetAdminUserById(adminUserId) + if err != nil || adminUser == nil { + c.JSON(http.StatusUnauthorized, gin.H{ + "message": "用户数据错误", + "code": consts.UserStatusError, + "data": "", + }) + c.Abort() + return + } + + if adminUser.Status == 2 { + c.JSON(http.StatusUnauthorized, gin.H{ + "message": "用户审核中", + "code": consts.UserStatusError, + "data": "", + }) + c.Abort() + return + } + + if adminUser.Status == 3 { + c.JSON(http.StatusUnauthorized, gin.H{ + "message": "用户状态异常", + "code": consts.UserStatusError, + "data": "", + }) + c.Abort() + return + } + + if adminUser.IsDisabled == 1 { + c.JSON(http.StatusUnauthorized, gin.H{ + "message": "用户已被禁用", + "code": consts.UserStatusError, + "data": "", + }) + c.Abort() + return + } + + if adminUser.IsDeleted == 1 { + c.JSON(http.StatusUnauthorized, gin.H{ + "message": "用户状态异常", + "code": consts.UserStatusError, + "data": "", + }) + c.Abort() + return + } + + c.Set("AdminUserId", adminUserId) // 用户id + + c.Next() + } +} diff --git a/api/middlewares/cors.go b/api/middlewares/cors.go new file mode 100644 index 0000000..9ee0ab4 --- /dev/null +++ b/api/middlewares/cors.go @@ -0,0 +1,29 @@ +package middlewares + +import ( + "github.com/gin-gonic/gin" + "net/http" +) + +// Cors +// @Description: 跨域中间件 +// @return gin.HandlerFunc +func Cors() gin.HandlerFunc { + return func(c *gin.Context) { + method := c.Request.Method + origin := c.Request.Header.Get("Origin") + if origin != "" { + c.Header("Access-Control-Allow-Origin", origin) + c.Header("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE, UPDATE") + c.Header("Access-Control-Allow-Headers", "Origin, X-Requested-With, Content-Type, Accept, Authorization") + c.Header("Access-Control-Expose-Headers", "Content-Length, Access-Control-Allow-Origin, Access-Control-Allow-Headers, Cache-Control, Content-Language, Content-Type") + c.Header("Access-Control-Allow-Credentials", "false") + c.Set("content-type", "application/json") + } + if method == "OPTIONS" { + c.AbortWithStatus(http.StatusNoContent) + return + } + c.Next() + } +} diff --git a/api/middlewares/jwt.go b/api/middlewares/jwt.go new file mode 100644 index 0000000..5922e00 --- /dev/null +++ b/api/middlewares/jwt.go @@ -0,0 +1,82 @@ +package middlewares + +import ( + "github.com/gin-gonic/gin" + "net/http" + "strconv" + "strings" + "vote-admin-video-api/consts" + "vote-admin-video-api/global" + "vote-admin-video-api/utils" +) + +// Jwt jwt认证 +func Jwt() gin.HandlerFunc { + return func(c *gin.Context) { + authorization := c.Request.Header.Get("Authorization") + if authorization == "" || !strings.HasPrefix(authorization, "Bearer ") { + c.JSON(http.StatusUnauthorized, gin.H{ + "message": "请求未授权", + "code": consts.TokenError, + "data": "", + }) + c.Abort() + return + } + + // 去除Bearer + authorization = authorization[7:] // 截取字符 + if authorization == "" { + c.JSON(http.StatusOK, gin.H{ + "message": "token错误/过期", + "code": consts.TokenError, + "data": "", + }) + + c.Abort() + return + } + + // 检测是否存在黑名单 + res, _ := global.Redis.Get(c, "jwt_black_"+authorization).Result() + if res != "" { + c.JSON(http.StatusOK, gin.H{ + "message": "token错误/过期", + "code": consts.TokenError, + "data": "", + }) + + c.Abort() + return + } + + // 解析jwt + t, err := utils.ParseJwt(authorization) + if err != nil { + c.JSON(http.StatusOK, gin.H{ + "message": "token错误/过期", + "code": consts.TokenError, + "data": "", + }) + + c.Abort() + return + } + + // 转换类型 + userId, err := strconv.ParseInt(t.UserId, 10, 64) + if err != nil { + c.JSON(http.StatusOK, gin.H{ + "message": "token错误", + "code": consts.TokenError, + "data": "", + }) + + c.Abort() + return + } + + c.Set("AdminUserId", userId) // 用户id + c.Next() + } +} diff --git a/api/middlewares/logrus.go b/api/middlewares/logrus.go new file mode 100644 index 0000000..5e8c080 --- /dev/null +++ b/api/middlewares/logrus.go @@ -0,0 +1,60 @@ +package middlewares + +import ( + "fmt" + "github.com/gin-gonic/gin" + "github.com/sirupsen/logrus" + "time" + "vote-admin-video-api/global" +) + +// Logrus 日志中间件 +func Logrus() gin.HandlerFunc { + return func(c *gin.Context) { + + // 开始时间 + startTime := time.Now() + + // 处理请求 + c.Next() + + // 获取 请求 参数 + params := make(map[string]string) + + paramsRaw, ok := c.Get("params") + if ok { + requestParams, ok := paramsRaw.(map[string]string) + if ok || len(requestParams) > 0 { + params = requestParams + } + } + + // 结束时间 + endTime := time.Now() + + // 执行时间 + latencyTime := fmt.Sprintf("%6v", endTime.Sub(startTime)) + + // 请求方式 + reqMethod := c.Request.Method + + // 请求路由 + reqUri := c.Request.RequestURI + + // 状态码 + statusCode := c.Writer.Status() + + // 请求IP + clientIP := c.ClientIP() + + // 日志格式 + global.Logger.WithFields(logrus.Fields{ + "http_status": statusCode, + "total_time": latencyTime, + "ip": clientIP, + "method": reqMethod, + "uri": reqUri, + "params": params, + }).Info("access") + } +} diff --git a/api/middlewares/requestParamsMiddleware.go b/api/middlewares/requestParamsMiddleware.go new file mode 100644 index 0000000..52d4ba9 --- /dev/null +++ b/api/middlewares/requestParamsMiddleware.go @@ -0,0 +1,92 @@ +package middlewares + +import ( + "bytes" + "encoding/json" + "fmt" + "github.com/gin-gonic/gin" + "io" + "net/http" + "vote-admin-video-api/consts" +) + +// RequestParamsMiddleware 获取请求参数中间件 +func RequestParamsMiddleware() gin.HandlerFunc { + return func(c *gin.Context) { + contentType := c.Request.Header.Get("Content-Type") + + params := make(map[string]string) + + // 判断请求参数类型 + switch contentType { + case "application/json": + // 解析 application/json 请求体 + bodyBytes, err := io.ReadAll(c.Request.Body) + if err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to read request body"}) + c.Abort() + return + } + + // 创建新的请求对象，并设置请求体数据 + c.Request.Body = io.NopCloser(bytes.NewBuffer(bodyBytes)) + + var jsonParams map[string]interface{} + err = json.Unmarshal(bodyBytes, &jsonParams) + if err != nil { + c.JSON(http.StatusBadRequest, gin.H{ + "message": "Invalid JSON data", + "code": consts.HttpError, + "data": "", + }) + c.Abort() + return + } + + for key, value := range jsonParams { + params[key] = fmt.Sprintf("%v", value) + } + + // 存储参数到上下文 + c.Set("params", params) + + case "multipart/form-data", "application/form-data", "application/x-www-form-urlencoded": + // 解析 Form 表单参数 + err := c.Request.ParseForm() + if err != nil { + c.JSON(http.StatusBadRequest, gin.H{ + "message": "Invalid form data", + "code": consts.HttpError, + "data": "", + }) + c.Abort() + return + } + + for key, values := range c.Request.Form { + if len(values) > 0 { + params[key] = fmt.Sprintf("%v", values[0]) + } + } + + // 存储参数到上下文 + c.Set("params", params) + + default: + // 解析 URL 参数 + queryParams := c.Request.URL.Query() + + for key, values := range queryParams { + if len(values) > 0 { + params[key] = fmt.Sprintf("%v", values[0]) + } + } + + // 存储参数到上下文 + c.Set("params", params) + } + + // 继续处理请求 + c.Next() + } +} diff --git a/api/model/AdminUser.go b/api/model/AdminUser.go new file mode 100644 index 0000000..ff4fb06 --- /dev/null +++ b/api/model/AdminUser.go @@ -0,0 +1,42 @@ +package model + +import ( + "gorm.io/gorm" + "time" + "vote-admin-video-api/global" +) + +// AdminUser 后台-用户表 +type AdminUser struct { + UserId int64 `gorm:"column:user_id;type:bigint(19);primary_key;comment:主键id" json:"user_id"` + Access string `gorm:"column:access;type:varchar(64);comment:账号;NOT NULL" json:"access"` + Password string `gorm:"column:password;type:varchar(128);comment:密码;NOT NULL" json:"password"` + Salt string `gorm:"column:salt;type:varchar(255);comment:密码掩码;NOT NULL" json:"salt"` + NickName string `gorm:"column:nick_name;type:varchar(255);comment:昵称" json:"nick_name"` + Status int `gorm:"column:status;type:tinyint(1);default:2;comment:状态（1:正常 2:审核中 3:审核失败）" json:"status"` + IsDeleted int `gorm:"column:is_deleted;type:tinyint(1);default:0;comment:是否被删除（0:否 1:是）" json:"is_deleted"` + IsDisabled int `gorm:"column:is_disabled;type:tinyint(1);default:0;comment:是否被禁用（0:否 1:是）" json:"is_disabled"` + Phone string `gorm:"column:phone;type:varchar(11);comment:手机号" json:"phone"` + Avatar string `gorm:"column:avatar;type:varchar(255);comment:头像" json:"avatar"` + Sex int `gorm:"column:sex;type:tinyint(1);comment:性别（1:男 2:女）" json:"sex"` + Email string `gorm:"column:email;type:varchar(100);comment:邮箱" json:"email"` + Model +} + +func (m *AdminUser) TableName() string { + return "admin_user" +} + +func (m *AdminUser) BeforeCreate(tx *gorm.DB) error { + if m.UserId == 0 { + m.UserId = global.Snowflake.Generate().Int64() + } + + m.CreatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("CreatedAt", m.CreatedAt) + + m.UpdatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("UpdatedAt", m.UpdatedAt) + + return nil +} diff --git a/api/model/Article.go b/api/model/Article.go new file mode 100644 index 0000000..62aea87 --- /dev/null +++ b/api/model/Article.go @@ -0,0 +1,37 @@ +package model + +import ( + "gorm.io/gorm" + "time" + "vote-admin-video-api/global" +) + +// Article 文章表 +type Article struct { + ArticleId int64 `gorm:"column:article_id;type:bigint(19);primary_key;comment:主键id" json:"article_id"` + ArticleTitle string `gorm:"column:article_title;type:varchar(200);comment:文章标题" json:"article_title"` + ArticleStatus int `gorm:"column:article_status;type:tinyint(1);default:1;comment:文章状态（1:正常 2:禁用）" json:"article_status"` + ArticleNumber string `gorm:"column:article_number;type:varchar(10);comment:文章编号;NOT NULL" json:"article_number"` + VoteNum uint `gorm:"column:vote_num;type:int(10) unsigned;default:0;comment:总票数" json:"vote_num"` + ArticleContent string `gorm:"column:article_content;type:text;comment:文章内容" json:"article_content"` + Model + ArticleAuthor []*ArticleAuthor `gorm:"foreignKey:ArticleId;references:article_id" json:"article_author"` +} + +func (m *Article) TableName() string { + return "article" +} + +func (m *Article) BeforeCreate(tx *gorm.DB) error { + if m.ArticleId == 0 { + m.ArticleId = global.Snowflake.Generate().Int64() + } + + m.CreatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("CreatedAt", m.CreatedAt) + + m.UpdatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("UpdatedAt", m.UpdatedAt) + + return nil +} diff --git a/api/model/ArticleAuthor.go b/api/model/ArticleAuthor.go new file mode 100644 index 0000000..67b1c4b --- /dev/null +++ b/api/model/ArticleAuthor.go @@ -0,0 +1,35 @@ +package model + +import ( + "gorm.io/gorm" + "time" + "vote-admin-video-api/global" +) + +// ArticleAuthor 文章-作者表 +type ArticleAuthor struct { + AuthorId int64 `gorm:"column:author_id;type:bigint(19);primary_key;comment:主键id" json:"author_id"` + ArticleId int64 `gorm:"column:article_id;type:bigint(19);comment:文章id;NOT NULL" json:"article_id"` + AuthorName string `gorm:"column:author_name;type:varchar(100);comment:作者姓名" json:"author_name"` + HospitalId int64 `gorm:"column:hospital_id;type:bigint(19);comment:作者所属医院id" json:"hospital_id"` + Model + BaseHospital *BaseHospital `gorm:"foreignKey:HospitalId;references:hospital_id" json:"base_hospital"` +} + +func (m *ArticleAuthor) TableName() string { + return "article_author" +} + +func (m *ArticleAuthor) BeforeCreate(tx *gorm.DB) error { + if m.AuthorId == 0 { + m.AuthorId = global.Snowflake.Generate().Int64() + } + + m.CreatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("CreatedAt", m.CreatedAt) + + m.UpdatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("UpdatedAt", m.UpdatedAt) + + return nil +} diff --git a/api/model/ArticleVoteDay.go b/api/model/ArticleVoteDay.go new file mode 100644 index 0000000..70491bd --- /dev/null +++ b/api/model/ArticleVoteDay.go @@ -0,0 +1,36 @@ +package model + +import ( + "gorm.io/gorm" + "time" + "vote-admin-video-api/global" +) + +// ArticleVoteDay 文章-每日投票 +type ArticleVoteDay struct { + VoteDayId int64 `gorm:"column:vote_day_id;type:bigint(19);primary_key;comment:主键id" json:"vote_day_id"` + ArticleId int64 `gorm:"column:article_id;type:bigint(19);comment:文章id;NOT NULL" json:"article_id"` + UserId int64 `gorm:"column:user_id;type:bigint(19);comment:用户id;NOT NULL" json:"user_id"` + VotedAt *LocalTime `gorm:"column:voted_at;type:date;comment:投票时间（日）" json:"voted_at"` + Model + Article *Article `gorm:"foreignKey:ArticleId;references:article_id" json:"article"` + User *User `gorm:"foreignKey:UserId;references:user_id" json:"user"` +} + +func (m *ArticleVoteDay) TableName() string { + return "article_vote_day" +} + +func (m *ArticleVoteDay) BeforeCreate(tx *gorm.DB) error { + if m.VoteDayId == 0 { + m.VoteDayId = global.Snowflake.Generate().Int64() + } + + m.CreatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("CreatedAt", m.CreatedAt) + + m.UpdatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("UpdatedAt", m.UpdatedAt) + + return nil +} diff --git a/api/model/BaseAgreement.go b/api/model/BaseAgreement.go new file mode 100644 index 0000000..a0d779c --- /dev/null +++ b/api/model/BaseAgreement.go @@ -0,0 +1,34 @@ +package model + +import ( + "gorm.io/gorm" + "time" + "vote-admin-video-api/global" +) + +// BaseAgreement 基础-协议 +type BaseAgreement struct { + AgreementId int64 `gorm:"column:agreement_id;type:bigint(19);primary_key;comment:主键id" json:"agreement_id"` + AgreementTitle string `gorm:"column:agreement_title;type:varchar(255);comment:协议标题" json:"agreement_title"` + AgreementType int `gorm:"column:agreement_type;type:tinyint(1);comment:协议类型（1:大赛介绍 2:投票规则）;NOT NULL" json:"agreement_type"` + AgreementContent string `gorm:"column:agreement_content;type:text;comment:协议内容" json:"agreement_content"` + Model +} + +func (m *BaseAgreement) TableName() string { + return "base_agreement" +} + +func (m *BaseAgreement) BeforeCreate(tx *gorm.DB) error { + if m.AgreementId == 0 { + m.AgreementId = global.Snowflake.Generate().Int64() + } + + m.CreatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("CreatedAt", m.CreatedAt) + + m.UpdatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("UpdatedAt", m.UpdatedAt) + + return nil +} diff --git a/api/model/BaseArea.go b/api/model/BaseArea.go new file mode 100644 index 0000000..e8c5c02 --- /dev/null +++ b/api/model/BaseArea.go @@ -0,0 +1,26 @@ +package model + +import ( + "gorm.io/gorm" + "vote-admin-video-api/global" +) + +// BaseArea 地区表 +type BaseArea struct { + AreaId int64 `gorm:"column:area_id;type:bigint(19);primary_key;comment:地区编号" json:"area_id"` + AreaName string `gorm:"column:area_name;type:varchar(255);comment:名称" json:"area_name"` + ParentId int64 `gorm:"column:parent_id;type:bigint(19);comment:上级编号" json:"parent_id"` + Zip string `gorm:"column:zip;type:varchar(10);comment:邮编" json:"zip"` + AreaType int `gorm:"column:area_type;type:tinyint(4);comment:类型(1:国家,2:省,3:市,4:区县)" json:"area_type"` +} + +func (m *BaseArea) TableName() string { + return "base_area" +} + +func (m *BaseArea) BeforeCreate(tx *gorm.DB) error { + if m.AreaId == 0 { + m.AreaId = global.Snowflake.Generate().Int64() + } + return nil +} diff --git a/api/model/BaseHospital.go b/api/model/BaseHospital.go new file mode 100644 index 0000000..2352e03 --- /dev/null +++ b/api/model/BaseHospital.go @@ -0,0 +1,46 @@ +package model + +import ( + "gorm.io/gorm" + "time" + "vote-admin-video-api/global" +) + +// BaseHospital 医院表 +type BaseHospital struct { + HospitalId int64 `gorm:"column:hospital_id;type:bigint(19);primary_key;comment:主键id" json:"hospital_id"` + HospitalName string `gorm:"column:hospital_name;type:varchar(255);comment:医院名称" json:"hospital_name"` + HospitalStatus int `gorm:"column:hospital_status;type:tinyint(1);default:1;comment:状态（0:禁用 1:正常 2:删除）" json:"hospital_status"` + HospitalLevelName string `gorm:"column:hospital_level_name;type:varchar(20);comment:医院等级名称" json:"hospital_level_name"` + PostCode string `gorm:"column:post_code;type:varchar(50);comment:邮政编码" json:"post_code"` + TelePhone string `gorm:"column:tele_phone;type:varchar(20);comment:电话" json:"tele_phone"` + ProvinceId int `gorm:"column:province_id;type:int(11);comment:省份id" json:"province_id"` + Province string `gorm:"column:province;type:varchar(50);comment:省份" json:"province"` + CityId int `gorm:"column:city_id;type:int(11);comment:城市id" json:"city_id"` + City string `gorm:"column:city;type:varchar(50);comment:城市" json:"city"` + CountyId int `gorm:"column:county_id;type:int(11);comment:区县id" json:"county_id"` + County string `gorm:"column:county;type:varchar(50);comment:区县" json:"county"` + Address string `gorm:"column:address;type:varchar(255);comment:地址" json:"address"` + Lat string `gorm:"column:lat;type:varchar(255);comment:纬度" json:"lat"` + Lng string `gorm:"column:lng;type:varchar(255);comment:经度" json:"lng"` + Desc string `gorm:"column:desc;type:varchar(255);comment:简介" json:"desc"` + Model +} + +func (m *BaseHospital) TableName() string { + return "base_hospital" +} + +func (m *BaseHospital) BeforeCreate(tx *gorm.DB) error { + if m.HospitalId == 0 { + m.HospitalId = global.Snowflake.Generate().Int64() + } + + m.CreatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("CreatedAt", m.CreatedAt) + + m.UpdatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("UpdatedAt", m.UpdatedAt) + + return nil +} diff --git a/api/model/Data.go b/api/model/Data.go new file mode 100644 index 0000000..9b19d4c --- /dev/null +++ b/api/model/Data.go @@ -0,0 +1,33 @@ +package model + +import ( + "gorm.io/gorm" + "time" + "vote-admin-video-api/global" +) + +// Data 数据表 +type Data struct { + DataId int64 `gorm:"column:data_id;type:bigint(19);primary_key;comment:主键id" json:"data_id"` + ViewNum uint `gorm:"column:view_num;type:int(10) unsigned;default:0;comment:浏览数量" json:"view_num"` + VoteNum uint `gorm:"column:vote_num;type:int(10) unsigned;default:0;comment:投票数量" json:"vote_num"` + Model +} + +func (m *Data) TableName() string { + return "data" +} + +func (m *Data) BeforeCreate(tx *gorm.DB) error { + if m.DataId == 0 { + m.DataId = global.Snowflake.Generate().Int64() + } + + m.CreatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("CreatedAt", m.CreatedAt) + + m.UpdatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("UpdatedAt", m.UpdatedAt) + + return nil +} diff --git a/api/model/SystemTime.go b/api/model/SystemTime.go new file mode 100644 index 0000000..5a1d487 --- /dev/null +++ b/api/model/SystemTime.go @@ -0,0 +1,33 @@ +package model + +import ( + "gorm.io/gorm" + "time" + "vote-admin-video-api/global" +) + +// SystemTime 配置-时间 +type SystemTime struct { + SystemTimeId int64 `gorm:"column:system_time_id;type:bigint(19);primary_key;comment:主键id" json:"system_time_id"` + StartTime *LocalTime `gorm:"column:start_time;type:datetime;comment:开始投票时间;NOT NULL" json:"start_time"` + EndTime *LocalTime `gorm:"column:end_time;type:datetime;comment:结束投票时间;NOT NULL" json:"end_time"` + Model +} + +func (m *SystemTime) TableName() string { + return "system_time" +} + +func (m *SystemTime) BeforeCreate(tx *gorm.DB) error { + if m.SystemTimeId == 0 { + m.SystemTimeId = global.Snowflake.Generate().Int64() + } + + m.CreatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("CreatedAt", m.CreatedAt) + + m.UpdatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("UpdatedAt", m.UpdatedAt) + + return nil +} diff --git a/api/model/User.go b/api/model/User.go new file mode 100644 index 0000000..76c5ba2 --- /dev/null +++ b/api/model/User.go @@ -0,0 +1,36 @@ +package model + +import ( + "gorm.io/gorm" + "time" + "vote-admin-video-api/global" +) + +// User 用户表 +type User struct { + UserId int64 `gorm:"column:user_id;type:bigint(19);primary_key;comment:用户id" json:"user_id"` + AppIden string `gorm:"column:app_iden;type:varchar(50);comment:app唯一标识" json:"app_iden"` + UserStatus int `gorm:"column:user_status;type:tinyint(1);default:1;comment:状态（1:正常 2:禁用）" json:"user_status"` + OpenId string `gorm:"column:open_id;type:varchar(100);comment:用户微信标识" json:"open_id"` + LoginAt *LocalTime `gorm:"column:login_at;type:datetime;comment:登陆时间" json:"login_at"` + LoginIp string `gorm:"column:login_ip;type:varchar(255);comment:登陆ip" json:"login_ip"` + Model +} + +func (m *User) TableName() string { + return "user" +} + +func (m *User) BeforeCreate(tx *gorm.DB) error { + if m.UserId == 0 { + m.UserId = global.Snowflake.Generate().Int64() + } + + m.CreatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("CreatedAt", m.CreatedAt) + + m.UpdatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("UpdatedAt", m.UpdatedAt) + + return nil +} diff --git a/api/model/Video.go b/api/model/Video.go new file mode 100644 index 0000000..21281a3 --- /dev/null +++ b/api/model/Video.go @@ -0,0 +1,38 @@ +package model + +import ( + "gorm.io/gorm" + "time" + "vote-admin-video-api/global" +) + +// Video 视频表 +type Video struct { + VideoId int64 `gorm:"column:video_id;type:bigint(19);primary_key;comment:主键id" json:"video_id"` + VideoTitle string `gorm:"column:video_title;type:varchar(200);comment:视频标题" json:"video_title"` + VideoStatus int `gorm:"column:video_status;type:tinyint(1);default:1;comment:视频状态（1:正常 2:禁用）" json:"video_status"` + VideoNumber string `gorm:"column:video_number;type:varchar(10);comment:视频编号;NOT NULL" json:"video_number"` + VoteNum uint `gorm:"column:vote_num;type:int(10) unsigned;default:0;comment:总票数" json:"vote_num"` + VideoNo string `gorm:"column:video_no;type:varchar(255);comment:视频编号（保利）" json:"video_no"` + VideoContent string `gorm:"column:video_content;type:text;comment:视频内容" json:"video_content"` + Model + VideoAuthor []*VideoAuthor `gorm:"foreignKey:VideoId;references:video_id" json:"video_author"` +} + +func (m *Video) TableName() string { + return "video" +} + +func (m *Video) BeforeCreate(tx *gorm.DB) error { + if m.VideoId == 0 { + m.VideoId = global.Snowflake.Generate().Int64() + } + + m.CreatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("CreatedAt", m.CreatedAt) + + m.UpdatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("UpdatedAt", m.UpdatedAt) + + return nil +} diff --git a/api/model/VideoAuthor.go b/api/model/VideoAuthor.go new file mode 100644 index 0000000..9c19a3e --- /dev/null +++ b/api/model/VideoAuthor.go @@ -0,0 +1,35 @@ +package model + +import ( + "gorm.io/gorm" + "time" + "vote-admin-video-api/global" +) + +// VideoAuthor 视频-作者表 +type VideoAuthor struct { + AuthorId int64 `gorm:"column:author_id;type:bigint(19);primary_key;comment:主键id" json:"author_id"` + VideoId int64 `gorm:"column:video_id;type:bigint(19);comment:视频id;NOT NULL" json:"video_id"` + AuthorName string `gorm:"column:author_name;type:varchar(100);comment:作者姓名" json:"author_name"` + HospitalId int64 `gorm:"column:hospital_id;type:bigint(19);comment:作者所属医院id" json:"hospital_id"` + Model + BaseHospital *BaseHospital `gorm:"foreignKey:HospitalId;references:hospital_id" json:"base_hospital"` +} + +func (m *VideoAuthor) TableName() string { + return "video_author" +} + +func (m *VideoAuthor) BeforeCreate(tx *gorm.DB) error { + if m.AuthorId == 0 { + m.AuthorId = global.Snowflake.Generate().Int64() + } + + m.CreatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("CreatedAt", m.CreatedAt) + + m.UpdatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("UpdatedAt", m.UpdatedAt) + + return nil +} diff --git a/api/model/VideoVoteDay.go b/api/model/VideoVoteDay.go new file mode 100644 index 0000000..0a66f82 --- /dev/null +++ b/api/model/VideoVoteDay.go @@ -0,0 +1,36 @@ +package model + +import ( + "gorm.io/gorm" + "time" + "vote-admin-video-api/global" +) + +// VideoVoteDay 视频-每日投票 +type VideoVoteDay struct { + VoteDayId int64 `gorm:"column:vote_day_id;type:bigint(19);primary_key;comment:主键id" json:"vote_day_id"` + VideoId int64 `gorm:"column:video_id;type:bigint(19);comment:视频id;NOT NULL" json:"video_id"` + UserId int64 `gorm:"column:user_id;type:bigint(19);comment:用户id;NOT NULL" json:"user_id"` + VotedAt *LocalTime `gorm:"column:voted_at;type:date;comment:投票时间（日）" json:"voted_at"` + Model + Video *Video `gorm:"foreignKey:VideoId;references:video_id" json:"video"` + User *User `gorm:"foreignKey:UserId;references:user_id" json:"user"` +} + +func (m *VideoVoteDay) TableName() string { + return "video_vote_day" +} + +func (m *VideoVoteDay) BeforeCreate(tx *gorm.DB) error { + if m.VoteDayId == 0 { + m.VoteDayId = global.Snowflake.Generate().Int64() + } + + m.CreatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("CreatedAt", m.CreatedAt) + + m.UpdatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("UpdatedAt", m.UpdatedAt) + + return nil +} diff --git a/api/model/model.go b/api/model/model.go new file mode 100644 index 0000000..58cffea --- /dev/null +++ b/api/model/model.go @@ -0,0 +1,87 @@ +package model + +import ( + "database/sql/driver" + "errors" + "fmt" + "gorm.io/gorm" + "strings" + "time" +) + +type Model struct { + CreatedAt LocalTime `gorm:"column:created_at;type:datetime;comment:创建时间" json:"created_at"` + UpdatedAt LocalTime `gorm:"column:updated_at;type:datetime;comment:修改时间" json:"updated_at"` +} + +// LocalTime 自定义数据类型 +type LocalTime time.Time + +func (t *LocalTime) UnmarshalJSON(data []byte) error { + if string(data) == "null" { + return nil + } + var err error + // 前端接收的时间字符串 + str := string(data) + // 去除接收的str收尾多余的" + timeStr := strings.Trim(str, "\"") + t1, err := time.Parse("2006-01-02 15:04:05", timeStr) + *t = LocalTime(t1) + return err +} + +func (t LocalTime) MarshalJSON() ([]byte, error) { + formatted := fmt.Sprintf("\"%v\"", time.Time(t).Format("2006-01-02 15:04:05")) + return []byte(formatted), nil +} + +func (t LocalTime) Value() (driver.Value, error) { + // MyTime 转换成 time.Time 类型 + tTime := time.Time(t) + return tTime.Format("2006-01-02 15:04:05"), nil +} + +func (t *LocalTime) Scan(v interface{}) error { + switch vt := v.(type) { + case time.Time: + // 字符串转成 time.Time 类型 + *t = LocalTime(vt) + default: + return errors.New("类型处理错误") + } + return nil +} + +func (t *LocalTime) String() string { + return fmt.Sprintf("hhh:%s", time.Time(*t).String()) +} + +func (t *LocalTime) IsEmpty() bool { + return time.Time(*t).IsZero() +} + +func (m *Model) BeforeUpdate(tx *gorm.DB) (err error) { + m.UpdatedAt = LocalTime(time.Now()) + tx.Statement.SetColumn("UpdatedAt", m.UpdatedAt) + + return nil +} + +func Paginate(page, pageSize int) func(db *gorm.DB) *gorm.DB { + return func(db *gorm.DB) *gorm.DB { + if page <= 0 { + page = 1 + } + + switch { + case pageSize > 100: + pageSize = 100 + case pageSize <= 0: + pageSize = 10 + } + + offset := (page - 1) * pageSize + return db.Offset(offset).Limit(pageSize) + } +} diff --git a/api/requests/Article.go b/api/requests/Article.go new file mode 100644 index 0000000..74e480c --- /dev/null +++ b/api/requests/Article.go @@ -0,0 +1,57 @@ +package requests + +type ArticleRequest struct { + GetArticlePage // 获取图文列表-分页 + PutArticle // 修改图文详情 + AddArticle // 新增图文详情 + PutArticleStatus // 操作图文状态 +} + +// GetArticlePage 获取图文列表-分页 +type GetArticlePage struct { + Page int `json:"page" form:"page" label:"页码"` + PageSize int `json:"page_size" form:"page_size" label:"每页个数"` + ArticleStatus *int `json:"article_status" form:"article_status" label:"文章状态"` // （1:正常 2:禁用） + Keyword string `json:"keyword" form:"keyword" label:"搜索关键字"` + Order *GetArticlePageOrder `json:"order" form:"order" label:"排序"` +} + +// GetArticlePageOrder 获取图文列表-分页-排序条件 +type GetArticlePageOrder struct { + VoteNum string `json:"vote_num" form:"vote_num" label:"排序"` // 总票数 +} + +// PutArticle 修改图文详情 +type PutArticle struct { + ArticleTitle string `json:"article_title" form:"article_title" label:"文章标题" validate:"required"` + ArticleStatus int `json:"article_status" form:"article_status" label:"文章状态" validate:"required,oneof=1 2"` // （1:正常 2:禁用） + VoteNum *uint `json:"vote_num" form:"vote_num" label:"票数"` + ArticleContent string `json:"article_content" form:"article_content" label:"文章内容" validate:"required"` + ArticleAuthor []*PutArticleAuthor `json:"article_author" form:"article_author" label:"作者" validate:"required"` +} + +// PutArticleAuthor 修改图文详情-作者 +type PutArticleAuthor struct { + AuthorName string `json:"author_name" form:"author_name" label:"作者姓名" validate:"required"` + HospitalId string `json:"hospital_id" form:"hospital_id" label:"作者所属医院id" validate:"required"` +} + +// AddArticle 新增图文详情 +type AddArticle struct { + ArticleTitle string `json:"article_title" form:"article_title" label:"文章标题" validate:"required"` + ArticleStatus int `json:"article_status" form:"article_status" label:"文章状态" validate:"required,oneof=1 2"` // （1:正常 2:禁用） + VoteNum *uint `json:"vote_num" form:"vote_num" label:"票数"` + ArticleContent string `json:"article_content" form:"article_content" label:"文章内容" validate:"required"` + ArticleAuthor []*PutArticleAuthor `json:"article_author" form:"article_author" label:"作者" validate:"required"` +} + +// AddArticleAuthor 新增图文详情-作者 +type AddArticleAuthor struct { + AuthorName string `json:"author_name" form:"author_name" label:"作者姓名" validate:"required"` + HospitalId string `json:"hospital_id" form:"hospital_id" label:"作者所属医院id" validate:"required"` +} + +// PutArticleStatus 操作图文状态 +type PutArticleStatus struct { + ArticleStatus int `json:"article_status" form:"article_status" label:"文章状态" validate:"required,oneof=1 2"` // （1:正常 2:禁用） +} diff --git a/api/requests/BaseAgreement.go b/api/requests/BaseAgreement.go new file mode 100644 index 0000000..72c2c79 --- /dev/null +++ b/api/requests/BaseAgreement.go @@ -0,0 +1,26 @@ +package requests + +type BaseAgreementRequest struct { + GetBaseAgreementPage // 获取协议列表-分页 + PutBaseAgreement // 修改协议 + AddBaseAgreement // 新增协议 +} + +// GetBaseAgreementPage 获取协议列表-分页 +type GetBaseAgreementPage struct { + Page int `json:"page" form:"page" label:"页码"` + PageSize int `json:"page_size" form:"page_size" label:"每页个数"` +} + +// PutBaseAgreement 修改协议 +type PutBaseAgreement struct { + AgreementTitle string `json:"agreement_title" form:"agreement_title" label:"协议标题"` + AgreementContent string `json:"agreement_content" form:"agreement_content" label:"协议内容"` +} + +// AddBaseAgreement 新增协议 +type AddBaseAgreement struct { + AgreementTitle string `json:"agreement_title" form:"agreement_title" label:"协议标题"` + AgreementType int `json:"agreement_type" form:"agreement_type" label:"协议类型" validate:"required,oneof=1 2"` // （1:大赛介绍 2:投票规则） + AgreementContent string `json:"agreement_content" form:"agreement_content" label:"协议内容"` +} diff --git a/api/requests/BaseHospital.go b/api/requests/BaseHospital.go new file mode 100644 index 0000000..9c6bf4c --- /dev/null +++ b/api/requests/BaseHospital.go @@ -0,0 +1,18 @@ +package requests + +type BaseHospitalRequest struct { + GetBaseHospitalList // 获取医院列表 +} + +// GetBaseHospitalList 获取医院列表 +type GetBaseHospitalList struct { + HospitalName string `json:"hospital_name" form:"hospital_name" label:"医院名称"` + HospitalLevelName string `json:"hospital_level_name" form:"hospital_level_name" label:"医院等级名称"` + HospitalStatus int `json:"hospital_status" form:"hospital_status" label:"状态"` // 状态（0:禁用 1:正常 2:删除） + ProvinceId int `json:"province_id" form:"province_id" label:"省份id"` + Province string `json:"province" form:"province" label:"省份"` + CityId int `json:"city_id" form:"city_id" label:"城市id"` + City string `json:"city" form:"city" label:"城市"` + CountyId int `json:"county_id" form:"county_id" label:"区县id"` + County string `json:"county" form:"county" label:"区县"` +} diff --git a/api/requests/Basic.go b/api/requests/Basic.go new file mode 100644 index 0000000..d6e7b20 --- /dev/null +++ b/api/requests/Basic.go @@ -0,0 +1,10 @@ +package requests + +type BasicRequest struct { + GetBasicAgreement // 获取协议详情 +} + +// GetBasicAgreement 获取协议详情 +type GetBasicAgreement struct { + AgreementType int `json:"agreement_type" form:"agreement_type" label:"协议类型" validate:"required,oneof=1 2"` // 协议类型（1:大赛介绍 2:投票规则） +} diff --git a/api/requests/Public.go b/api/requests/Public.go new file mode 100644 index 0000000..f8a7c09 --- /dev/null +++ b/api/requests/Public.go @@ -0,0 +1,21 @@ +package requests + +type PublicRequest struct { + Login // 登陆 + GetIndexData // 首页动态统计数据 +} + +// Login 登陆 +type Login struct { + Access string `json:"access" form:"access" validate:"required" label:"用户名"` // 用户名 + Password string `json:"password" form:"password" validate:"required" label:"密码"` // 密码 + Captcha string `json:"captcha" form:"captcha" validate:"required" label:"验证码"` // 验证码 + CaptchaId string `json:"captchaId" form:"captchaId" validate:"required"` // 验证码ID +} + +// GetIndexData 首页动态统计数据 +type GetIndexData struct { + Type int `json:"type" form:"type" validate:"required,oneof=1 2 3" label:"分类"` // 分类（1:新增用户数 2:图文投票数 3:视频投票数） + StartTime string `json:"start_time" form:"start_time" validate:"required" label:"开始时间"` + EndTime string `json:"end_time" form:"end_time" validate:"required" label:"结束时间"` +} diff --git a/api/requests/System.go b/api/requests/System.go new file mode 100644 index 0000000..c1b4d61 --- /dev/null +++ b/api/requests/System.go @@ -0,0 +1,11 @@ +package requests + +type SystemRequest struct { + PutSystemTime // 修改投票时间 +} + +// PutSystemTime 修改投票时间 +type PutSystemTime struct { + StartTime string `json:"start_time" form:"start_time" label:"开始投票时间" validate:"required"` + EndTime string `json:"end_time" form:"end_time" label:"结束投票时间" validate:"required"` +} diff --git a/api/requests/User.go b/api/requests/User.go new file mode 100644 index 0000000..394ccf6 --- /dev/null +++ b/api/requests/User.go @@ -0,0 +1,21 @@ +package requests + +type UserRequest struct { + GetUserPage // 获取用户列表-分页 + PutUserStatus // 操作用户状态 +} + +// GetUserPage 获取用户列表-分页 +type GetUserPage struct { + Page int `json:"page" form:"page" label:"页码"` + PageSize int `json:"page_size" form:"page_size" label:"每页个数"` + UserId string `json:"user_id" form:"user_id" label:"用户id"` + UserStatus int `json:"user_status" form:"user_status" label:"状态"` + AppIden string `json:"app_iden" form:"app_iden" label:"app唯一标识"` + OpenId string `json:"open_id" form:"open_id" label:"用户微信标识"` +} + +// PutUserStatus 操作用户状态 +type PutUserStatus struct { + UserStatus int `json:"user_status" form:"user_status" label:"删除状态" validate:"required,oneof=1 2"` // 状态（1:正常 2:禁用） +} diff --git a/api/requests/UserVoteDay.go b/api/requests/UserVoteDay.go new file mode 100644 index 0000000..e98eb9d --- /dev/null +++ b/api/requests/UserVoteDay.go @@ -0,0 +1,36 @@ +package requests + +type UserVoteDayRequest struct { + GetArticleUserVoteDayPage // 用户投票记录列表-图文-分页 + GetVideoUserVoteDayPage // 用户投票记录列表-视频-分页 + GetArticleVoteDayPage // 投票记录列表-图文-分页 + GetVideoVoteDayPage // 投票记录列表-视频-分页 +} + +// GetArticleUserVoteDayPage 用户投票记录列表-图文-分页 +type GetArticleUserVoteDayPage struct { + Page int `json:"page" form:"page" label:"页码"` + PageSize int `json:"page_size" form:"page_size" label:"每页个数"` + UserId string `json:"user_id" form:"user_id" label:"用户id" validate:"required"` +} + +// GetVideoUserVoteDayPage 用户投票记录列表-视频-分页 +type GetVideoUserVoteDayPage struct { + Page int `json:"page" form:"page" label:"页码"` + PageSize int `json:"page_size" form:"page_size" label:"每页个数"` + UserId string `json:"user_id" form:"user_id" label:"用户id" validate:"required"` +} + +// GetArticleVoteDayPage 投票记录列表-图文-分页 +type GetArticleVoteDayPage struct { + Page int `json:"page" form:"page" label:"页码"` + PageSize int `json:"page_size" form:"page_size" label:"每页个数"` + ArticleId string `json:"article_id" form:"article_id" label:"文章id" validate:"required"` +} + +// GetVideoVoteDayPage 投票记录列表-视频-分页 +type GetVideoVoteDayPage struct { + Page int `json:"page" form:"page" label:"页码"` + PageSize int `json:"page_size" form:"page_size" label:"每页个数"` + VideoId string `json:"video_id" form:"video_id" label:"视频id" validate:"required"` +} diff --git a/api/requests/Video.go b/api/requests/Video.go new file mode 100644 index 0000000..86d12c4 --- /dev/null +++ b/api/requests/Video.go @@ -0,0 +1,60 @@ +package requests + +type VideoRequest struct { + GetVideoPage // 获取视频列表-分页 + PutVideo // 修改视频详情 + AddVideo // 新增视频详情 + PutVideoStatus // 操作视频状态 +} + +// GetVideoPage 获取视频列表-分页 +type GetVideoPage struct { + Page int `json:"page" form:"page" label:"页码"` + PageSize int `json:"page_size" form:"page_size" label:"每页个数"` + Keyword string `json:"keyword" form:"keyword" label:"搜索关键字"` + VideoStatus *int `json:"video_status" form:"Video_status" label:"视频状态"` // （1:正常 2:禁用） + VideoNo string `json:"video_no" form:"video_no" label:"视频编号"` // （保利） + Order *GetVideoPageOrder `json:"order" form:"order" label:"排序"` +} + +// GetVideoPageOrder 获取视频列表-分页-排序条件 +type GetVideoPageOrder struct { + VoteNum string `json:"vote_num" form:"vote_num" label:"排序"` // 总票数 +} + +// PutVideo 修改视频详情 +type PutVideo struct { + VideoTitle string `json:"video_title" form:"video_title" label:"视频标题" validate:"required"` + VideoStatus int `json:"video_status" form:"video_status" label:"视频状态" validate:"required,oneof=1 2"` // （1:正常 2:禁用） + VoteNum *uint `json:"vote_num" form:"vote_num" label:"总票数"` + VideoNo string `json:"video_no" form:"video_no" label:"视频编号" validate:"required"` // （保利） + VideoContent string `json:"video_content" form:"video_content" label:"视频内容"` + VideoAuthor []*PutVideoAuthor `json:"video_author" form:"video_author" label:"作者" validate:"required"` +} + +// PutVideoAuthor 修改视频详情-作者 +type PutVideoAuthor struct { + AuthorName string `json:"author_name" form:"author_name" label:"作者姓名" validate:"required"` + HospitalId string `json:"hospital_id" form:"hospital_id" label:"作者所属医院id" validate:"required"` +} + +// AddVideo 新增视频详情 +type AddVideo struct { + VideoTitle string `json:"video_title" form:"video_title" label:"视频标题" validate:"required"` + VideoStatus int `json:"video_status" form:"video_status" label:"视频状态" validate:"required,oneof=1 2"` // （1:正常 2:禁用） + VoteNum *uint `json:"vote_num" form:"vote_num" label:"总票数"` + VideoNo string `json:"video_no" form:"video_no" label:"视频编号" validate:"required"` // （保利） + VideoContent string `json:"video_content" form:"video_content" label:"视频内容"` + VideoAuthor []*PutVideoAuthor `json:"video_author" form:"video_author" label:"作者" validate:"required"` +} + +// AddVideoAuthor 新增视频详情-作者 +type AddVideoAuthor struct { + AuthorName string `json:"author_name" form:"author_name" label:"作者姓名" validate:"required"` + HospitalId string `json:"hospital_id" form:"hospital_id" label:"作者所属医院id" validate:"required"` +} + +// PutVideoStatus 操作视频状态 +type PutVideoStatus struct { + VideoStatus int `json:"video_status" form:"video_status" label:"状态" validate:"required,oneof=1 2"` // （1:正常 2:禁用） +} diff --git a/api/requests/base.go b/api/requests/base.go new file mode 100644 index 0000000..309b903 --- /dev/null +++ b/api/requests/base.go @@ -0,0 +1,4 @@ +package requests + +type Requests struct { +} diff --git a/api/responses/responses.go b/api/responses/responses.go new file mode 100644 index 0000000..201e252 --- /dev/null +++ b/api/responses/responses.go @@ -0,0 +1,52 @@ +package responses + +import ( + "github.com/gin-gonic/gin" + "net/http" + "vote-admin-video-api/consts" +) + +type res struct { + Code int `json:"code"` + Data interface{} `json:"data"` + Message string `json:"message"` +} + +func result(code int, data interface{}, msg string, c *gin.Context) { + //if data == nil { + // data = gin.H{} + //} + c.JSON(http.StatusOK, res{ + code, + data, + msg, + }) +} + +func Ok(c *gin.Context) { + result(consts.HttpSuccess, map[string]interface{}{}, "成功", c) +} + +func OkWithMessage(message string, c *gin.Context) { + result(consts.HttpSuccess, map[string]interface{}{}, message, c) +} + +func OkWithData(data interface{}, c *gin.Context) { + result(consts.HttpSuccess, data, "成功", c) +} + +func OkWithDetailed(data interface{}, message string, c *gin.Context) { + result(consts.HttpSuccess, data, message, c) +} + +func Fail(c *gin.Context) { + result(consts.HttpError, map[string]interface{}{}, "失败", c) +} + +func FailWithMessage(message string, c *gin.Context) { + result(consts.HttpError, map[string]interface{}{}, message, c) +} + +func FailWithDetailed(data interface{}, message string, c *gin.Context) { + result(consts.HttpError, data, message, c) +} diff --git a/api/router/router.go b/api/router/router.go new file mode 100644 index 0000000..8e2b3e2 --- /dev/null +++ b/api/router/router.go @@ -0,0 +1,229 @@ +package router + +import ( + "fmt" + "github.com/gin-gonic/gin" + "net/http" + "vote-admin-video-api/api/controller" + "vote-admin-video-api/api/exception" + "vote-admin-video-api/api/middlewares" + "vote-admin-video-api/config" + "vote-admin-video-api/consts" +) + +// Init 初始化路由 +func Init() *gin.Engine { + r := gin.New() + + // 环境设置 + if config.C.Env == "prod" { + gin.SetMode(gin.ReleaseMode) + } + + // 获取请求参数中间件-json格式下会导致接口获取不到请求数据 + r.Use(middlewares.RequestParamsMiddleware()) + + // 日志中间件 + r.Use(middlewares.Logrus()) + + // 异常 + r.Use(gin.Recovery()) + + // 404处理 + r.NoRoute(func(c *gin.Context) { + path := c.Request.URL.Path + method := c.Request.Method + c.JSON(http.StatusNotFound, gin.H{ + "msg": fmt.Sprintf("%s %s not found", method, path), + "code": consts.ClientHttpNotFound, + "data": "", + }) + }) + + // 异常处理 + r.Use(exception.Recover()) + + // 跨域处理 + r.Use(middlewares.Cors()) + + // 加载基础路由 + api := controller.Api{} + + // 公开路由-不验证权限 + publicRouter(r, api) + + // 验证jwt + r.Use(middlewares.Jwt()) + + // 验证权限 + r.Use(middlewares.Auth()) + + // 私有路由-验证权限 + privateRouter(r, api) + + // 公共路由-验证权限 + adminRouter(r, api) + + // 基础数据-验证权限 + basicRouter(r, api) + + return r +} + +// publicRouter 公开路由-不验证权限 +func publicRouter(r *gin.Engine, api controller.Api) { + adminGroup := r.Group("/admin") + + // 登录 + adminGroup.POST("/login", api.Public.Login) + + // 验证码 + adminGroup.GET("/captcha", api.Public.GetCaptcha) + + // 配置 + systemGroup := adminGroup.Group("/system") + { + // 编辑器配置 + editorGroup := systemGroup.Group("/editor") + { + // 编辑器-获取配置 + editorGroup.GET("", api.Editor.GetEditorConfig) + + // 编辑器-上传 + editorGroup.POST("", api.Editor.EditorUpload) + } + } +} + +// adminRouter 公共路由-验证权限 +func adminRouter(r *gin.Engine, api controller.Api) { + adminGroup := r.Group("/admin") + + // 首页 + indexGroup := adminGroup.Group("/index") + { + // 首页 + indexGroup.GET("", api.Public.GetIndex) + + // 首页动态统计数据 + indexGroup.GET("/data", api.Public.GetIndexData) + } +} + +// basicRouter 基础数据-验证权限 +func basicRouter(r *gin.Engine, api controller.Api) { + +} + +// privateRouter 私有路由-验证权限 +func privateRouter(r *gin.Engine, api controller.Api) { + adminGroup := r.Group("/admin") + + // 图文 + articleGroup := adminGroup.Group("/article") + { + // 获取图文列表-分页 + articleGroup.POST("/page", api.Article.GetArticlePage) + + // 获取图文详情 + articleGroup.GET("/:article_id", api.Article.GetArticle) + + // 修改图文详情 + articleGroup.PUT("/:article_id", api.Article.PutArticle) + + // 新增图文详情 + articleGroup.POST("", api.Article.AddArticle) + + // 操作图文状态 + articleGroup.PUT("/status/:article_id", api.Article.PutArticleStatus) + + // 投票记录列表-图文-分页 + articleGroup.GET("/vote/page", api.UserVoteDay.GetArticleVotePage) + } + + // 视频 + videoGroup := adminGroup.Group("/video") + { + // 获取视频列表-分页 + videoGroup.POST("/page", api.Video.GetVideoPage) + + // 获取视频详情 + videoGroup.GET("/:video_id", api.Video.GetVideo) + + // 修改视频详情 + videoGroup.PUT("/:video_id", api.Video.PutVideo) + + // 新增视频详情 + videoGroup.POST("", api.Video.AddVideo) + + // 操作视频状态 + videoGroup.PUT("/status/:video_id", api.Video.PutVideoStatus) + + // 投票记录列表-视频-分页 + videoGroup.GET("/vote/page", api.UserVoteDay.GetVideoVotePage) + } + + // 基础数据 + basicGroup := adminGroup.Group("/basic") + { + // 协议 + agreementGroup := basicGroup.Group("/agreement") + { + // 获取协议列表-分页 + agreementGroup.GET("/page", api.BaseAgreement.GetBaseAgreementPage) + + // 获取协议详情 + agreementGroup.GET("/:agreement_id", api.BaseAgreement.GetBaseAgreement) + + // 修改协议 + agreementGroup.PUT("/:agreement_id", api.BaseAgreement.PutBaseAgreement) + + // 新增协议 + agreementGroup.POST("", api.BaseAgreement.AddBaseAgreement) + } + + // 医院管理 + hospitalGroup := basicGroup.Group("/hospital") + { + // 获取医院列表-限制条数 + hospitalGroup.GET("/page", api.BaseHospital.GetHospitalList) + } + } + + // 用户 + userGroup := adminGroup.Group("/user") + { + // 获取用户列表-分页 + userGroup.GET("/page", api.User.GetUserPage) + + // 获取用户详情 + userGroup.GET("/:user_id", api.User.GetUser) + + // 操作用户状态 + userGroup.PUT("/status/:user_id", api.User.PutUserStatus) + + // 用户投票记录 + voteGroup := userGroup.Group("/vote") + { + // 用户投票记录列表-图文-分页 + voteGroup.GET("/article/page", api.UserVoteDay.GetUserArticleVotePage) + + // 用户投票记录列表-视频-分页 + voteGroup.GET("/video/page", api.UserVoteDay.GetUserVideoVotePage) + } + } + + // 配置 + systemGroup := adminGroup.Group("/system") + { + // 投票时间 + timeGroup := systemGroup.Group("/time") + { + // 获取投票时间详情 + timeGroup.GET("/:system_time_id", api.System.GetSystemTime) + + // 修改投票时间 + timeGroup.PUT("/:system_time_id", api.System.PutSystemTime) + } + } +} diff --git a/api/service/Public.go b/api/service/Public.go new file mode 100644 index 0000000..53099e8 --- /dev/null +++ b/api/service/Public.go @@ -0,0 +1,17 @@ +package service + +import ( + "net/http" +) + +type PublicService struct { +} + +// GetUserIP 获取用户ip +func (r *PublicService) GetUserIP(h *http.Request) string { + forwarded := h.Header.Get("X-FORWARDED-FOR") + if forwarded != "" { + return forwarded + } + return h.RemoteAddr +} diff --git a/api/service/SystemTime.go b/api/service/SystemTime.go new file mode 100644 index 0000000..5cfd164 --- /dev/null +++ b/api/service/SystemTime.go @@ -0,0 +1,49 @@ +package service + +import ( + "context" + "time" + "vote-admin-video-api/api/dao" + "vote-admin-video-api/global" +) + +type SystemTimeService struct { +} + +// CheckVoteValidStatus 检测投票有效期 +// bool true:未结束 false:已结束 +func (r *SystemTimeService) CheckVoteValidStatus() bool { + redisKey := "VoteSystemTime" + res, _ := global.Redis.Get(context.Background(), redisKey).Result() + if res == "" { + // 获取配置-时间 + systemTimeDao := dao.SystemTimeDao{} + systemTime, err := systemTimeDao.GetSystemTimeById(1) + if err != nil { + return false + } + + if systemTime.EndTime == nil { + return false + } + + // 结束时间 + endTime := time.Time(*systemTime.EndTime) + + // 当前时间 + now := time.Now() + + duration := endTime.Sub(now) + if duration < 0 { + return false + } + + // 添加缓存 + _, err = global.Redis.Set(context.Background(), redisKey, "1", duration).Result() + if err != nil { + return false + } + } + + return true +} diff --git a/api/service/User.go b/api/service/User.go new file mode 100644 index 0000000..be3a40a --- /dev/null +++ b/api/service/User.go @@ -0,0 +1,109 @@ +package service + +import ( + "context" + "fmt" + "time" + "vote-admin-video-api/config" + "vote-admin-video-api/global" +) + +type UserService struct { +} + +// AddUserVoteDayCache 新增用户每日投票记录缓存 +// t 1:文章 2:视频 +func (r *UserService) AddUserVoteDayCache(userId, id int64, t int) bool { + now := time.Now() + + redisKey := "UserVoteDay" + fmt.Sprintf("%d", userId) + fmt.Sprintf("%d", id) + now.Format("2006-01-02") + fmt.Sprintf("%d", t) + + // 缓存过期时间 + year, month, day := now.Date() + location := now.Location() + validTime := time.Date(year, month, day, 23, 59, 59, 0, location) + duration := validTime.Sub(now) + if duration < 0 { + return false + } + + if config.C.Env == "dev" { + duration = 60 * 5 * time.Second + } + + // 添加缓存 + _, err := global.Redis.Set(context.Background(), redisKey, "1", duration).Result() + if err != nil { + return false + } + + return true +} + +// CheckUserVoteDay 检测用户今日是否投票 +// t 1:文章 2:视频 +func (r *UserService) CheckUserVoteDay(userId, id int64, t int) bool { + if t != 1 && t != 2 { + return false + } + + now := time.Now() + redisKey := "UserVoteDay" + fmt.Sprintf("%d", userId) + fmt.Sprintf("%d", id) + now.Format("2006-01-02") + fmt.Sprintf("%d", t) + res, _ := global.Redis.Get(context.Background(), redisKey).Result() + if res == "" { + return false + } + + if res == "1" { + return true + } + + return false + //if res == "" { + // // 是否已投票（1:已投票 0:未投票） + // isVote := "0" + // if t == 1 { + // // 文章 + // maps := make(map[string]interface{}) + // maps["article_id"] = id + // maps["user_id"] = userId + // maps["voted_at"] = now.Format("2006-01-02") + // + // articleVoteDayDao := dao.ArticleVoteDayDao{} + // articleVoteDay, _ := articleVoteDayDao.GetArticleVoteDayById(id) + // if articleVoteDay != nil { + // isVote = "1" + // } + // } + // + // if t == 2 { + // // 视频 + // maps := make(map[string]interface{}) + // maps["article_id"] = id + // maps["user_id"] = userId + // maps["voted_at"] = now.Format("2006-01-02") + // + // videoVoteDayDao := dao.VideoVoteDayDao{} + // videoVoteDay, _ := videoVoteDayDao.GetVideoVoteDayById(id) + // if videoVoteDay != nil { + // isVote = "1" + // } + // } + // + // // 添加缓存 + // if isVote == "1" { + // result := r.AddUserVoteDayCache(userId, id, t) + // if result == false { + // return false + // } + // } + // + // res = isVote + //} + // + //if res == "0" { + // return false + //} else { + // return true + //} +} diff --git a/config.yaml b/config.yaml new file mode 100644 index 0000000..513998b --- /dev/null +++ b/config.yaml @@ -0,0 +1,65 @@ +port: 8508 # 启动端口 + +env: dev # 环境配置 + +snowflake: 1 # 雪花算法机器id + +# [数据库配置] +mysql: + host: '42.193.16.243' + username: root + password: 'gdxz123456^%$d' + port: 30001 + db-name: vote-video + max-idle-cons: 5 + max-open-cons: 20 + debug: true + +log: + file-path: "./log/" + file-name: "vote-admin-video-api.log" + +# [redis] +redis: + host: '139.155.127.177' + port: 30002 + password: gdxz2022&dj. + pool-size: 100 + db: 5 + +# [jwt] +jwt: + sign-key: 123456899 # 私钥 + ttl : 24 # 过期时间 小时 + algo : HS256 # 加密方式 + +oss: + oss-access-key: LTAI5tKmFrVCghcxX7yHyGhm + oss-access-key-secret: q1aiIZCJJuf92YbKk2cSXnPES4zx26 + oss-bucket: dev-vote-system + oss-endpoint: oss-cn-beijing.aliyuncs.com + oss-custom-domain-name: https://dev-vote-system.oss-cn-beijing.aliyuncs.com + +# [阿里大鱼短信] +dysms: + dysms-access-key: LTAI4GGygjsKhyBwvvC3CghV + dysms-access-secret: rcx7lO9kQxG10m8NqNPEfEtT9IS8EI + +# [微信] +wechat: + app-id: wx68affaa9d23528f8 + app-secret: 2963c90242ddb2421c939591ad9e903d + test-pay-app-id: wxa4132ef4701ac5e4 + single-pay-notify-url: callback/wxpay/single # 单项支付回调地址 + single-refund-notify-url: callback/wxpay/single/refund # 单项退款回调地址 + member-pay-notify-url: callback/wxpay/member # 会员支付回调地址 + member-refund-notify-url: callback/wxpay/member/refund # 会员退款回调地址 + notify-domain: https://dev-hepa.igandan.com/api/ + pay-1281030301: + mch-id: 1281030301 + v3-api-secret: sB2tCkT70uwEy7cQCu1llA6nilTbek6F + mch-certificate-serial-number: 3D1685894CEDD41753A470211F975D40AE1375F5 + platform-certs: extend/weChat/certs/1281030301/wechatpay_5B5C8A69CC86D1127F6B6AA06AAAF10531EEFE90.pem + private-key: extend/weChat/certs/1281030301/apiclient_key.pem + certificate: extend/weChat/certs/1281030301/apiclient_cert.pem + diff --git a/config/config.go b/config/config.go new file mode 100644 index 0000000..5785c32 --- /dev/null +++ b/config/config.go @@ -0,0 +1,16 @@ +package config + +var C Config + +type Config struct { + Port int `mapstructure:"port" json:"port" yaml:"port"` + Env string `mapstructure:"env" json:"Env" yaml:"Env"` + Mysql Mysql `mapstructure:"mysql" json:"mysql" yaml:"mysql"` + Log Log `mapstructure:"log" json:"log" yaml:"log"` + Redis Redis `mapstructure:"redis" json:"redis" yaml:"redis"` + Jwt Jwt `mapstructure:"jwt" json:"jwt" yaml:"jwt"` + Oss Oss `mapstructure:"oss" json:"oss" yaml:"oss"` + Snowflake int64 `mapstructure:"snowflake" json:"snowflake" yaml:"snowflake"` + Dysms Dysms `mapstructure:"dysms" json:"dysms" yaml:"dysms"` + Wechat Wechat `mapstructure:"wechat" json:"wechat" yaml:"wechat"` +} diff --git a/config/dysms.go b/config/dysms.go new file mode 100644 index 0000000..7d02d49 --- /dev/null +++ b/config/dysms.go @@ -0,0 +1,6 @@ +package config + +type Dysms struct { + DysmsAccessKey string `mapstructure:"dysms-access-key" json:"dysms-access-key" yaml:"dysms-access-key"` + DysmsAccessSecret string `mapstructure:"dysms-access-secret" json:"dysms-access-secret" yaml:"dysms-access-secret"` +} diff --git a/config/jwt.go b/config/jwt.go new file mode 100644 index 0000000..9f59bed --- /dev/null +++ b/config/jwt.go @@ -0,0 +1,7 @@ +package config + +type Jwt struct { + SignKey string `mapstructure:"sign-key" json:"sign-key" yaml:"sign-key"` // 私钥 + Ttl int `mapstructure:"ttl" json:"ttl" yaml:"ttl"` // 过期时间 小时 + Algo string `mapstructure:"algo" json:"algo" yaml:"algo"` // 加密方式 +} diff --git a/config/log.go b/config/log.go new file mode 100644 index 0000000..ae86ff5 --- /dev/null +++ b/config/log.go @@ -0,0 +1,6 @@ +package config + +type Log struct { + FilePath string `mapstructure:"file-path" json:"file-path" yaml:"file-path"` // 日志目录 + FileName string `mapstructure:"file-name" json:"file-name" yaml:"file-name"` // 日志名称 +} diff --git a/config/mysql.go b/config/mysql.go new file mode 100644 index 0000000..5135c8e --- /dev/null +++ b/config/mysql.go @@ -0,0 +1,12 @@ +package config + +type Mysql struct { + Host string `mapstructure:"host" json:"host" yaml:"host"` // 服务器地址 + Port int `mapstructure:"port" json:"port" yaml:"port"` // 端口 + DbName string `mapstructure:"db-name" json:"db-name" yaml:"db-name"` // 数据库名 + Username string `mapstructure:"username" json:"username" yaml:"username"` // 数据库用户名 + Password string `mapstructure:"password" json:"password" yaml:"password"` // 数据库密码 + MaxIdleConns int `mapstructure:"max-idle-cons" json:"MaxIdleConns" yaml:"max-idle-cons"` // 空闲中的最大连接数 + MaxOpenConns int `mapstructure:"max-open-cons" json:"MaxOpenConns" yaml:"max-open-cons"` // 打开到数据库的最大连接数 + Debug bool `mapstructure:"debug" json:"debug" yaml:"debug"` // 是否开启Gorm全局日志 +} diff --git a/config/oss.go b/config/oss.go new file mode 100644 index 0000000..7c508b4 --- /dev/null +++ b/config/oss.go @@ -0,0 +1,9 @@ +package config + +type Oss struct { + OssAccessKey string `mapstructure:"oss-access-key" json:"oss-access-key" yaml:"oss-access-key"` + OssAccessKeySecret string `mapstructure:"oss-access-key-secret" json:"oss-access-key-secret" yaml:"oss-access-key-secret"` + OssBucket string `mapstructure:"oss-bucket" json:"oss-bucket" yaml:"oss-bucket"` + OssEndpoint string `mapstructure:"oss-endpoint" json:"oss-endpoint" yaml:"oss-endpoint"` + OssCustomDomainName string `mapstructure:"oss-custom-domain-name" json:"oss-custom-domain-name" yaml:"oss-custom-domain-name"` +} diff --git a/config/redis.go b/config/redis.go new file mode 100644 index 0000000..324aaba --- /dev/null +++ b/config/redis.go @@ -0,0 +1,9 @@ +package config + +type Redis struct { + Host string `mapstructure:"host" json:"host" yaml:"host"` // 服务器地址 + Port int `mapstructure:"port" json:"port" yaml:"port"` // 服务器端口 + Password string `mapstructure:"password" json:"password" yaml:"password"` // 密码 + PoolSize int `mapstructure:"pool-size" json:"pool-size" yaml:"pool-size"` // 连接池大小 + Db int `mapstructure:"db" json:"db" yaml:"db"` // 数据库 +} diff --git a/config/wechat.go b/config/wechat.go new file mode 100644 index 0000000..9a7fd05 --- /dev/null +++ b/config/wechat.go @@ -0,0 +1,23 @@ +package config + +type Wechat struct { + AppId string `mapstructure:"app-id" json:"app-id" yaml:"app-id"` + AppSecret string `mapstructure:"app-secret" json:"app-secret" yaml:"app-secret"` + TestPayAppId string `mapstructure:"test-pay-app-id" json:"test-pay-app-id" yaml:"test-pay-app-id"` + SinglePayNotifyUrl string `mapstructure:"single-pay-notify-url" json:"single-pay-notify-url" yaml:"single-pay-notify-url"` // 单项支付回调地址 + SingleRefundNotifyUrl string `mapstructure:"single-refund-notify-url" json:"single-refund-notify-url" yaml:"single-refund-notify-url"` // 单项退款回调地址 + MemberPayNotifyUrl string `mapstructure:"member-pay-notify-url" json:"member-pay-notify-url" yaml:"member-pay-notify-url"` // 会员支付回调地址 + MemberRefundNotifyUrl string `mapstructure:"member-refund-notify-url" json:"member-refund-notify-url" yaml:"member-refund-notify-url"` // 会员退款回调地址 + NotifyDomain string `mapstructure:"notify-domain" json:"notify-domain" yaml:"notify-domain"` + Pay1281030301 Pay1281030301 `mapstructure:"pay-1281030301" json:"pay-1281030301" yaml:"pay-1281030301"` +} + +// Pay1281030301 app +type Pay1281030301 struct { + MchId string `mapstructure:"mch-id" json:"mch-id" yaml:"mch-id"` // 商户号 + V3ApiSecret string `mapstructure:"v3-api-secret" json:"v3-api-secret" yaml:"v3-api-secret"` // 商户APIv3密钥 + MchCertificateSerialNumber string `mapstructure:"mch-certificate-serial-number" json:"mch-certificate-serial-number" yaml:"mch-certificate-serial-number"` // 商户证书序列号 + PlatformCerts string `mapstructure:"platform-certs" json:"platform-certs" yaml:"platform-certs"` // 平台证书 + PrivateKey string `mapstructure:"private-key" json:"private-key" yaml:"private-key"` + Certificate string `mapstructure:"certificate" json:"certificate" yaml:"certificate"` +} diff --git a/consts/http.go b/consts/http.go new file mode 100644 index 0000000..00614e5 --- /dev/null +++ b/consts/http.go @@ -0,0 +1,24 @@ +package consts + +// 业务状态码 +const ( + HttpSuccess = 200 // 成功 + + HttpError = -1 // 失败 + + UserStatusError = 201 // 用户状态异常 + + ClientHttpError = 400 // 错误请求 + + ClientHttpUnauthorized = 401 // 未授权 + + HttpProhibit = 403 // 禁止请求 + + ClientHttpNotFound = 404 // 资源未找到 + + TokenError = 405 // token错误/无效 + + TokenExptired = 406 // token过期 + + ServerError = 500 // 服务器异常 +) diff --git a/core/logrus.go b/core/logrus.go new file mode 100644 index 0000000..9247baa --- /dev/null +++ b/core/logrus.go @@ -0,0 +1,46 @@ +package core + +import ( + "github.com/sirupsen/logrus" + "os" + "path" + "path/filepath" + "vote-admin-video-api/config" + "vote-admin-video-api/global" +) + +// Logrus 日志记录到文件 +func Logrus() *logrus.Logger { + // 日志文件 + fileName := path.Join(config.C.Log.FilePath, config.C.Log.FileName) + + // 获取文件夹路径 + dirPath := filepath.Dir(fileName) + + // 创建文件夹（如果不存在） + err := os.MkdirAll(dirPath, os.ModePerm) + if err != nil { + panic("初始化日志文件失败") + } + + // 写入文件 + src, err := os.OpenFile(fileName, os.O_CREATE|os.O_APPEND|os.O_RDWR, os.ModePerm) + if err != nil { + panic("初始化日志文件失败") + } + + global.Logger = logrus.New() + + // 设置输出 + global.Logger.Out = src + + // 设置日志级别 + global.Logger.SetLevel(logrus.DebugLevel) + + // 设置日志格式 + global.Logger.SetFormatter(&logrus.TextFormatter{ + TimestampFormat: "2006-01-02 15:04:05", + }) + + return global.Logger +} diff --git a/core/mysql.go b/core/mysql.go new file mode 100644 index 0000000..0e71475 --- /dev/null +++ b/core/mysql.go @@ -0,0 +1,56 @@ +package core + +import ( + "fmt" + "gorm.io/driver/mysql" + "gorm.io/gorm" + "gorm.io/gorm/logger" + "time" + "vote-admin-video-api/config" + "vote-admin-video-api/global" +) + +func Mysql() { + var err error + + m := config.C.Mysql + dsn := fmt.Sprintf("%s:%s@(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local&timeout=%s", m.Username, + m.Password, m.Host, m.Port, m.DbName, "10s") + + // newLogger := logger.New( + // global.Logger, + // logger.Config{ + // SlowThreshold: time.Second, // Slow SQL threshold + // LogLevel: logger.Info, // Log level + // IgnoreRecordNotFoundError: true, // Ignore ErrRecordNotFound error for logger + // ParameterizedQueries: false, // Don't include params in the SQL log + // Colorful: false, // Disable color + // }, + // ) + + global.Db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{ + Logger: logger.Default.LogMode(logger.Info), + // Logger: newLogger, + }) + + if err != nil { + fmt.Println(err.Error()) + panic(err.Error()) + } + + sqlDB, _ := global.Db.DB() + + // SetMaxIdleConns 设置空闲连接池中连接的最大数量 + sqlDB.SetMaxIdleConns(m.MaxIdleConns) + + // SetMaxOpenConns 设置打开数据库连接的最大数量。 + sqlDB.SetMaxOpenConns(m.MaxOpenConns) + + // SetConnMaxLifetime 设置了连接可复用的最大时间。 + sqlDB.SetConnMaxLifetime(time.Hour) + + // 调试模式 + //Db.LogMode(m.Debug) // 打印sql + // Db.SingularTable(true) // 全局禁用表名复数 + fmt.Println("初始化数据库成功......") +} diff --git a/core/redis.go b/core/redis.go new file mode 100644 index 0000000..cc09103 --- /dev/null +++ b/core/redis.go @@ -0,0 +1,25 @@ +package core + +import ( + "context" + "fmt" + "github.com/go-redis/redis/v8" + "strconv" + "vote-admin-video-api/config" + "vote-admin-video-api/global" +) + +// Redis redis缓存 +func Redis() { + global.Redis = redis.NewClient(&redis.Options{ + Addr: config.C.Redis.Host + ":" + strconv.Itoa(config.C.Redis.Port), + Password: config.C.Redis.Password, + DB: config.C.Redis.Db, + PoolSize: config.C.Redis.PoolSize, + }) + _, err := global.Redis.Ping(context.Background()).Result() + if err != nil { + panic("redis初始化失败! " + err.Error()) + } + fmt.Println("初始化redis成功......") +} diff --git a/core/snowflake.go b/core/snowflake.go new file mode 100644 index 0000000..c911a29 --- /dev/null +++ b/core/snowflake.go @@ -0,0 +1,21 @@ +package core + +import ( + "fmt" + "github.com/bwmarrin/snowflake" + "vote-admin-video-api/config" + "vote-admin-video-api/global" +) + +// Snowflake 雪花算法 +func Snowflake() { + // 创建雪花算法实例 + node, err := snowflake.NewNode(config.C.Snowflake) + if err != nil { + panic("snowflake初始化失败! " + err.Error()) + } + + global.Snowflake = node + + fmt.Println("初始化snowflake成功......") +} diff --git a/core/validator.go b/core/validator.go new file mode 100644 index 0000000..4c58318 --- /dev/null +++ b/core/validator.go @@ -0,0 +1,95 @@ +package core + +import ( + "github.com/go-playground/locales/zh" + ut "github.com/go-playground/universal-translator" + "github.com/go-playground/validator/v10" + zhTranslations "github.com/go-playground/validator/v10/translations/zh" + "reflect" + "vote-admin-video-api/global" + "vote-admin-video-api/utils" +) + +// Validator 验证器 +func Validator() { + chzh := zh.New() + uni := ut.New(chzh, chzh) + global.Trans, _ = uni.GetTranslator("zh") + + global.Validate = validator.New() + + // 通过label标签返回自定义错误内容 + global.Validate.RegisterTagNameFunc(func(field reflect.StructField) string { + label := field.Tag.Get("label") + if label == "" { + return field.Name + } + return label + }) + _ = zhTranslations.RegisterDefaultTranslations(global.Validate, global.Trans) + // 注册自定义函数和标签 + // 手机号验证 + _ = global.Validate.RegisterValidation("Mobile", mobile) // 注册自定义函数，前一个参数是struct里tag自定义，后一个参数是自定义的函数 + + // 自定义required错误内容 + _ = global.Validate.RegisterTranslation("required", global.Trans, func(ut ut.Translator) error { + return ut.Add("required", "{0}为必填字段!", false) // see universal-translator for details + }, func(ut ut.Translator, fe validator.FieldError) string { + t, _ := ut.T("required", fe.Field()) + return t + }) + + // 自定义max错误内容 + _ = global.Validate.RegisterTranslation("max", global.Trans, func(ut ut.Translator) error { + return ut.Add("max", "{0}超出最大长度", false) // see universal-translator for details + }, func(ut ut.Translator, fe validator.FieldError) string { + t, _ := ut.T("max", fe.Field()) + return t + }) + + // 自定义min错误内容 + _ = global.Validate.RegisterTranslation("min", global.Trans, func(ut ut.Translator) error { + return ut.Add("min", "{0}超出最小长度", false) // see universal-translator for details + }, func(ut ut.Translator, fe validator.FieldError) string { + t, _ := ut.T("min", fe.Field()) + return t + }) + + // 自定义lt错误内容 + _ = global.Validate.RegisterTranslation("lt", global.Trans, func(ut ut.Translator) error { + return ut.Add("lt", "{0}超出最大值", false) // see universal-translator for details + }, func(ut ut.Translator, fe validator.FieldError) string { + t, _ := ut.T("lt", fe.Field()) + return t + }) + + // 自定义min错误内容 + _ = global.Validate.RegisterTranslation("gt", global.Trans, func(ut ut.Translator) error { + return ut.Add("gt", "{0}不满足最小值", false) // see universal-translator for details + }, func(ut ut.Translator, fe validator.FieldError) string { + t, _ := ut.T("gt", fe.Field()) + return t + }) + + // 自定义email错误内容 + _ = global.Validate.RegisterTranslation("email", global.Trans, func(ut ut.Translator) error { + return ut.Add("email", "{0}邮件格式错误", false) // see universal-translator for details + }, func(ut ut.Translator, fe validator.FieldError) string { + t, _ := ut.T("email", fe.Field()) + return t + }) + + // 自定义mobile错误内容 + _ = global.Validate.RegisterTranslation("Mobile", global.Trans, func(ut ut.Translator) error { + return ut.Add("mobile", "手机号格式错误", false) // see universal-translator for details + }, func(ut ut.Translator, fe validator.FieldError) string { + t, _ := ut.T("mobile", fe.Field()) + return t + }) + +} + +// 自定义手机号验证 +func mobile(fl validator.FieldLevel) bool { + return utils.RegexpMobile(fl.Field().String()) +} diff --git a/core/viper.go b/core/viper.go new file mode 100644 index 0000000..63d8f56 --- /dev/null +++ b/core/viper.go @@ -0,0 +1,41 @@ +package core + +import ( + "fmt" + "github.com/fsnotify/fsnotify" + "github.com/spf13/viper" + "vote-admin-video-api/config" +) + +// Viper 初始化配置文件 +func Viper() { + // 如需增加环境判断 在此处增加 + // 可根据 命令行 > 环境变量 > 默认值 等优先级进行判别读取 + viper.New() + viper.SetConfigName("config") + viper.AddConfigPath("./") + viper.SetConfigType("yaml") + if err := viper.ReadInConfig(); err != nil { + if _, ok := err.(viper.ConfigFileNotFoundError); ok { + panic("未找到该文件") + } else { + panic("读取失败") + } + } + + // 将读取的配置信息保存至全局变量Conf + if err := viper.Unmarshal(&config.C); err != nil { + panic(fmt.Errorf("解析配置文件失败, err:%s \n", err)) + } + + // 自动监听配置修改 + viper.WatchConfig() + + // 配置文件发生变化后同步到全局变量Conf + viper.OnConfigChange(func(e fsnotify.Event) { + fmt.Println("config file changed:", e.Name) + if err := viper.Unmarshal(&config.C); err != nil { + panic(fmt.Errorf("重载配置文件失败, err:%s \n", err)) + } + }) +} diff --git a/extend/Editor/Editor.go b/extend/Editor/Editor.go new file mode 100644 index 0000000..4023a6a --- /dev/null +++ b/extend/Editor/Editor.go @@ -0,0 +1,128 @@ +package Editor + +// Config 定义了上传和管理图片、视频、文件等的配置信息 +type Config struct { + // 图片上传配置 + ImageActionName string `json:"imageActionName"` // 执行上传图片的action名称，默认值：image + ImageFieldName string `json:"imageFieldName"` // 提交的图片表单名称，默认值：file + ImageMaxSize int `json:"imageMaxSize"` // 上传大小限制，单位B，默认值：2048000 + ImageAllowFiles []string `json:"imageAllowFiles"` // 允许的图片格式，默认值：[".png", ".jpg", ".jpeg"] + ImageCompressEnable bool `json:"imageCompressEnable"` // 是否压缩图片，默认值：true + ImageCompressBorder int `json:"imageCompressBorder"` // 图片压缩最长边限制，默认值：1600 + ImageInsertAlign string `json:"imageInsertAlign"` // 插入图片时的浮动方式，默认值：none + ImageUrlPrefix string `json:"imageUrlPrefix"` // 图片访问路径前缀，默认值：空 + + // 涂鸦上传配置 + ScrawlActionName string `json:"scrawlActionName"` // 执行上传涂鸦的action名称，默认值：scrawl + ScrawlFieldName string `json:"scrawlFieldName"` // 提交的涂鸦表单名称，默认值：file + ScrawlMaxSize int `json:"scrawlMaxSize"` // 涂鸦上传大小限制，单位B，默认值：2048000 + ScrawlUrlPrefix string `json:"scrawlUrlPrefix"` // 涂鸦访问路径前缀，默认值：空 + ScrawlInsertAlign string `json:"scrawlInsertAlign"` // 插入涂鸦时的浮动方式，默认值：none + + // 截图上传配置 + SnapscreenActionName string `json:"snapscreenActionName"` // 执行上传截图的action名称，默认值：snap + SnapscreenUrlPrefix string `json:"snapscreenUrlPrefix"` // 截图访问路径前缀，默认值：空 + SnapscreenInsertAlign string `json:"snapscreenInsertAlign"` // 插入截图时的浮动方式，默认值：none + + // 图片抓取配置 + CatcherActionName string `json:"catcherActionName"` // 执行抓取远程图片的action名称，默认值：catch + CatcherFieldName string `json:"catcherFieldName"` // 提交的图片列表表单名称，默认值：source + CatcherLocalDomain []string `json:"catcherLocalDomain"` // 例外的图片抓取域名，默认值：["127.0.0.1", "localhost"] + CatcherUrlPrefix string `json:"catcherUrlPrefix"` // 抓取图片访问路径前缀，默认值：空 + CatcherMaxSize int `json:"catcherMaxSize"` // 抓取图片上传大小限制，单位B，默认值：10485760 + CatcherAllowFiles []string `json:"catcherAllowFiles"` // 允许的抓取图片格式，默认值：[".png", ".jpg", ".jpeg"] + + // 视频上传配置 + VideoActionName string `json:"videoActionName"` // 执行上传视频的action名称，默认值：video + VideoFieldName string `json:"videoFieldName"` // 提交的视频表单名称，默认值：file + VideoUrlPrefix string `json:"videoUrlPrefix"` // 视频访问路径前缀，默认值：空 + VideoMaxSize int `json:"videoMaxSize"` // 上传视频大小限制，单位B，默认值：102400000 + VideoAllowFiles []string `json:"videoAllowFiles"` // 允许的视频格式，默认值：[".mp4"] + + // 文件上传配置 + FileActionName string `json:"fileActionName"` // 执行上传文件的action名称，默认值：file + FileFieldName string `json:"fileFieldName"` // 提交的文件表单名称，默认值：file + FileUrlPrefix string `json:"fileUrlPrefix"` // 文件访问路径前缀，默认值：空 + FileMaxSize int `json:"fileMaxSize"` // 上传文件大小限制，单位B，默认值：104857600 + FileAllowFiles []string `json:"fileAllowFiles"` // 允许的文件格式，默认值：[".zip", ".pdf", ".doc"] + + // 图片列表配置 + ImageManagerActionName string `json:"imageManagerActionName"` // 执行图片管理的action名称，默认值：listImage + ImageManagerListSize int `json:"imageManagerListSize"` // 每次列出文件数量，默认值：20 + ImageManagerUrlPrefix string `json:"imageManagerUrlPrefix"` // 图片管理访问路径前缀，默认值：空 + ImageManagerInsertAlign string `json:"imageManagerInsertAlign"` // 插入的图片浮动方式，默认值：none + ImageManagerAllowFiles []string `json:"imageManagerAllowFiles"` // 列出的图片文件类型，默认值：[".jpg", ".png", ".jpeg"] + + // 文件列表配置 + FileManagerActionName string `json:"fileManagerActionName"` // 执行文件管理的action名称，默认值：listFile + FileManagerUrlPrefix string `json:"fileManagerUrlPrefix"` // 指定要列出文件的目录，默认值：空 + FileManagerListSize int `json:"fileManagerListSize"` // 每次列出文件数量，默认值：20 + FileManagerAllowFiles []string `json:"fileManagerAllowFiles"` // 列出的文件类型，默认值：[".zip", ".pdf", ".doc"] + + // 公式配置 + FormulaConfig FormulaConfig `json:"formulaConfig"` // 公式渲染相关配置 +} + +// FormulaConfig 定义了公式渲染的相关配置 +type FormulaConfig struct { + ImageUrlTemplate string `json:"imageUrlTemplate"` // 公式渲染的图片URL模板 +} + +// UploadDto 上传返回数据 +type UploadDto struct { + Original string `json:"original"` + State string `json:"state"` + Title string `json:"title"` + Type string `json:"type"` + Url string `json:"url"` +} + +// GetConfig 封装方法，返回配置信息 +func GetConfig() Config { + return Config{ + ImageActionName: "image", + ImageFieldName: "file", + ImageMaxSize: 8388608, + ImageAllowFiles: []string{".jpg", ".png", ".jpeg"}, + ImageCompressEnable: false, + ImageCompressBorder: 5000, + ImageInsertAlign: "none", + ImageUrlPrefix: "", + ScrawlActionName: "crawl", + ScrawlFieldName: "file", + ScrawlMaxSize: 10485760, + ScrawlUrlPrefix: "", + ScrawlInsertAlign: "none", + SnapscreenActionName: "snap", + SnapscreenUrlPrefix: "", + SnapscreenInsertAlign: "none", + CatcherActionName: "catch", + CatcherFieldName: "source", + CatcherLocalDomain: []string{"127.0.0.1", "localhost"}, + CatcherUrlPrefix: "", + CatcherMaxSize: 10485760, + CatcherAllowFiles: []string{".jpg", ".png", ".jpeg"}, + VideoActionName: "video", + VideoFieldName: "file", + VideoUrlPrefix: "", + VideoMaxSize: 104857600, + VideoAllowFiles: []string{".mp4"}, + FileActionName: "file", + FileFieldName: "file", + FileUrlPrefix: "", + FileMaxSize: 104857600, + FileAllowFiles: []string{".zip", ".pdf", ".doc"}, + ImageManagerActionName: "listImage", + ImageManagerListSize: 20, + ImageManagerUrlPrefix: "", + ImageManagerInsertAlign: "none", + ImageManagerAllowFiles: []string{".jpg", ".png", ".jpeg"}, + FileManagerActionName: "listFile", + FileManagerUrlPrefix: "", + FileManagerListSize: 20, + FileManagerAllowFiles: []string{".zip", ".pdf", ".doc"}, + FormulaConfig: FormulaConfig{ + ImageUrlTemplate: "https://r.latexeasy.com/image.svg?{}", + }, + } +} diff --git a/extend/aliyun/dysms.go b/extend/aliyun/dysms.go new file mode 100644 index 0000000..7a19e1f --- /dev/null +++ b/extend/aliyun/dysms.go @@ -0,0 +1,104 @@ +// Package aliyun 短信 +package aliyun + +import ( + "encoding/json" + "errors" + openapi "github.com/alibabacloud-go/darabonba-openapi/v2/client" + dysmsapi20170525 "github.com/alibabacloud-go/dysmsapi-20170525/v3/client" + util "github.com/alibabacloud-go/tea-utils/v2/service" + "github.com/alibabacloud-go/tea/tea" + "vote-admin-video-api/config" +) + +func createClient() (_result *dysmsapi20170525.Client, _err error) { + accessKeyId := config.C.Dysms.DysmsAccessKey + accessKeySecret := config.C.Dysms.DysmsAccessSecret + + openapiConfig := &openapi.Config{ + // 必填，您的 AccessKey ID + AccessKeyId: &accessKeyId, + // 必填，您的 AccessKey Secret + AccessKeySecret: &accessKeySecret, + } + // Endpoint 请参考 https://api.aliyun.com/product/Dysmsapi + openapiConfig.Endpoint = tea.String("dysmsapi.aliyuncs.com") + _result = &dysmsapi20170525.Client{} + _result, _err = dysmsapi20170525.NewClient(openapiConfig) + return _result, _err +} + +// SendSms 发送短信 +func SendSms(phoneNumber, templateCode, sceneDesc string, templateParam map[string]interface{}) error { + client, err := createClient() + if err != nil { + return err + } + + params, err := json.Marshal(templateParam) + if err != nil { + return err + } + + sendSmsRequest := &dysmsapi20170525.SendSmsRequest{ + PhoneNumbers: tea.String(phoneNumber), + SignName: tea.String("肝胆相照"), + TemplateCode: tea.String(templateCode), + TemplateParam: tea.String(string(params)), + } + + tryErr := func() (e error) { + defer func() { + if r := tea.Recover(recover()); r != nil { + e = r + } + }() + + // 初始化运行时配置。 + runtime := &util.RuntimeOptions{} + // 读取超时 + runtime.SetReadTimeout(10000) + // 连接超时 + runtime.SetConnectTimeout(5000) + + // 复制代码运行请自行打印 API 的返回值 + response, err := client.SendSms(sendSmsRequest) + if err != nil { + return err + } + + if response.Body == nil { + return errors.New("短信发送失败") + } + + if response.Body.Code != nil && *response.Body.Code != "OK" { + if response.Body.Message != nil { + return errors.New(*response.Body.Message) + } + } + + // 检测唯一值返回 + if response.Body.RequestId == nil { + if response.Body.Message != nil { + return errors.New(*response.Body.Message) + } + } + + return nil + }() + + if tryErr != nil { + var sdkError = &tea.SDKError{} + if t, ok := tryErr.(*tea.SDKError); ok { + sdkError = t + } else { + sdkError.Message = tea.String(tryErr.Error()) + } + // 如有需要，请打印 error + _, err = util.AssertAsString(sdkError.Message) + if err != nil { + return err + } + } + return err +} diff --git a/extend/aliyun/oss.go b/extend/aliyun/oss.go new file mode 100644 index 0000000..498f76b --- /dev/null +++ b/extend/aliyun/oss.go @@ -0,0 +1,186 @@ +package aliyun + +import ( + "bytes" + "crypto/hmac" + "crypto/sha1" + "encoding/base64" + "encoding/json" + "fmt" + "github.com/aliyun/aliyun-oss-go-sdk/oss" + "io" + "os" + "strings" + "time" + "vote-admin-video-api/config" +) + +// GetOssSignResponse 获取oss签名返回值 +type GetOssSignResponse struct { + AccessId string `json:"access_id"` // 主键id + Host string `json:"host"` + Policy string `json:"policy"` + Signature string `json:"signature"` + Expire int64 `json:"expire"` + Callback string `json:"callback"` + Dir string `json:"dir"` +} + +// GetOssSign 获取oss签名 +func GetOssSign(dir string) (*GetOssSignResponse, error) { + now := time.Now() + expire := 30 // 设置该policy超时时间是30s，即这个policy过了这个有效时间，将不能访问。 + end := now.Add(time.Second * time.Duration(expire)) + expiration := strings.Replace(end.Format("2006-01-02T15:04:05.000Z"), "+00:00", ".000Z", 1) + + start := []interface{}{"starts-with", "$key", dir} + conditions := [][]interface{}{start} + + arr := map[string]interface{}{ + "expiration": expiration, + "conditions": conditions, + } + policy, _ := json.Marshal(arr) + base64Policy := base64.StdEncoding.EncodeToString(policy) + stringToSign := base64Policy + h := hmac.New(sha1.New, []byte(config.C.Oss.OssAccessKeySecret)) + h.Write([]byte(stringToSign)) + signature := base64.StdEncoding.EncodeToString(h.Sum(nil)) + + response := &GetOssSignResponse{ + AccessId: config.C.Oss.OssAccessKey, + Host: config.C.Oss.OssCustomDomainName, + Policy: base64Policy, + Signature: signature, + Expire: end.Unix(), + Callback: "", + Dir: dir, + } + + return response, nil +} + +// CreateClient 创建客户端 +func CreateClient() (*oss.Client, error) { + // 创建OSSClient实例。 + client, err := oss.New(config.C.Oss.OssEndpoint, config.C.Oss.OssAccessKey, config.C.Oss.OssAccessKeySecret) + if err != nil { + return nil, err + } + + return client, nil +} + +// GetCusTomObjectToRAM 下载自定义风格文件到内存 +func GetCusTomObjectToRAM(filename string, style string) (string, error) { + if style == "" { + style = "image/resize" + } + + ossClient, err := CreateClient() + if err != nil { + return "", err + } + + // yourBucketName填写存储空间名称。 + bucket, err := ossClient.Bucket(config.C.Oss.OssBucket) + if err != nil { + return "", err + } + + // 下载文件到流。 + body, err := bucket.GetObject(filename, oss.Process(style)) + if err != nil { + return "", err + } + + // 数据读取完成后，获取的流必须关闭，否则会造成连接泄漏，导致请求无连接可用，程序无法正常工作。 + defer func(body io.ReadCloser) { + _ = body.Close() + }(body) + + data, err := io.ReadAll(body) + if err != nil { + fmt.Println("Error:", err) + os.Exit(-1) + } + + return string(data), nil +} + +// GetObjectToRAM 下载文件到内存 +func GetObjectToRAM(filename string) (string, error) { + ossClient, err := CreateClient() + if err != nil { + return "", err + } + + // yourBucketName填写存储空间名称。 + bucket, err := ossClient.Bucket(config.C.Oss.OssBucket) + if err != nil { + return "", err + } + + // 下载文件到流。 + body, err := bucket.GetObject(filename) + if err != nil { + return "", err + } + + // 数据读取完成后，获取的流必须关闭，否则会造成连接泄漏，导致请求无连接可用，程序无法正常工作。 + defer func(body io.ReadCloser) { + _ = body.Close() + }(body) + + data, err := io.ReadAll(body) + if err != nil { + fmt.Println("Error:", err) + os.Exit(-1) + } + + return string(data), nil +} + +// GetObjectToLocal 下载文件到本地 +func GetObjectToLocal(filename, local string) (bool, error) { + ossClient, err := CreateClient() + if err != nil { + return false, err + } + + // yourBucketName填写存储空间名称。 + bucket, err := ossClient.Bucket(config.C.Oss.OssBucket) + if err != nil { + return false, err + } + + // 下载文件到本地文件，并保存到指定的本地路径中。如果指定的本地文件存在会覆盖，不存在则新建。 + // 如果未指定本地路径，则下载后的文件默认保存到示例程序所属项目对应本地路径中。 + // 依次填写Object完整路径（例如exampledir/exampleobject.txt）和本地文件的完整路径(例如D:\\localpath\\examplefile.txt)。Object完整路径中不能包含Bucket名称。 + err = bucket.GetObjectToFile(filename, local) + if err != nil { + return false, err + } + + return true, nil +} + +// PutObjectByte 上传文件 +func PutObjectByte(filename string, content []byte) (bool, error) { + ossClient, err := CreateClient() + if err != nil { + return false, err + } + + // yourBucketName填写存储空间名称。 + bucket, err := ossClient.Bucket(config.C.Oss.OssBucket) + if err != nil { + return false, err + } + + err = bucket.PutObject(filename, bytes.NewReader(content)) + if err != nil { + return false, err + } + return true, nil +} diff --git a/extend/app/base.go b/extend/app/base.go new file mode 100644 index 0000000..7ee8ebc --- /dev/null +++ b/extend/app/base.go @@ -0,0 +1,92 @@ +package app + +import ( + "encoding/json" + "sort" + "strings" + "vote-admin-video-api/config" + "vote-admin-video-api/utils" +) + +const ( + apiUrl = "https://dev-wx.igandan.com" // 接口地址 + secretKey = "RY8pcn04#TSdzHVX6YgWnyCue9!T&QP^" // 产品私有密钥，服务端生成签名信息使用，请严格保管，避免泄露 + platform = "suanyisuan" // 所属平台 + devImagePrefix = "https://dev-doc.igandan.com/app" // 测试环境图片地址前缀 + prodImagePrefix = "https://dev-doc.igandan.com/app" // 正式环境图片地址前缀 +) + +// GenSignature 生成签名信息 +func GenSignature(params map[string]interface{}) (string, error) { + // 对map的key进行排序，包括多层嵌套的情况 + data := sortMapRecursively(params) + + // 转换为JSON + jsonData, err := json.Marshal(data) + if err != nil { + return "", err + } + + sing := utils.HmacSHA256(string(jsonData), secretKey) + return sing, nil +} + +// sortMapRecursively 对map的key进行排序，包括多层嵌套的情况 +func sortMapRecursively(data map[string]interface{}) map[string]interface{} { + sortedMap := make(map[string]interface{}) + keys := make([]string, 0, len(data)) + + // 收集所有的key + for key := range data { + keys = append(keys, key) + } + + // 对key进行排序 + sort.Strings(keys) + + // 通过排序后的key插入新map中 + for _, key := range keys { + value := data[key] + switch valueTyped := value.(type) { + case map[string]interface{}: + // 如果是嵌套的map，递归调用 + sortedMap[key] = sortMapRecursively(valueTyped) + case []interface{}: + // 如果是嵌套的slice，对其中的map进行递归调用 + for i, v := range valueTyped { + if vMap, ok := v.(map[string]interface{}); ok { + valueTyped[i] = sortMapRecursively(vMap) + } + } + sortedMap[key] = valueTyped + default: + // 否则直接插入 + sortedMap[key] = value + } + } + + return sortedMap +} + +// HandleImagePrefix 处理app图片前缀 +func HandleImagePrefix(u string) (string, error) { + if u == "" { + return "", nil + } + + // 去除oss前缀 + u = utils.RemoveOssDomain(u) + + var imgPath string + if config.C.Env == "prod" { + imgPath = strings.Replace(u, devImagePrefix, "", 1) + } else { + imgPath = strings.Replace(u, prodImagePrefix, "", 1) + } + + if imgPath == "/null" { + return "", nil + } + + return imgPath, nil +} diff --git a/extend/app/pay.go b/extend/app/pay.go new file mode 100644 index 0000000..02fd9da --- /dev/null +++ b/extend/app/pay.go @@ -0,0 +1,53 @@ +package app + +import ( + "encoding/json" + "errors" + "fmt" +) + +// PayOrderRequest 获取订单支付请求数据 +type PayOrderRequest struct { + OrderId string `json:"orderId" label:"订单编号" validate:"required"` // 订单id +} + +// PayOrderDataResponse 获取订单支付返回数据-data +type PayOrderDataResponse struct { + AppId string `json:"appid"` // 公众号id + Total int `json:"total"` // 订单总金额(精确到分) + Description string `json:"description"` // 订单描述 + OpenId string `json:"openid"` // 下单用户 + OutTradeNo string `json:"out_trade_no"` // 商户订单 + Attach string `json:"attach"` // 附加信息 + NotifyUrl string `json:"notify_url"` // 异步接收微信支付结果通知的回调地址 + GoodName string `json:"goodName"` // 商品名称 +} + +// VerifySignature 验证签名 +func VerifySignature(req PayOrderRequest, requestSign string) error { + // 将 JSON 数据编码为字节数组 + jsonData, err := json.Marshal(req) + if err != nil { + return err + } + + maps := make(map[string]interface{}) + err = json.Unmarshal(jsonData, &maps) + if err != nil { + return err + } + + // 生成签名 + sign, err := GenSignature(maps) + if err != nil { + return err + } + + fmt.Println(sign) + // 对比签名 + if sign != requestSign { + return errors.New("签名错误") + } + + return nil +} diff --git a/extend/app/userCase.go b/extend/app/userCase.go new file mode 100644 index 0000000..b21a653 --- /dev/null +++ b/extend/app/userCase.go @@ -0,0 +1,242 @@ +package app + +import ( + "bytes" + "encoding/json" + "errors" + "io" + "net/http" + "strconv" + "time" + "vote-admin-video-api/utils" +) + +// GetUserCaseByAppIdenRequest 根据app唯一标识获取用户病例信息-请求数据 +type GetUserCaseByAppIdenRequest struct { + PatientUuid string `json:"patientUuid"` // 患者 uuid + Platform string `json:"platform"` // 所属平台 + Timestamp string `json:"timestamp"` // 当前时间戳（10位） +} + +// UpdateUserCaseRequest 修改疾病信息-请求数据 +type UpdateUserCaseRequest struct { + IsAllergy *int `json:"isAllergy"` // 是否过敏史 0否 1是 + AllergyInfo string `json:"allergyInfo"` // 过敏史详情 + IsHospital *int `json:"isHospital"` // 是否去医院 0否 1是 + IsMedication *int `json:"isMedication"` // 是否服药 0否 1是 + MedicationInfo string `json:"medicationInfo"` // 正在服用的药物 + LiverStatus string `json:"liverStatus"` // 目前肝脏状态 + OtherDisease string `json:"otherDisease"` // 合并其他慢性疾病 （多个英文逗号分隔拼接） + DiseasesList []*DiseasesListRequest `json:"diseasesList"` // 所患疾病列表 + PatientUuid string `json:"patientUuid"` // 患者 uuid + Platform string `json:"platform"` // 所属平台 + Timestamp string `json:"timestamp"` // 当前时间戳（10位） +} + +type DiseasesListRequest struct { + Uuid string `json:"uuid"` // 疾病 uuid + Year *int `json:"year"` // 患病时长 + Info string `json:"info"` // 丙肝基因型（仅针对丙肝） + Name string `json:"name"` // 疾病名称 +} + +// GetUserCaseByAppIdenResponse 根据app唯一标识获取用户病例信息-返回数据 +type GetUserCaseByAppIdenResponse struct { + Code int `json:"code"` // 接口调用状态。200：正常；其它值：调用出错 + Msg string `json:"msg"` // 结果说明。如果接口调用出错，那么返回错误描述。成功则返回 ok + Data *GetUserCaseByAppIdenData `json:"data"` // 接口返回结果，各个接口自定义，数据结构参考具体文档说明 + Success bool `json:"success"` + Message string `json:"message"` +} + +// UpdateUserCaseResponse 修改用户病例-返回数据 +type UpdateUserCaseResponse struct { + Code int `json:"code"` // 接口调用状态。200：正常；其它值：调用出错 + Msg string `json:"msg"` // 结果说明。如果接口调用出错，那么返回错误描述。成功则返回 ok + Data string `json:"data"` // 接口返回结果，各个接口自定义，数据结构参考具体文档说明 + Success bool `json:"success"` + Message string `json:"message"` +} + +// GetUserCaseByAppIdenData 根据app唯一标识获取用户病例信息-data详细数据 +type GetUserCaseByAppIdenData struct { + AllergyInfo string `json:"allergyInfo" description:"过敏史详情"` + PatientUUID string `json:"patientUuid" description:"患者 uuid"` + DiseasesList []*DiseasesListData `json:"diseasesList" description:"所患疾病列表"` + MedicationInfo string `json:"medicationInfo" description:"正在服用的药物"` + OtherDisease string `json:"otherDisease" description:"合并其他慢性疾病 （多个英文逗号分隔拼接）"` + IsMedication *int `json:"isMedication" description:"是否服药 0否 1是"` + IsHospital *int `json:"isHospital" description:"是否去医院 0否 1是"` + IsAllergy *int `json:"isAllergy" description:"是否过敏史 0否 1是"` + LiverStatus string `json:"liverStatus" description:"目前肝脏状态"` +} + +// DiseasesListData 根据app唯一标识获取用户病例信息-data详细数据-所患疾病数据 +type DiseasesListData struct { + UUID string `json:"uuid" description:"疾病 uuid"` + Year *int `json:"year" description:"患病时长"` + Info string `json:"info" description:"丙肝基因型（仅针对丙肝）"` + Name string `json:"name" description:"疾病名称"` +} + +// GetUserCaseByAppIden 根据app唯一标识获取用户病例信息 +func GetUserCaseByAppIden(appIden string) (g *GetUserCaseByAppIdenResponse, err error) { + // 准备要发送的 JSON 数据 + requestData := GetUserCaseByAppIdenRequest{ + PatientUuid: appIden, + Platform: platform, + Timestamp: strconv.FormatInt(time.Now().Unix(), 10), + } + + // 将 JSON 数据编码为字节数组 + jsonData, err := json.Marshal(requestData) + if err != nil { + return g, err + } + + maps := make(map[string]interface{}) + err = json.Unmarshal(jsonData, &maps) + if err != nil { + return g, err + } + + // 获取请求签名 + sign, err := GenSignature(maps) + if err != nil { + return g, err + } + + // 准备请求体 + requestBody := bytes.NewBuffer(jsonData) + + // 设置请求 URL + url := apiUrl + "/patient-api/getDiseaseInfo" + + // 创建 POST 请求 + req, err := http.NewRequest("POST", url, requestBody) + if err != nil { + return g, err + } + + // 设置请求头 + req.Header.Set("Content-Type", "application/json") + req.Header.Set("sign", sign) + + // 发送请求 + client := &http.Client{} + resp, err := client.Do(req) + if err != nil { + return g, err + } + + defer func(Body io.ReadCloser) { + _ = Body.Close() + }(resp.Body) + + body, err := io.ReadAll(resp.Body) + if err != nil { + return g, err + } + + // 检查响应状态码 + if resp.StatusCode != 200 { + return g, errors.New("失败") + } + + err = json.Unmarshal(body, &g) + if err != nil { + // json解析失败 + return g, err + } + + utils.LogJsonInfo("获取app数据返回", g) + + if g.Code != 200 { + if g.Msg != "" { + return g, errors.New(g.Msg) + } else { + return g, errors.New("失败") + } + } + + return g, nil +} + +// UpdateUserCase 修改用户病例 +func UpdateUserCase(reqData UpdateUserCaseRequest) (g *UpdateUserCaseResponse, err error) { + reqData.Platform = platform + reqData.Timestamp = strconv.FormatInt(time.Now().Unix(), 10) + + // 将 JSON 数据编码为字节数组 + jsonData, err := json.Marshal(reqData) + if err != nil { + return g, err + } + + maps := make(map[string]interface{}) + err = json.Unmarshal(jsonData, &maps) + if err != nil { + return g, err + } + + // 获取请求签名 + sign, err := GenSignature(maps) + if err != nil { + return g, err + } + + // 准备请求体 + requestBody := bytes.NewBuffer(jsonData) + + // 设置请求 URL + url := apiUrl + "/patient-api/upDiseaseInfo" + + // 创建 POST 请求 + req, err := http.NewRequest("POST", url, requestBody) + if err != nil { + return g, err + } + + // 设置请求头 + req.Header.Set("Content-Type", "application/json") + req.Header.Set("sign", sign) + + // 发送请求 + client := &http.Client{} + resp, err := client.Do(req) + if err != nil { + return g, err + } + + defer func(Body io.ReadCloser) { + _ = Body.Close() + }(resp.Body) + + body, err := io.ReadAll(resp.Body) + if err != nil { + return g, err + } + + // 检查响应状态码 + if resp.StatusCode != 200 { + return g, errors.New("失败") + } + + err = json.Unmarshal(body, &g) + if err != nil { + // json解析失败 + return g, err + } + + utils.LogJsonInfo("修改app数据返回", g) + + if g.Code != 200 { + if g.Msg != "" { + return g, errors.New(g.Msg) + } else { + return g, errors.New("失败") + } + } + + return g, nil +} diff --git a/extend/app/userInfo.go b/extend/app/userInfo.go new file mode 100644 index 0000000..aa196e2 --- /dev/null +++ b/extend/app/userInfo.go @@ -0,0 +1,239 @@ +package app + +import ( + "bytes" + "encoding/json" + "errors" + "io" + "net/http" + "strconv" + "time" + "vote-admin-video-api/utils" +) + +// GetInfoByMobileRequest 根据手机号获取用户信息-请求数据 +type GetInfoByMobileRequest struct { + Mobile string `json:"mobile"` // 手机号 + Platform string `json:"platform"` // 所属平台 + Timestamp string `json:"timestamp"` // 当前时间戳（10位） +} + +// UpdateInfoRequest 修改用户信息-请求数据 +type UpdateInfoRequest struct { + Birthday string `json:"birthday"` // 出生日期 + IsPegnant *int `json:"isPegnant"` // 是否怀孕 1无计划 2计划中 3已怀孕 4家有宝宝 + Sex *int `json:"sex"` // 性别 0男 1女 + Weight *int `json:"weight"` // 体重 KG + ExpectedDateOfChildbirth string `json:"expectedDateOfChildbirth"` // 预产期 + IsHbv *int `json:"isHbv"` // 市区 id + NationUuid string `json:"nationUuid"` // 民族 uuid + PatientUuid string `json:"patientUuid"` // 患者 uuid + Name string `json:"name"` // 姓名 + ProvId *int64 `json:"provId"` // 省份 id + CityId *int64 `json:"cityId"` // 城市 id + CountyId *int64 `json:"countyId"` // 市区 id + Height *int `json:"height"` // 身高 cm + Platform string `json:"platform"` // 所属平台 + Timestamp string `json:"timestamp"` // 当前时间戳（10位） +} + +// GetInfoByMobileResponse 根据手机号获取用户信息-返回数据 +type GetInfoByMobileResponse struct { + Code int `json:"code"` // 接口调用状态。200：正常；其它值：调用出错 + Msg string `json:"msg"` // 结果说明。如果接口调用出错，那么返回错误描述。成功则返回 ok + Data GetInfoByMobileData `json:"data"` // 接口返回结果，各个接口自定义，数据结构参考具体文档说明 + Success bool `json:"success"` + Message string `json:"message"` +} + +// UpdateInfoResponse 修改用户信息-返回数据 +type UpdateInfoResponse struct { + Code int `json:"code"` // 接口调用状态。200：正常；其它值：调用出错 + Msg string `json:"msg"` // 结果说明。如果接口调用出错，那么返回错误描述。成功则返回 ok + Data string `json:"data"` // 接口返回结果，各个接口自定义，数据结构参考具体文档说明 + Success bool `json:"success"` + Message string `json:"message"` +} + +// GetInfoByMobileData 根据手机号获取用户信息-data详细数据 +type GetInfoByMobileData struct { + Birthday string `json:"birthday" description:"出生日期"` + IsPregnant *int `json:"isPregnant" description:"是否怀孕 1无计划 2计划中 3已怀孕 4家有宝宝"` + Sex *int `json:"sex" description:"性别 0男 1女"` + Mobile string `json:"mobile" description:"手机号"` + Photo string `json:"photo" description:"头像地址"` + Weight *int `json:"weight" description:"体重 KG"` + CityID *int64 `json:"cityId" description:"城市 id"` + ExpectedDateOfChildbirth string `json:"expectedDateOfChildbirth" description:"预产期"` + CountyID *int64 `json:"countyId" description:"市区 id"` + IsHBV *int `json:"isHbv" description:"有无 肝硬化或肝癌家族史 0无1有2未知"` + NationUUID string `json:"nationUuid" description:"民族 uuid"` + PatientUUID string `json:"patientUuid" description:"患者 uuid"` + Name string `json:"name" description:"姓名"` + ProvinceID *int64 `json:"provId" description:"省份 id"` + Height *int `json:"height" description:"身高 cm"` + OpenId string `json:"openid" description:"openid"` + UnionId string `json:"unionid" description:"unionid"` +} + +// GetInfoByMobile 根据手机号获取用户信息 +func GetInfoByMobile(mobile string) (g *GetInfoByMobileResponse, err error) { + // 准备要发送的 JSON 数据 + requestData := GetInfoByMobileRequest{ + Mobile: mobile, + Platform: platform, + Timestamp: strconv.FormatInt(time.Now().Unix(), 10), + } + + // 将 JSON 数据编码为字节数组 + jsonData, err := json.Marshal(requestData) + if err != nil { + return g, err + } + + maps := make(map[string]interface{}) + err = json.Unmarshal(jsonData, &maps) + if err != nil { + return g, err + } + + // 获取请求签名 + sign, err := GenSignature(maps) + if err != nil { + return g, err + } + + // 准备请求体 + requestBody := bytes.NewBuffer(jsonData) + + // 设置请求 URL + url := apiUrl + "/patient-api/getInfo" + + // 创建 POST 请求 + req, err := http.NewRequest("POST", url, requestBody) + if err != nil { + return g, err + } + + // 设置请求头 + req.Header.Set("Content-Type", "application/json") + req.Header.Set("sign", sign) + + // 发送请求 + client := &http.Client{} + resp, err := client.Do(req) + if err != nil { + return g, err + } + + defer func(Body io.ReadCloser) { + _ = Body.Close() + }(resp.Body) + + body, err := io.ReadAll(resp.Body) + if err != nil { + return g, err + } + + // 检查响应状态码 + if resp.StatusCode != 200 { + return g, errors.New("失败") + } + + err = json.Unmarshal(body, &g) + if err != nil { + // json解析失败 + return g, err + } + + utils.LogJsonInfo("获取app数据返回", g) + + if g.Code != 200 { + if g.Msg != "" { + return g, errors.New(g.Msg) + } else { + return g, errors.New("失败") + } + } + + return g, nil +} + +// UpdateInfo 修改用户信息 +func UpdateInfo(reqData UpdateInfoRequest) (g *UpdateInfoResponse, err error) { + reqData.Platform = platform + reqData.Timestamp = strconv.FormatInt(time.Now().Unix(), 10) + + // 将 JSON 数据编码为字节数组 + jsonData, err := json.Marshal(reqData) + if err != nil { + return g, err + } + + maps := make(map[string]interface{}) + err = json.Unmarshal(jsonData, &maps) + if err != nil { + return g, err + } + + // 获取请求签名 + sign, err := GenSignature(maps) + if err != nil { + return g, err + } + + // 准备请求体 + requestBody := bytes.NewBuffer(jsonData) + + // 设置请求 URL + url := apiUrl + "/patient-api/updateInfo" + + // 创建 POST 请求 + req, err := http.NewRequest("POST", url, requestBody) + if err != nil { + return g, err + } + + // 设置请求头 + req.Header.Set("Content-Type", "application/json") + req.Header.Set("sign", sign) + + // 发送请求 + client := &http.Client{} + resp, err := client.Do(req) + if err != nil { + return g, err + } + + defer func(Body io.ReadCloser) { + _ = Body.Close() + }(resp.Body) + + body, err := io.ReadAll(resp.Body) + if err != nil { + return g, err + } + + // 检查响应状态码 + if resp.StatusCode != 200 { + return g, errors.New("失败") + } + + err = json.Unmarshal(body, &g) + if err != nil { + // json解析失败 + return g, err + } + + utils.LogJsonInfo("修改app数据返回", g) + + if g.Code != 200 { + if g.Msg != "" { + return g, errors.New(g.Msg) + } else { + return g, errors.New("失败") + } + } + + return g, nil +} diff --git a/extend/weChat/base.go b/extend/weChat/base.go new file mode 100644 index 0000000..771e317 --- /dev/null +++ b/extend/weChat/base.go @@ -0,0 +1,103 @@ +package weChat + +import ( + "context" + "errors" + "github.com/wechatpay-apiv3/wechatpay-go/core" + "github.com/wechatpay-apiv3/wechatpay-go/core/option" + "github.com/wechatpay-apiv3/wechatpay-go/services/payments" + "github.com/wechatpay-apiv3/wechatpay-go/services/refunddomestic" + "github.com/wechatpay-apiv3/wechatpay-go/utils" + "vote-admin-video-api/config" +) + +// 创建客户端 +func createClient() (*core.Client, error) { + mchId := config.C.Wechat.Pay1281030301.MchId // 商户号 + mchCertificateSerialNumber := config.C.Wechat.Pay1281030301.MchCertificateSerialNumber // 商户证书序列号 + v3ApiSecret := config.C.Wechat.Pay1281030301.V3ApiSecret // 商户APIv3密钥 + privateKeyPath := config.C.Wechat.Pay1281030301.PrivateKey // 商户私钥文件地址 + + if mchId == "" { + return nil, errors.New("商户号错误") + } + + mchPrivateKey, err := utils.LoadPrivateKeyWithPath(privateKeyPath) + if err != nil { + return nil, errors.New("微信支付生成失败") + } + + ctx := context.Background() + // 使用商户私钥等初始化 client，并使它具有自动定时获取微信支付平台证书的能力 + opts := []core.ClientOption{ + option.WithWechatPayAutoAuthCipher(mchId, mchCertificateSerialNumber, mchPrivateKey, v3ApiSecret), + } + + client, err := core.NewClient(ctx, opts...) + if err != nil { + return nil, errors.New(err.Error()) + } + + return client, nil +} + +// WxPayResult 支付结果 +type WxPayResult struct { + OrderStatus int `json:"order_status"` // 订单状态（1:待支付 2:已完成 3:已取消） + PayStatus int `json:"pay_status"` // 支付状态（1:未支付 2:已支付 3:支付中 4:支付失败 5:支付超时 6:支付关闭 7:已撤销 8:转入退款） + PayTime *string `json:"pay_time"` // 支付时间 +} + +// WxPayRefundResult 退款结果 +type WxPayRefundResult struct { + RefundStatus int `json:"refundStatus"` // 订单退款状态（0:无退款 1:申请退款 2:退款中 3:退款成功 4:拒绝退款 5:退款关闭 6:退款异常 7:部分退款） + SuccessTime *string `json:"successTime"` // 退款成功时间 +} + +// HandlePayStatus 处理支付状态 +func HandlePayStatus(t *payments.Transaction) (w *WxPayResult, err error) { + w = &WxPayResult{} + + switch *t.TradeState { + case "SUCCESS": // 支付成功 + w.OrderStatus = 2 + w.PayStatus = 2 + w.PayTime = t.SuccessTime + case "CLOSED": // 已关闭 + w.PayStatus = 6 + case "REVOKED": // 已撤销（付款码支付） + w.PayStatus = 7 + case "USERPAYING": // 用户支付中（付款码支付） + w.PayStatus = 3 + case "PAYERROR": // 支付失败(其他原因，如银行返回失败) + w.PayStatus = 4 + default: + return nil, errors.New("未知支付状态") + } + + return w, nil +} + +// HandlePayRefundStatus 处理退款状态 +func HandlePayRefundStatus(r *refunddomestic.Refund) (w *WxPayRefundResult, err error) { + w = &WxPayRefundResult{} + + switch *r.Status { + case "SUCCESS": // 退款成功 + w.RefundStatus = 3 + if r.SuccessTime != nil { + successTime := r.SuccessTime.Format("2006-01-02 15:04:05") + w.SuccessTime = &successTime + } + case "CLOSED": // 退款关闭 + w.RefundStatus = 5 + case "PROCESSING": // 退款处理中 + w.RefundStatus = 2 + case "ABNORMAL": // 退款异常 + return nil, errors.New("退款状态错误") + default: + return nil, errors.New("退款状态错误") + } + + return w, nil +} diff --git a/extend/weChat/certs/1281030301/apiclient_key.pem b/extend/weChat/certs/1281030301/apiclient_key.pem new file mode 100644 index 0000000..35a62ce --- /dev/null +++ b/extend/weChat/certs/1281030301/apiclient_key.pem @@ -0,0 +1,28 @@ +-----BEGIN PRIVATE KEY----- +MIIEvQIBADANBgkqhkiG9w0BAQEFAASCBKcwggSjAgEAAoIBAQDR4fiYRUjnLQA+ +TPMslDvQIIyZm4ajWFcSXg6yozhdi/O+vuRhwgyeGXsE4SpvuxkK4HPZaocrsjqe +68Y46DyhudzXIBy50IF2UeN5ilj7ydq5TJbBOi1iOF1lgOUNK6DGiOQ+folVNKyN +sjrsA3RHUKHF/JdOfcpSGzvV79FnmjSll/KsdxEc9LCiAa5t7GYA9Hx7AIDG4i8b +03d5EJXvE8BGUZQItE6VIpHwyySTHtGQkPyI6lqMKt2Jlx0TFHu0GkTEmhuA2Z5C +5gOqYWclQO+X2ORgulL9Q3mNcpfy9uBynI/CJDbSvVTkNfwYB0Ny1knu/+in7C6p +IPJYfZpPAgMBAAECggEAWb5tJPcjSC5W10ziAiLUPJdeZ2Q4OupQOPtc/4eJV367 +V8maMC7gZE3y61A4bBQtjhgRkVrat5V7OW8JkFXFb0XhJ1+EyPNeGDDFureseuWC +EA+uuqrcsw306a0mw+3uzlXEevByWquuSNx4E2katE/HDLiIHjjtZRReDomAGfLw +vLNZ40RgdNSXmwpVRoHOZnvX+C2Hxh4fKsrxs1HRCjFsSpTxKwsK6/Kdudj4hsaO +Z5glqf4qopPlQIIkToxOe5p7ukwo7vqaqogixx3jOjruOMpzxZ24wU5AxneLcOYr +ZC1659Uwv96TsKpueakWw0lq3TXkv9Qs+wDNI+NC6QKBgQDx4hWXplLYPYwgGgxG +L9Pk+If9aDS9wj78JUaYBXKR5atXmf/62ufahW4IEDfmMugC1bm9MoYC7CmHv1nk +nxckB84B20NOSiaxbkOaJuMHDtnNeXCmL8Gvtb8fTni1AM/g6XEleEw1r40LNZAq +kdyQ1+OUFhRQGOvAe1SRr1rdGwKBgQDeIcauwMPkc3Seh6Fb66FQ4Lgmt1GvvuiG +UJiwgwPFbEaKKczBHBL6hlKoAFweFa2Qw2xJ0H8jwb8RnkkNcAKYUyMc2wi85+Ih +d568pvXMdYl4DGeGdd/sXKf4dGDjcb2AHzNfDuklbCsAojtO3pLWOs5U/RSKe8ps +hhWa8nPO3QKBgQCIxuKU3X1tP+hz4qbcLYFxscQcXIeuYiABrwZrQnFV5Pxtzex9 +Krn+zIK61oj1iAXATKD6Ro6XKnoVg/POHtQUEMHCNP2rUKzumj5p9eFdBV3OHgTA +RLMOrARGLLZ/C9WBBiBwIsVdekaUdxZtrAuAcEQFYjLcVCtDrbnVo8YKzwKBgEsg +V0cBMP+RwM5hBsTE45Er/3wwofLzeUb7+Tgxh1P887qEuphRO2X5ifkB7iXKpSIB +xh0M5AMe4tU9mG1wBaCo9YYr2j+xmTxCbbBWM2mMEwtD/rtuIGabS7/u9FnYPQQZ +CVHMBDRA6iZTuAVLp5PG3cPGuGzBw0uC6cm22E4NAoGAKrbg3nFPKtzBJZ8Z0iNH +G2gNotmHXbd4+gs4e3Iz9xsEabm4wCoEibNojNdMkX8zX267ebgfvesGEobROjW9 +Intvh39xi3aQ2Q5gvXzKdY0lDzgVvQ7udmLQ4dCUyYnpLr7Yac0asRxLge2esCWV +x2uW2YTiUW3zsbHN/N/vJ5I= +-----END PRIVATE KEY----- diff --git a/extend/weChat/certs/1636644248/apiclient_cert.p12 b/extend/weChat/certs/1636644248/apiclient_cert.p12 new file mode 100644 index 0000000..b5df8dd Binary files /dev/null and b/extend/weChat/certs/1636644248/apiclient_cert.p12 differ diff --git a/extend/weChat/certs/1636644248/apiclient_cert.pem b/extend/weChat/certs/1636644248/apiclient_cert.pem new file mode 100644 index 0000000..101b476 --- /dev/null +++ b/extend/weChat/certs/1636644248/apiclient_cert.pem @@ -0,0 +1,25 @@ +-----BEGIN CERTIFICATE----- +MIIELjCCAxagAwIBAgIUfewObFfg3HHwd/AvUkBlZq85vrswDQYJKoZIhvcNAQEL +BQAwXjELMAkGA1UEBhMCQ04xEzARBgNVBAoTClRlbnBheS5jb20xHTAbBgNVBAsT +FFRlbnBheS5jb20gQ0EgQ2VudGVyMRswGQYDVQQDExJUZW5wYXkuY29tIFJvb3Qg +Q0EwHhcNMjMwMzAxMDExNzE3WhcNMjgwMjI4MDExNzE3WjCBhzETMBEGA1UEAwwK +MTYzNjY0NDI0ODEbMBkGA1UECgwS5b6u5L+h5ZWG5oi357O757ufMTMwMQYDVQQL +DCrljJfkuqzmrKPmrKPnm7jnhaflgaXlurfnp5HmioDmnInpmZDlhazlj7gxCzAJ +BgNVBAYMAkNOMREwDwYDVQQHDAhTaGVuWmhlbjCCASIwDQYJKoZIhvcNAQEBBQAD +ggEPADCCAQoCggEBAK7x9ywApasrs+xIt0XxHhYOK4xZeuNEj5o7pzTcZMRkd6CB +1eX2ydlUfDe+FcNS03cwLNtUeAfrZXlhA807b4HuReaFqCrbt87hfIF/lOBpePdN +sSr8Wi6OKfPakuLNZ4RCOdlvgxPMhOf2b9VFQwns8h72H6JrFz7xR+Heundy3KGH +kEoG2qcl7nKyhkUVSRSH4/yzsxsDIwkpXjEiSL87E+A6GKik7jphYc+vfV9NURiA +JwbVWbQMhpj3YLRxgXadLS4xMryB59fYKm+VFMNg/jSZG55Jz2DW02aR2h3KjegY +SGA9qMKojkFVOrZvxKWOvtEWw2JQ/1dRvzYpgw0CAwEAAaOBuTCBtjAJBgNVHRME +AjAAMAsGA1UdDwQEAwID+DCBmwYDVR0fBIGTMIGQMIGNoIGKoIGHhoGEaHR0cDov +L2V2Y2EuaXRydXMuY29tLmNuL3B1YmxpYy9pdHJ1c2NybD9DQT0xQkQ0MjIwRTUw +REJDMDRCMDZBRDM5NzU0OTg0NkMwMUMzRThFQkQyJnNnPUhBQ0M0NzFCNjU0MjJF +MTJCMjdBOUQzM0E4N0FEMUNERjU5MjZFMTQwMzcxMA0GCSqGSIb3DQEBCwUAA4IB +AQCBjjBeJTRtnsIISCfoQ7pj70dil5LTBNjpzV9swGOG7sY9dTMbHV0K2BUyRvbD +eY2fuCcr3Qt3L1RykyomKbxD4O6b/ZPLTkiq2m+hq34g2Ig7zUx0a0sNKf/NPAq6 +6pVu5/XdIKYXz2OsByBI8aN2hkcUApmM1qlm+gb4FSmTzkdHaFRblhDlKfiRdAw8 +T4yliaOFovqhf/S8o5Xa76kQMOb3uJ/oE4KX02kp/ig5wZKj9nay46AyovBXOj+Z +0oaeIPFDXCPzZV7LA4E45zDl43fL8r+9OzaVprRZ9ASO0cgnPsHyS+o/mpMEBOR4 +NIwdIaqYRQ/Rh6eZQvrqDZ4r +-----END CERTIFICATE----- diff --git a/extend/weChat/certs/1636644248/apiclient_key.pem b/extend/weChat/certs/1636644248/apiclient_key.pem new file mode 100644 index 0000000..8e9da3d --- /dev/null +++ b/extend/weChat/certs/1636644248/apiclient_key.pem @@ -0,0 +1,28 @@ +-----BEGIN PRIVATE KEY----- +MIIEvAIBADANBgkqhkiG9w0BAQEFAASCBKYwggSiAgEAAoIBAQCu8fcsAKWrK7Ps +SLdF8R4WDiuMWXrjRI+aO6c03GTEZHeggdXl9snZVHw3vhXDUtN3MCzbVHgH62V5 +YQPNO2+B7kXmhagq27fO4XyBf5TgaXj3TbEq/Foujinz2pLizWeEQjnZb4MTzITn +9m/VRUMJ7PIe9h+iaxc+8Ufh3rp3ctyhh5BKBtqnJe5ysoZFFUkUh+P8s7MbAyMJ +KV4xIki/OxPgOhiopO46YWHPr31fTVEYgCcG1Vm0DIaY92C0cYF2nS0uMTK8gefX +2CpvlRTDYP40mRueSc9g1tNmkdodyo3oGEhgPajCqI5BVTq2b8Sljr7RFsNiUP9X +Ub82KYMNAgMBAAECggEAT7uNyGs/FkVjykPV67WZ3bl1lZDOljgQLt4TNd9gubWE +ZA3om9efZULBHnKu3oeoQ0EcoJXd4tYhOHHD1szI5HHhP9AYtffPzSUtpqOsCZ9o +d2XcYlgDDgbTDgXHPkEZdcjtLrFJD0P+Ku5BR/U6OZLZQs0v28ltHc2/0iy91WQt +iB/NSf88dY1L5RVZc7dZfb8nFB2/BsITUXDFJpzdim3AXKvNYzDajs/yuoTp5r/b +2EOjUIGPP55W/8iPwbezgDToF79toum6V7yUpCq5P+QQvnnbJfoc1yXB48zHjmgP +I2P1cDVtFs+V9Edb6qsU3e4CMyBQg5Uykk3TUqZRIQKBgQDnXgGk8XnI1kG9jZMR +wMkra87Dx76vrBdM47n+c2I1xubM9WSeSUxtXGyryTwJ9i6/BUyc3Eyb2wi4Q8AK +1ZIEFhvIUkBJpUgevAksj4eX6z0qDwDc3ZhgAAWOplnOCCyJuOBKmZks4GaQb4Gv +/aSFVQ9jefOc5e9RIDNzkHFkaQKBgQDBkigvmusjad95Jt34TwyE4Dr9z2LzBker +6ebcyQmv4YdKvq0KaaVMrroH7pNu7CSFrj4CGqdS/Gg1BtK8Xaxn+gEWT6RPcsi1 +mPwy/7oiK0GodzNI6RW5HDZEHKG2icEb4ycDnHeLfThKmP/gckPifjcJHrP5OX1A +V6qrq4iFBQKBgGdgB1gNVJ65rJHnCckq3Dd8advsCXUwbRC7x0S7hSwF/OWi1xwq +H+3VF/EBbsP8rRJIadzESa5xhUnfa5Trq9wLjMpKhdLh+IFS/r5cOvdT8fYy0e3d +TNHH8LO1+/YkjNHUOtLaIih88xah284ohDPWt5N4z7JQwkb7HkIKTb/RAoGARt51 +7Afx8sM+WCLMva5jTPqzXl1hQsyXzO8T4N2RuFz/pXPt8pP/OvX1khXc0I2QSYkj +lq2feRiEJnXbDa/WATNc1ohOBfBmX2YlX56UzRG9Nip+EkGT/HPBwmohIq2Ij+c4 +T3AnrGAqDdW6SLhM9k1zZNli1uofW0E9cSCaGOkCgYAI29eelAwUxpbDXJB6vSyr +LBvDV+C+/3OW4AjlJ5lWSzY8oMn21Xzp03MwXOXOSFuR6vTMkJDKJ3zIDHBt9vJ2 +evNmfMKzqNdsvNjORb79GAZ0paBF1XGzXvPO9JSUi6oZxNg+pW8oxIzx0xFIgtZL +PxnzFj2IbraxYwuR1CI6rQ== +-----END PRIVATE KEY----- diff --git a/extend/weChat/certs/1636644248/wechatpay_112FCCD1B9ECC8292703AB7363C73D74B6AFDC1A.pem b/extend/weChat/certs/1636644248/wechatpay_112FCCD1B9ECC8292703AB7363C73D74B6AFDC1A.pem new file mode 100644 index 0000000..ab18f68 --- /dev/null +++ b/extend/weChat/certs/1636644248/wechatpay_112FCCD1B9ECC8292703AB7363C73D74B6AFDC1A.pem @@ -0,0 +1,24 @@ +-----BEGIN CERTIFICATE----- +MIIEFDCCAvygAwIBAgIUES/M0bnsyCknA6tzY8c9dLav3BowDQYJKoZIhvcNAQEL +BQAwXjELMAkGA1UEBhMCQ04xEzARBgNVBAoTClRlbnBheS5jb20xHTAbBgNVBAsT +FFRlbnBheS5jb20gQ0EgQ2VudGVyMRswGQYDVQQDExJUZW5wYXkuY29tIFJvb3Qg +Q0EwHhcNMjMwMzAxMDExNzE2WhcNMjgwMjI4MDExNzE2WjBuMRgwFgYDVQQDDA9U +ZW5wYXkuY29tIHNpZ24xEzARBgNVBAoMClRlbnBheS5jb20xHTAbBgNVBAsMFFRl +bnBheS5jb20gQ0EgQ2VudGVyMQswCQYDVQQGDAJDTjERMA8GA1UEBwwIU2hlblpo +ZW4wggEiMA0GCSqGSIb3DQEBAQUAA4IBDwAwggEKAoIBAQCvM2YfnzwBvd+B5Ro1 +z+LaCVYhxia9/hQwRhi5ag2HmeZZNgMNf4+bC75Nv+t3RW0lklbM9qJSBk03eEBG +75Q33VPGy//WCJZgXlHV+rZ7ejzWtwh9Muhd60HFXhIsM3pwlfWpPo6bkJie9Na0 +wMsgg/8UxsNydhEZF6HFLdqY+zqGOjRRCduIyJcFhtrkjNUMIFAOSkHBaGJjmGrm +OXigAnYAsaD1VWLOoblA0HlOs144KQ/5Shj74Ggk2pFo/YDN+i5hazHZo9hZnjmX +G5BV3KDJD0k3Gn/qymhiLlzGsYW79P+BbsmE/M6jKIP2jxcDDdpek0Z6Lk0Cz/au +02UPAgMBAAGjgbkwgbYwCQYDVR0TBAIwADALBgNVHQ8EBAMCA/gwgZsGA1UdHwSB +kzCBkDCBjaCBiqCBh4aBhGh0dHA6Ly9ldmNhLml0cnVzLmNvbS5jbi9wdWJsaWMv +aXRydXNjcmw/Q0E9MUJENDIyMEU1MERCQzA0QjA2QUQzOTc1NDk4NDZDMDFDM0U4 +RUJEMiZzZz1IQUNDNDcxQjY1NDIyRTEyQjI3QTlEMzNBODdBRDFDREY1OTI2RTE0 +MDM3MTANBgkqhkiG9w0BAQsFAAOCAQEAJX6C/QMYF0F3IiK9P8tW2DKN8y9vCU20 +6Ws8u4bcO3AaiSmsdAJ6I1MyZkUg3dKijtnDbieY2P364IEp48TmI4k6UJwP4+/f +i0NseOm3BmAJ3mBoNmuFul+61opKpeV67AYQuhbehVA4cDPqKb/hP0tEcb6nefIO +mnys5GReLWLFV2XFR1h9QtsohPOEYlpBl6lmKHNoQZdAPtSHq2iVFLbuD7GMKoj7 +Br2puJEQgxla1AsNxAYjGttEIO9I30+L5av7SKesSGXvL6G5eOvFrZl0S4EsNVz4 +QeL4qmtFEKDi2eaxeyLuWe9TMI/IiI3ngekoDyTwdnUzadumbio+dg== +-----END CERTIFICATE----- \ No newline at end of file diff --git a/extend/weChat/certs/1659662936/apiclient_cert.p12 b/extend/weChat/certs/1659662936/apiclient_cert.p12 new file mode 100644 index 0000000..1398295 Binary files /dev/null and b/extend/weChat/certs/1659662936/apiclient_cert.p12 differ diff --git a/extend/weChat/certs/1659662936/apiclient_cert.pem b/extend/weChat/certs/1659662936/apiclient_cert.pem new file mode 100644 index 0000000..514b2af --- /dev/null +++ b/extend/weChat/certs/1659662936/apiclient_cert.pem @@ -0,0 +1,25 @@ +-----BEGIN CERTIFICATE----- +MIIENzCCAx+gAwIBAgIUEvql9hcIt5W7UzeukVSU4twsqHswDQYJKoZIhvcNAQEL +BQAwXjELMAkGA1UEBhMCQ04xEzARBgNVBAoTClRlbnBheS5jb20xHTAbBgNVBAsT +FFRlbnBheS5jb20gQ0EgQ2VudGVyMRswGQYDVQQDExJUZW5wYXkuY29tIFJvb3Qg +Q0EwHhcNMjMxMjE5MDUxNTA4WhcNMjgxMjE3MDUxNTA4WjCBkDETMBEGA1UEAwwK +MTY1OTY2MjkzNjEbMBkGA1UECgwS5b6u5L+h5ZWG5oi357O757ufMTwwOgYDVQQL +DDPmiJDpg73ph5HniZvmrKPmrKPnm7jnhafkupLogZTnvZHljLvpmaLmnInpmZDl +hazlj7gxCzAJBgNVBAYTAkNOMREwDwYDVQQHDAhTaGVuWmhlbjCCASIwDQYJKoZI +hvcNAQEBBQADggEPADCCAQoCggEBAPCINl3HwUEM2CzerR5u4GbUnvRJDjVmaEbm +qHlaYRUrrGDIF/Q63QEbcUh+PP+zZ0Grln7hGT5dwPj6e8jwmVvtEqO8mhBQzynH +6MZUONVfxj7L5Pf/+vZtB/PPGKdupZ9tUuodqOrbLycE8xWmSUDPrKEZ9iOEYamz +p0s9fPWwcZM8Vf4Gqx8d9ItnFJjVmF8aIJfsvcGNvKWD44JUzZItcsWz/srzxRAH +2moGIM8eNf1UNm3k6q43dfoRbZryWmVIyw49xybAKzm5e4Q4z8jCwGoQWgs2tSF4 +uM0/SxK2j1XRGnVmwRzcdDq/rCPlbqynHXZyUh8Te1540HJKZK8CAwEAAaOBuTCB +tjAJBgNVHRMEAjAAMAsGA1UdDwQEAwID+DCBmwYDVR0fBIGTMIGQMIGNoIGKoIGH +hoGEaHR0cDovL2V2Y2EuaXRydXMuY29tLmNuL3B1YmxpYy9pdHJ1c2NybD9DQT0x +QkQ0MjIwRTUwREJDMDRCMDZBRDM5NzU0OTg0NkMwMUMzRThFQkQyJnNnPUhBQ0M0 +NzFCNjU0MjJFMTJCMjdBOUQzM0E4N0FEMUNERjU5MjZFMTQwMzcxMA0GCSqGSIb3 +DQEBCwUAA4IBAQAe4CfX/m0Xz6aI/Studd5pbdb/M7cyTjTwYSoBUNKtEmBwQFCu +CsFOEGuqmhq4ciMkooDWY/6mccgluZtdiYB8PQAIk1+IKBgpmhSuJP6yONecoit0 +JBWzFeSLkBxKkq6CfHhb9wW4EK0oPtgE6LjqLMj13fLy+Mxf4Eu4fgSlsdgGLpBt +bXIPbL12bCcP5hJLGRoLbECyx9LaOe1D9RAxwcTx/i+8C1s3IajkkyZnoPasLMs6 +cdSIp20QSct0klMgpEobmV+Z59NMKAQC/pbaSRrT/PDHUTDAuwA7W+COALg6H26c +MAGyeSTnwhc9WccmohKPgi3UcXh/Sn84QqSs +-----END CERTIFICATE----- diff --git a/extend/weChat/certs/1659662936/apiclient_key.pem b/extend/weChat/certs/1659662936/apiclient_key.pem new file mode 100644 index 0000000..0a65bca --- /dev/null +++ b/extend/weChat/certs/1659662936/apiclient_key.pem @@ -0,0 +1,28 @@ +-----BEGIN PRIVATE KEY----- +MIIEvgIBADANBgkqhkiG9w0BAQEFAASCBKgwggSkAgEAAoIBAQDwiDZdx8FBDNgs +3q0ebuBm1J70SQ41ZmhG5qh5WmEVK6xgyBf0Ot0BG3FIfjz/s2dBq5Z+4Rk+XcD4 ++nvI8Jlb7RKjvJoQUM8px+jGVDjVX8Y+y+T3//r2bQfzzxinbqWfbVLqHajq2y8n +BPMVpklAz6yhGfYjhGGps6dLPXz1sHGTPFX+BqsfHfSLZxSY1ZhfGiCX7L3Bjbyl +g+OCVM2SLXLFs/7K88UQB9pqBiDPHjX9VDZt5OquN3X6EW2a8lplSMsOPccmwCs5 +uXuEOM/IwsBqEFoLNrUheLjNP0sSto9V0Rp1ZsEc3HQ6v6wj5W6spx12clIfE3te +eNBySmSvAgMBAAECggEBAOVndDqjLozFHFRHGGOjKgDJHsUr/BIwFpowmVQMP/V6 +DtDLZhU4ItpQew3B4JmbWIrIhSODupjBwC92bqLp3cpP5Gwnj+SpvbtCf57QatgO +nTv9KObizE9FE8WTqhbeL7ZLBT5mhVlhLKqRTOpECy92IlYQNbIQKzk4MAFRpqGH +CqSMtD0ng6DiMDxdQYLgIDCZIKamv0lF3rQymINQ6VZHpdlB0f3Gp/T/yWzn0nmJ +5eOrBhzFKGpevsp00FmvFE7HSqnogC7a3ieLluhfRPyB4QZQjfMorEGKwS8i7I8U +ZLA9FhtsaZpvOeNSZoVh3z0sblKUOvhXvL+jgRQK4PkCgYEA+M0QxM3VXst+deDr +ZkpmhWdPUQykVdSbfOdR8thYDu30QdjKe/T0X75wcqKh+s2GPK2GZsgeU02+EWqv +gVZMgFwGKFCBzDiCs9QwOTO+32OWP62yapFWNba1g4K7Un7MJRsJIj0u0nwjik68 +hec5KFYTOYttzk9oOLgXLFM6zm0CgYEA933lgvPITFykd2kGKg1v4oA52PPmq9LT +2dvVkDpIxioaUUHPALwFLhHLT+w/RPwCfP9gqZAQQFAfpMiTezm+ULA54rLHF2Tj +Lm0q3ko98sFW3Ltk1ehw8CdU0p5bii7a7AbuS6V36iTIrC53rhl7N+WL82uMw2w/ +a8Gz3fIl3gsCgYA+XsHeZC8iBWddS5YXXX1X4e8bRU0JCzQzWpXLh/qDO5mozBzu +eBiuy8HKqwRqKA2HtoRjzbT0cx+7o//9L1IcN3V/s7bmKCBzzjSMknE99OwcaIG6 +f1aaPoRARIyLAKhSgPWINMhBEcejC1vtQWqttu441cAgIP3ighulC/RI/QKBgQDE +lxKqdK8USTqzR4+H8+h+CNDqjsMalXuGwGLiEAoirur8xMODl9adg7D2KXkQeQYY ++Fp2FmNyUrLwGAtehL2yJmm0s8IFyTPUew7kSCDxJbaz2377k4mymet86iFYoGNx +vQeouyWHrfRhIQAcIU2JVyNWFoZX7TJrjBAuKtz9hQKBgGUFIyhQiPYRWDocvuSi +McyzYov2AnNehn24vsXC3SCXT804KPOX8RY4he63zWMGaLfCmSq1fhsJnf2pvffK +W+9ZXlIH6dwgQ9QyDNjKRVCGwTASsWOyqgn1Be0VDuuoXlREyGNQSn2RpvL1SMFH +PW/ePgdfoFkzU5WFJFPnvlU2 +-----END PRIVATE KEY----- diff --git a/extend/weChat/certs/1659662936/wechatpay_5B5C8A69CC86D1127F6B6AA06AAAF10531EEFE90.pem b/extend/weChat/certs/1659662936/wechatpay_5B5C8A69CC86D1127F6B6AA06AAAF10531EEFE90.pem new file mode 100644 index 0000000..290c3c2 --- /dev/null +++ b/extend/weChat/certs/1659662936/wechatpay_5B5C8A69CC86D1127F6B6AA06AAAF10531EEFE90.pem @@ -0,0 +1,24 @@ +-----BEGIN CERTIFICATE----- +MIIEFDCCAvygAwIBAgIUW1yKacyG0RJ/a2qgaqrxBTHu/pAwDQYJKoZIhvcNAQEL +BQAwXjELMAkGA1UEBhMCQ04xEzARBgNVBAoTClRlbnBheS5jb20xHTAbBgNVBAsT +FFRlbnBheS5jb20gQ0EgQ2VudGVyMRswGQYDVQQDExJUZW5wYXkuY29tIFJvb3Qg +Q0EwHhcNMjMxMjE5MDUxNTA3WhcNMjgxMjE3MDUxNTA3WjBuMRgwFgYDVQQDDA9U +ZW5wYXkuY29tIHNpZ24xEzARBgNVBAoMClRlbnBheS5jb20xHTAbBgNVBAsMFFRl +bnBheS5jb20gQ0EgQ2VudGVyMQswCQYDVQQGEwJDTjERMA8GA1UEBwwIU2hlblpo +ZW4wggEiMA0GCSqGSIb3DQEBAQUAA4IBDwAwggEKAoIBAQClp0R6MR6NwEkqRJNF +aD1INE0uHoRg1of0D8mTVwoWvADgt9gutpeJ1uTweAtrXkvu9NALxTNaG51tsZa8 +2fsJ00/Ry1IYUb6xG7kAuh0h4KLgWUUhoXDOKW0B1K/7g3AhCAylUzhD/ghD1Y8F +hlMtw87oNzfTE51I/2sGZgnPX2IqIEOjxx1F8Ebzfh5shJGN40guMpBItsCWYe7s +zj1zdyERL6zxQN2A7o2QvDeX1EsNIdwoEECv06tWjUEGJioFwb3OKBu5n9jJT+Og +EcJFmRasLRQWsooutsDCO7y5wIKrCY9n52eZgk9QlHkYWyiwUqpnrrJ1DC+ueJei +QptFAgMBAAGjgbkwgbYwCQYDVR0TBAIwADALBgNVHQ8EBAMCA/gwgZsGA1UdHwSB +kzCBkDCBjaCBiqCBh4aBhGh0dHA6Ly9ldmNhLml0cnVzLmNvbS5jbi9wdWJsaWMv +aXRydXNjcmw/Q0E9MUJENDIyMEU1MERCQzA0QjA2QUQzOTc1NDk4NDZDMDFDM0U4 +RUJEMiZzZz1IQUNDNDcxQjY1NDIyRTEyQjI3QTlEMzNBODdBRDFDREY1OTI2RTE0 +MDM3MTANBgkqhkiG9w0BAQsFAAOCAQEAhWwscSAE1OBR3AGBOuFHs3vFJ+y37z5/ +EoB9FwVu4zXP1dih101zl83iwIdPEHIR2skXcTjHRI2qdOvu7X5JmWOJP+51RGtX +Y+aWXKfhRzRQOUlpNyltlgsGKzbIXLLBzQjMzBNv+n/HX4q9F0TV3SW4zTiMlhD8 ++bGGGwuIhziWpK9qvr3RPU1j+0bggHhIre+cNolnh1FepS4Gt964zhx6THtrS/jI +I64UBBh6moBq7zB5QYloBhW464c7GCEEv5/AdcxGhAe+vuL/mkVRNsSRxVPIxPE6 ++qoIiNBmQvL/mL+4UKfX6b9h4wrUQUdQP3ljRdpL3a5YTMTUJuoJSQ== +-----END CERTIFICATE----- \ No newline at end of file diff --git a/extend/weChat/close.go b/extend/weChat/close.go new file mode 100644 index 0000000..d276b53 --- /dev/null +++ b/extend/weChat/close.go @@ -0,0 +1,62 @@ +package weChat + +import ( + "context" + "errors" + "github.com/wechatpay-apiv3/wechatpay-go/core" + "github.com/wechatpay-apiv3/wechatpay-go/services/payments/app" + "github.com/wechatpay-apiv3/wechatpay-go/services/payments/jsapi" + "vote-admin-video-api/config" +) + +// CloseJsapiOrder 关闭jsapi支付订单 +func CloseJsapiOrder(outTradeNo string) error { + client, err := createClient() + if err != nil { + return err + } + + svc := jsapi.JsapiApiService{Client: client} + + req := jsapi.CloseOrderRequest{ + OutTradeNo: &outTradeNo, + Mchid: core.String(config.C.Wechat.Pay1281030301.MchId), + } + + result, err := svc.CloseOrder(context.TODO(), req) + if err != nil { + return err + } + + if result.Response.StatusCode != 204 { + return errors.New("关闭订单失败") + } + + return nil +} + +// CloseAppOrder 关闭app支付订单 +func CloseAppOrder(outTradeNo string) error { + client, err := createClient() + if err != nil { + return err + } + + svc := app.AppApiService{Client: client} + + req := app.CloseOrderRequest{ + OutTradeNo: &outTradeNo, + Mchid: core.String(config.C.Wechat.Pay1281030301.MchId), + } + + result, err := svc.CloseOrder(context.TODO(), req) + if err != nil { + return err + } + + if result.Response.StatusCode != 204 { + return errors.New("关闭订单失败") + } + + return nil +} diff --git a/extend/weChat/parseNotify.go b/extend/weChat/parseNotify.go new file mode 100644 index 0000000..4df8d31 --- /dev/null +++ b/extend/weChat/parseNotify.go @@ -0,0 +1,82 @@ +package weChat + +import ( + "context" + "errors" + "github.com/gin-gonic/gin" + "github.com/wechatpay-apiv3/wechatpay-go/core/auth/verifiers" + "github.com/wechatpay-apiv3/wechatpay-go/core/downloader" + "github.com/wechatpay-apiv3/wechatpay-go/core/notify" + "github.com/wechatpay-apiv3/wechatpay-go/services/payments" + "github.com/wechatpay-apiv3/wechatpay-go/services/refunddomestic" + "github.com/wechatpay-apiv3/wechatpay-go/utils" + "vote-admin-video-api/config" +) + +// PayParseNotify 支付回调通知的验签与解密 +func PayParseNotify(c *gin.Context) (notifyReq *notify.Request, t *payments.Transaction, err error) { + mchId := config.C.Wechat.Pay1281030301.MchId // 商户号 + mchCertificateSerialNumber := config.C.Wechat.Pay1281030301.MchCertificateSerialNumber // 商户证书序列号 + v3ApiSecret := config.C.Wechat.Pay1281030301.V3ApiSecret // 商户APIv3密钥 + privateKeyPath := "extend/weChat/certs/" + config.C.Wechat.Pay1281030301.MchId + "/apiclient_key.pem" // 商户私钥文件地址 + + // 使用 utils 提供的函数从本地文件中加载商户私钥，商户私钥会用来生成请求的签名 + mchPrivateKey, err := utils.LoadPrivateKeyWithPath(privateKeyPath) + if err != nil { + return nil, nil, errors.New("微信支付生成失败") + } + + // 1. 使用 `RegisterDownloaderWithPrivateKey` 注册下载器 + err = downloader.MgrInstance().RegisterDownloaderWithPrivateKey(c, mchPrivateKey, mchCertificateSerialNumber, mchId, v3ApiSecret) + if err != nil { + return nil, nil, err + } + + // 2. 获取商户号对应的微信支付平台证书访问器 + certificateVisitor := downloader.MgrInstance().GetCertificateVisitor(mchId) + // 3. 使用证书访问器初始化 `notify.Handler` + handler, err := notify.NewRSANotifyHandler(v3ApiSecret, verifiers.NewSHA256WithRSAVerifier(certificateVisitor)) + + transaction := new(payments.Transaction) + notifyReq, err = handler.ParseNotifyRequest(context.Background(), c.Request, transaction) + // 如果验签未通过，或者解密失败 + if err != nil { + return nil, nil, err + } + + return notifyReq, transaction, nil +} + +// RefundParseNotify 退款回调通知的验签与解密 +func RefundParseNotify(c *gin.Context) (notifyReq *notify.Request, t *refunddomestic.Refund, err error) { + mchId := config.C.Wechat.Pay1281030301.MchId // 商户号 + mchCertificateSerialNumber := config.C.Wechat.Pay1281030301.MchCertificateSerialNumber // 商户证书序列号 + v3ApiSecret := config.C.Wechat.Pay1281030301.V3ApiSecret // 商户APIv3密钥 + privateKeyPath := "extend/weChat/certs/" + config.C.Wechat.Pay1281030301.MchId + "/apiclient_key.pem" // 商户私钥文件地址 + + // 使用 utils 提供的函数从本地文件中加载商户私钥，商户私钥会用来生成请求的签名 + mchPrivateKey, err := utils.LoadPrivateKeyWithPath(privateKeyPath) + if err != nil { + return nil, nil, errors.New("微信支付生成失败") + } + + // 1. 使用 `RegisterDownloaderWithPrivateKey` 注册下载器 + err = downloader.MgrInstance().RegisterDownloaderWithPrivateKey(c, mchPrivateKey, mchCertificateSerialNumber, mchId, v3ApiSecret) + if err != nil { + return nil, nil, err + } + + // 2. 获取商户号对应的微信支付平台证书访问器 + certificateVisitor := downloader.MgrInstance().GetCertificateVisitor(mchId) + // 3. 使用证书访问器初始化 `notify.Handler` + handler, err := notify.NewRSANotifyHandler(v3ApiSecret, verifiers.NewSHA256WithRSAVerifier(certificateVisitor)) + + refund := new(refunddomestic.Refund) + notifyReq, err = handler.ParseNotifyRequest(context.Background(), c.Request, refund) + // 如果验签未通过，或者解密失败 + if err != nil { + return nil, nil, err + } + + return notifyReq, refund, nil +} diff --git a/extend/weChat/prepay.go b/extend/weChat/prepay.go new file mode 100644 index 0000000..08e49df --- /dev/null +++ b/extend/weChat/prepay.go @@ -0,0 +1,142 @@ +package weChat + +import ( + "context" + "errors" + "github.com/wechatpay-apiv3/wechatpay-go/core" + "github.com/wechatpay-apiv3/wechatpay-go/services/payments/app" + "github.com/wechatpay-apiv3/wechatpay-go/services/payments/jsapi" +) + +/** +JSAPI下单 +APP下单 +*/ + +// JsapiRequest 请求数据-JSAPI下单 +type JsapiRequest struct { + AppId string `json:"appid" comment:"公众号ID"` + MchId string `json:"mchid" comment:"直连商户号"` + Description string `json:"description" comment:"商品描述"` + OutTradeNo string `json:"out_trade_no" comment:"商户订单号"` + NotifyUrl string `json:"notify_url" comment:"回调地址"` + Amount JsapiRequestAmountRequest `json:"amount" comment:"订单金额"` + Payer JsapiRequestPayerRequest `json:"payer" comment:"支付者"` +} + +// JsapiRequestAmountRequest 订单金额信息 +type JsapiRequestAmountRequest struct { + Total int64 `json:"total" comment:"订单总金额"` + Currency string `json:"currency" comment:"货币类型"` +} + +// JsapiRequestPayerRequest 支付者信息 +type JsapiRequestPayerRequest struct { + OpenId string `json:"openid" comment:"openid"` +} + +// AppRequest 请求数据-APP下单 +type AppRequest struct { + AppId string `json:"appid" comment:"公众号ID"` + MchId string `json:"mchid" comment:"直连商户号"` + Description string `json:"description" comment:"商品描述"` + OutTradeNo string `json:"out_trade_no" comment:"商户订单号"` + NotifyUrl string `json:"notify_url" comment:"回调地址"` + Amount AppRequestAmountRequest `json:"amount" comment:"订单金额"` +} + +// AppRequestAmountRequest 订单金额信息 +type AppRequestAmountRequest struct { + Total int64 `json:"total" comment:"订单总金额"` + Currency string `json:"currency" comment:"货币类型"` +} + +// GetJsapiPrepay JSAPI下单-获取jsapi预支付交易会话 +func (r JsapiRequest) GetJsapiPrepay() (prepay *jsapi.PrepayWithRequestPaymentResponse, err error) { + client, err := createClient() + if err != nil { + return nil, err + } + + svc := jsapi.JsapiApiService{Client: client} + // 得到prepay_id，以及调起支付所需的参数和签名 + PrepayRequest := jsapi.PrepayRequest{ + Appid: core.String(r.AppId), + Mchid: core.String(r.MchId), + Description: core.String(r.Description), + OutTradeNo: core.String(r.OutTradeNo), + NotifyUrl: core.String(r.NotifyUrl), + Amount: &jsapi.Amount{ + Total: core.Int64(r.Amount.Total), + Currency: core.String("CNY"), + }, + Payer: &jsapi.Payer{ + Openid: core.String(r.Payer.OpenId), + }, + } + + resp, result, err := svc.PrepayWithRequestPayment(context.Background(), + PrepayRequest, + ) + + if err != nil { + return nil, err + } + + if result.Response.StatusCode != 200 { + return nil, errors.New("发起支付失败") + } + + if resp.PrepayId == nil { + return nil, errors.New("发起支付失败") + } + + if err != nil { + return nil, err + } + + return resp, nil +} + +// GetAppPrepay APP下单-获取app预支付交易会话 +func (r AppRequest) GetAppPrepay() (prepay *app.PrepayWithRequestPaymentResponse, err error) { + client, err := createClient() + if err != nil { + return nil, err + } + + svc := app.AppApiService{Client: client} + // 得到prepay_id，以及调起支付所需的参数和签名 + + resp, result, err := svc.PrepayWithRequestPayment(context.Background(), + app.PrepayRequest{ + Appid: core.String(r.AppId), + Mchid: core.String(r.MchId), + Description: core.String(r.Description), + OutTradeNo: core.String(r.OutTradeNo), + NotifyUrl: core.String(r.NotifyUrl), + Amount: &app.Amount{ + Total: core.Int64(r.Amount.Total), + Currency: core.String("CNY"), + }, + }, + ) + + if err != nil { + return nil, err + } + + if result.Response.StatusCode != 200 { + return nil, errors.New("发起支付失败") + } + + if resp.PrepayId == nil { + return nil, errors.New("发起支付失败") + } + + if err != nil { + return nil, err + } + + return resp, nil +} diff --git a/extend/weChat/refund.go b/extend/weChat/refund.go new file mode 100644 index 0000000..86c1a02 --- /dev/null +++ b/extend/weChat/refund.go @@ -0,0 +1,46 @@ +package weChat + +import ( + "context" + "github.com/wechatpay-apiv3/wechatpay-go/core" + "github.com/wechatpay-apiv3/wechatpay-go/services/refunddomestic" +) + +type RefundRequest struct { + TransactionId string `json:"transaction_id" comment:"微信订单号"` + OutTradeNo string `json:"out_trade_no" comment:"商户订单号"` + OutRefundNo string `json:"out_refund_no" comment:"退款订单号"` + Reason string `json:"reason" comment:"退款原因"` + RefundAmount int64 `json:"refund_amount" comment:"退款金额"` + PaymentAmountTotal int64 `json:"payment_amount_total" comment:"支付金额"` + NotifyUrl string `json:"notify_url" comment:"回调地址"` +} + +// Refund 退款 +func (r RefundRequest) Refund() (*refunddomestic.Refund, error) { + client, err := createClient() + if err != nil { + return nil, err + } + + refundRequest := refunddomestic.CreateRequest{ + TransactionId: core.String(r.TransactionId), + OutTradeNo: core.String(r.OutTradeNo), + OutRefundNo: core.String(r.OutRefundNo), + Reason: core.String(r.Reason), + NotifyUrl: core.String(r.NotifyUrl), + Amount: &refunddomestic.AmountReq{ + Currency: core.String("CNY"), + Refund: core.Int64(r.RefundAmount), + Total: core.Int64(r.PaymentAmountTotal), + }, + } + + svc := refunddomestic.RefundsApiService{Client: client} + resp, _, err := svc.Create(context.Background(), refundRequest) + if err != nil { + return nil, err + } + + return resp, nil +} diff --git a/extend/weChat/userInfo.go b/extend/weChat/userInfo.go new file mode 100644 index 0000000..92b1131 --- /dev/null +++ b/extend/weChat/userInfo.go @@ -0,0 +1,76 @@ +package weChat + +import ( + "encoding/json" + "errors" + "io" + "net/http" +) + +// GetUserInfoResponse 网页授权拉取用户信息返回值 +type GetUserInfoResponse struct { + OpenId string `json:"openid" form:"openid" label:"openid"` + Nickname string `json:"nickname" form:"nickname" label:"用户昵称"` + Sex int `json:"sex" form:"sex" label:"性别"` // 用户的性别，值为1时是男性，值为2时是女性，值为0时是未知 + Province string `json:"province" form:"province" label:"省份"` + City string `json:"city" form:"city" label:"城市"` + Country string `json:"country" form:"country" label:"国家"` + HeadImgUrl string `json:"headimgurl" form:"headimgurl" label:"头像"` + UnionId string `json:"unionid" form:"unionid" label:"unionid"` + Errcode *int `json:"errcode" form:"errcode" label:"errcode"` + Errmsg string `json:"errmsg" form:"errmsg" label:"errmsg"` +} + +// GetUserInfo 网页授权拉取用户信息 +func GetUserInfo(accessToken, openId string) (r *GetUserInfoResponse, err error) { + if accessToken == "" { + return nil, errors.New("授权失败") + } + + if openId == "" { + return nil, errors.New("授权失败") + } + + // 拼接请求数据 + requestUrl := "https://api.weixin.qq.com/sns/userinfo?" + + "access_token=" + accessToken + + "&openid=" + openId + + "&lang=zh_CN" + + // 发送GET请求 + resp, err := http.Get(requestUrl) + if err != nil { + return nil, err + } + + defer func(Body io.ReadCloser) { + _ = Body.Close() + }(resp.Body) + + if resp.StatusCode != 200 { + return nil, errors.New("请求失败") + } + + // 读取响应体 + respBody, err := io.ReadAll(resp.Body) + if err != nil { + return nil, err + } + + var response GetUserInfoResponse + err = json.Unmarshal([]byte(respBody), &response) + if err != nil { + // json解析失败 + return nil, err + } + + if response.Errcode != nil { + if response.Errmsg != "" { + return nil, errors.New(response.Errmsg) + } else { + return nil, errors.New("请求失败") + } + } + + return &response, nil +} diff --git a/extend/weChat/webAccessToken.go b/extend/weChat/webAccessToken.go new file mode 100644 index 0000000..724f210 --- /dev/null +++ b/extend/weChat/webAccessToken.go @@ -0,0 +1,68 @@ +package weChat + +import ( + "encoding/json" + "errors" + "io" + "net/http" + "vote-admin-video-api/config" +) + +// GetWebAccessTokenResponse 获取网页授权access_token返回值 +type GetWebAccessTokenResponse struct { + AccessToken string `json:"access_token" form:"access_token" label:"网页授权接口调用凭证"` + ExpiresIn int `json:"expires_in" form:"expires_in" label:"access_token接口调用凭证超时时间"` + RefreshToken string `json:"refresh_token" form:"refresh_token" label:"用户刷新access_token"` + OpenId string `json:"openid" form:"openid" label:"openid"` + Scope string `json:"scope" form:"scope" label:"scope"` + UnionId string `json:"unionid" form:"unionid" label:"unionid"` + Errcode *int `json:"errcode" form:"errcode" label:"errcode"` + Errmsg string `json:"errmsg" form:"errmsg" label:"errmsg"` +} + +// GetWebAccessToken 获取网页授权access_token +func GetWebAccessToken(code string) (r *GetWebAccessTokenResponse, err error) { + // 拼接请求数据 + requestUrl := "https://api.weixin.qq.com/sns/oauth2/access_token?" + + "appid=" + config.C.Wechat.AppId + + "&secret=" + config.C.Wechat.AppSecret + + "&code=" + code + + "&grant_type=authorization_code" + + // 发送GET请求 + resp, err := http.Get(requestUrl) + if err != nil { + return nil, err + } + + defer func(Body io.ReadCloser) { + _ = Body.Close() + }(resp.Body) + + if resp.StatusCode != 200 { + return nil, errors.New("请求失败") + } + + // 读取响应体 + respBody, err := io.ReadAll(resp.Body) + if err != nil { + return nil, err + } + + var response GetWebAccessTokenResponse + err = json.Unmarshal([]byte(respBody), &response) + if err != nil { + // json解析失败 + return nil, err + } + + if response.Errcode != nil { + if response.Errmsg != "" { + return nil, errors.New(response.Errmsg) + } else { + return nil, errors.New("请求失败") + } + } + + return &response, nil +} diff --git a/global/global.go b/global/global.go new file mode 100644 index 0000000..c026d8c --- /dev/null +++ b/global/global.go @@ -0,0 +1,21 @@ +package global + +import ( + "github.com/bwmarrin/snowflake" + ut "github.com/go-playground/universal-translator" + "github.com/go-playground/validator/v10" + "github.com/go-redis/redis/v8" + "github.com/sirupsen/logrus" + "gorm.io/gorm" +) + +// 全局变量 +var ( + Db *gorm.DB // 数据库 + Logger *logrus.Logger // 日志 + Redis *redis.Client // redis + Validate *validator.Validate // 验证器 + Trans ut.Translator // Validate/v10 全局验证器 + Snowflake *snowflake.Node // 雪花算法 + UserId int64 // 用户id +) diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..cc9952e --- /dev/null +++ b/go.mod @@ -0,0 +1,98 @@ +module vote-admin-video-api + +go 1.19 + +require ( + github.com/alibabacloud-go/darabonba-openapi/v2 v2.0.9 + github.com/alibabacloud-go/dysmsapi-20170525/v3 v3.0.6 + github.com/alibabacloud-go/tea v1.2.2 + github.com/alibabacloud-go/tea-utils/v2 v2.0.6 + github.com/aliyun/aliyun-oss-go-sdk v3.0.2+incompatible + github.com/bwmarrin/snowflake v0.3.0 + github.com/disintegration/imaging v1.6.2 + github.com/facebookarchive/grace v0.0.0-20180706040059-75cf19382434 + github.com/fsnotify/fsnotify v1.7.0 + github.com/gin-gonic/gin v1.10.0 + github.com/go-playground/locales v0.14.1 + github.com/go-playground/universal-translator v0.18.1 + github.com/go-playground/validator/v10 v10.22.0 + github.com/go-redis/redis/v8 v8.11.5 + github.com/golang-jwt/jwt/v5 v5.2.1 + github.com/mojocn/base64Captcha v1.3.6 + github.com/sirupsen/logrus v1.9.3 + github.com/spf13/viper v1.19.0 + github.com/wechatpay-apiv3/wechatpay-go v0.2.20 + github.com/xuri/excelize/v2 v2.8.1 + gorm.io/driver/mysql v1.5.7 + gorm.io/gorm v1.25.11 +) + +require ( + github.com/alibabacloud-go/alibabacloud-gateway-spi v0.0.4 // indirect + github.com/alibabacloud-go/debug v1.0.0 // indirect + github.com/alibabacloud-go/endpoint-util v1.1.0 // indirect + github.com/alibabacloud-go/openapi-util v0.1.0 // indirect + github.com/alibabacloud-go/tea-utils v1.3.1 // indirect + github.com/alibabacloud-go/tea-xml v1.1.3 // indirect + github.com/aliyun/credentials-go v1.3.1 // indirect + github.com/bytedance/sonic v1.11.6 // indirect + github.com/bytedance/sonic/loader v0.1.1 // indirect + github.com/cespare/xxhash/v2 v2.1.2 // indirect + github.com/clbanning/mxj/v2 v2.5.5 // indirect + github.com/cloudwego/base64x v0.1.4 // indirect + github.com/cloudwego/iasm v0.2.0 // indirect + github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect + github.com/facebookgo/clock v0.0.0-20150410010913-600d898af40a // indirect + github.com/facebookgo/ensure v0.0.0-20200202191622-63f1cf65ac4c // indirect + github.com/facebookgo/freeport v0.0.0-20150612182905-d4adf43b75b9 // indirect + github.com/facebookgo/grace v0.0.0-20180706040059-75cf19382434 // indirect + github.com/facebookgo/httpdown v0.0.0-20180706035922-5979d39b15c2 // indirect + github.com/facebookgo/stack v0.0.0-20160209184415-751773369052 // indirect + github.com/facebookgo/stats v0.0.0-20151006221625-1b76add642e4 // indirect + github.com/facebookgo/subset v0.0.0-20200203212716-c811ad88dec4 // indirect + github.com/gabriel-vasile/mimetype v1.4.3 // indirect + github.com/gin-contrib/sse v0.1.0 // indirect + github.com/go-sql-driver/mysql v1.7.0 // indirect + github.com/goccy/go-json v0.10.2 // indirect + github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 // indirect + github.com/hashicorp/hcl v1.0.0 // indirect + github.com/jinzhu/inflection v1.0.0 // indirect + github.com/jinzhu/now v1.1.5 // indirect + github.com/json-iterator/go v1.1.12 // indirect + github.com/klauspost/cpuid/v2 v2.2.7 // indirect + github.com/leodido/go-urn v1.4.0 // indirect + github.com/magiconair/properties v1.8.7 // indirect + github.com/mattn/go-isatty v0.0.20 // indirect + github.com/mitchellh/mapstructure v1.5.0 // indirect + github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect + github.com/modern-go/reflect2 v1.0.2 // indirect + github.com/mohae/deepcopy v0.0.0-20170929034955-c48cc78d4826 // indirect + github.com/pelletier/go-toml/v2 v2.2.2 // indirect + github.com/richardlehane/mscfb v1.0.4 // indirect + github.com/richardlehane/msoleps v1.0.3 // indirect + github.com/sagikazarmark/locafero v0.4.0 // indirect + github.com/sagikazarmark/slog-shim v0.1.0 // indirect + github.com/sourcegraph/conc v0.3.0 // indirect + github.com/spf13/afero v1.11.0 // indirect + github.com/spf13/cast v1.6.0 // indirect + github.com/spf13/pflag v1.0.5 // indirect + github.com/subosito/gotenv v1.6.0 // indirect + github.com/tjfoc/gmsm v1.3.2 // indirect + github.com/twitchyliquid64/golang-asm v0.15.1 // indirect + github.com/ugorji/go/codec v1.2.12 // indirect + github.com/xuri/efp v0.0.0-20231025114914-d1ff6096ae53 // indirect + github.com/xuri/nfp v0.0.0-20230919160717-d98342af3f05 // indirect + go.uber.org/atomic v1.9.0 // indirect + go.uber.org/multierr v1.9.0 // indirect + golang.org/x/arch v0.8.0 // indirect + golang.org/x/crypto v0.23.0 // indirect + golang.org/x/exp v0.0.0-20230905200255-921286631fa9 // indirect + golang.org/x/image v0.14.0 // indirect + golang.org/x/net v0.25.0 // indirect + golang.org/x/sys v0.20.0 // indirect + golang.org/x/text v0.15.0 // indirect + golang.org/x/time v0.5.0 // indirect + google.golang.org/protobuf v1.34.1 // indirect + gopkg.in/ini.v1 v1.67.0 // indirect + gopkg.in/yaml.v3 v3.0.1 // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..df42ced --- /dev/null +++ b/go.sum @@ -0,0 +1,322 @@ +github.com/agiledragon/gomonkey v2.0.2+incompatible h1:eXKi9/piiC3cjJD1658mEE2o3NjkJ5vDLgYjCQu0Xlw= +github.com/agiledragon/gomonkey v2.0.2+incompatible/go.mod h1:2NGfXu1a80LLr2cmWXGBDaHEjb1idR6+FVlX5T3D9hw= +github.com/alibabacloud-go/alibabacloud-gateway-spi v0.0.4 h1:iC9YFYKDGEy3n/FtqJnOkZsene9olVspKmkX5A2YBEo= +github.com/alibabacloud-go/alibabacloud-gateway-spi v0.0.4/go.mod h1:sCavSAvdzOjul4cEqeVtvlSaSScfNsTQ+46HwlTL1hc= +github.com/alibabacloud-go/darabonba-openapi/v2 v2.0.2/go.mod h1:5JHVmnHvGzR2wNdgaW1zDLQG8kOC4Uec8ubkMogW7OQ= +github.com/alibabacloud-go/darabonba-openapi/v2 v2.0.9 h1:fxMCrZatZfXq5nLcgkmWBXmU3FLC1OR+m/SqVtMqflk= +github.com/alibabacloud-go/darabonba-openapi/v2 v2.0.9/go.mod h1:bb+Io8Sn2RuM3/Rpme6ll86jMyFSrD1bxeV/+v61KeU= +github.com/alibabacloud-go/debug v0.0.0-20190504072949-9472017b5c68/go.mod h1:6pb/Qy8c+lqua8cFpEy7g39NRRqOWc3rOwAy8m5Y2BY= +github.com/alibabacloud-go/debug v1.0.0 h1:3eIEQWfay1fB24PQIEzXAswlVJtdQok8f3EVN5VrBnA= +github.com/alibabacloud-go/debug v1.0.0/go.mod h1:8gfgZCCAC3+SCzjWtY053FrOcd4/qlH6IHTI4QyICOc= +github.com/alibabacloud-go/dysmsapi-20170525/v3 v3.0.6 h1:UTl97mt2qfavxveqCkaVg4tKaZUPzA9RKbFIRaIdtdg= +github.com/alibabacloud-go/dysmsapi-20170525/v3 v3.0.6/go.mod h1:UWpcGrWwTbES9QW7OQ7xDffukMJ/l7lzioixIz8+lgY= +github.com/alibabacloud-go/endpoint-util v1.1.0 h1:r/4D3VSw888XGaeNpP994zDUaxdgTSHBbVfZlzf6b5Q= +github.com/alibabacloud-go/endpoint-util v1.1.0/go.mod h1:O5FuCALmCKs2Ff7JFJMudHs0I5EBgecXXxZRyswlEjE= +github.com/alibabacloud-go/openapi-util v0.0.11/go.mod h1:sQuElr4ywwFRlCCberQwKRFhRzIyG4QTP/P4y1CJ6Ws= +github.com/alibabacloud-go/openapi-util v0.1.0 h1:0z75cIULkDrdEhkLWgi9tnLe+KhAFE/r5Pb3312/eAY= +github.com/alibabacloud-go/openapi-util v0.1.0/go.mod h1:sQuElr4ywwFRlCCberQwKRFhRzIyG4QTP/P4y1CJ6Ws= +github.com/alibabacloud-go/tea v1.1.0/go.mod h1:IkGyUSX4Ba1V+k4pCtJUc6jDpZLFph9QMy2VUPTwukg= +github.com/alibabacloud-go/tea v1.1.7/go.mod h1:/tmnEaQMyb4Ky1/5D+SE1BAsa5zj/KeGOFfwYm3N/p4= +github.com/alibabacloud-go/tea v1.1.8/go.mod h1:/tmnEaQMyb4Ky1/5D+SE1BAsa5zj/KeGOFfwYm3N/p4= +github.com/alibabacloud-go/tea v1.1.17/go.mod h1:nXxjm6CIFkBhwW4FQkNrolwbfon8Svy6cujmKFUq98A= +github.com/alibabacloud-go/tea v1.1.19/go.mod h1:nXxjm6CIFkBhwW4FQkNrolwbfon8Svy6cujmKFUq98A= +github.com/alibabacloud-go/tea v1.2.1/go.mod h1:qbzof29bM/IFhLMtJPrgTGK3eauV5J2wSyEUo4OEmnA= +github.com/alibabacloud-go/tea v1.2.2 h1:aTsR6Rl3ANWPfqeQugPglfurloyBJY85eFy7Gc1+8oU= +github.com/alibabacloud-go/tea v1.2.2/go.mod h1:CF3vOzEMAG+bR4WOql8gc2G9H3EkH3ZLAQdpmpXMgwk= +github.com/alibabacloud-go/tea-utils v1.3.1 h1:iWQeRzRheqCMuiF3+XkfybB3kTgUXkXX+JMrqfLeB2I= +github.com/alibabacloud-go/tea-utils v1.3.1/go.mod h1:EI/o33aBfj3hETm4RLiAxF/ThQdSngxrpF8rKUDJjPE= +github.com/alibabacloud-go/tea-utils/v2 v2.0.0/go.mod h1:U5MTY10WwlquGPS34DOeomUGBB0gXbLueiq5Trwu0C4= +github.com/alibabacloud-go/tea-utils/v2 v2.0.3/go.mod h1:sj1PbjPodAVTqGTA3olprfeeqqmwD0A5OQz94o9EuXQ= +github.com/alibabacloud-go/tea-utils/v2 v2.0.6 h1:ZkmUlhlQbaDC+Eba/GARMPy6hKdCLiSke5RsN5LcyQ0= +github.com/alibabacloud-go/tea-utils/v2 v2.0.6/go.mod h1:qxn986l+q33J5VkialKMqT/TTs3E+U9MJpd001iWQ9I= +github.com/alibabacloud-go/tea-xml v1.1.2/go.mod h1:Rq08vgCcCAjHyRi/M7xlHKUykZCEtyBy9+DPF6GgEu8= +github.com/alibabacloud-go/tea-xml v1.1.3 h1:7LYnm+JbOq2B+T/B0fHC4Ies4/FofC4zHzYtqw7dgt0= +github.com/alibabacloud-go/tea-xml v1.1.3/go.mod h1:Rq08vgCcCAjHyRi/M7xlHKUykZCEtyBy9+DPF6GgEu8= +github.com/aliyun/aliyun-oss-go-sdk v3.0.2+incompatible h1:8psS8a+wKfiLt1iVDX79F7Y6wUM49Lcha2FMXt4UM8g= +github.com/aliyun/aliyun-oss-go-sdk v3.0.2+incompatible/go.mod h1:T/Aws4fEfogEE9v+HPhhw+CntffsBHJ8nXQCwKr0/g8= +github.com/aliyun/credentials-go v1.1.2/go.mod h1:ozcZaMR5kLM7pwtCMEpVmQ242suV6qTJya2bDq4X1Tw= +github.com/aliyun/credentials-go v1.3.1 h1:uq/0v7kWrxmoLGpqjx7vtQ/s03f0zR//0br/xWDTE28= +github.com/aliyun/credentials-go v1.3.1/go.mod h1:8jKYhQuDawt8x2+fusqa1Y6mPxemTsBEN04dgcAcYz0= +github.com/bwmarrin/snowflake v0.3.0 h1:xm67bEhkKh6ij1790JB83OujPR5CzNe8QuQqAgISZN0= +github.com/bwmarrin/snowflake v0.3.0/go.mod h1:NdZxfVWX+oR6y2K0o6qAYv6gIOP9rjG0/E9WsDpxqwE= +github.com/bytedance/sonic v1.11.6 h1:oUp34TzMlL+OY1OUWxHqsdkgC/Zfc85zGqw9siXjrc0= +github.com/bytedance/sonic v1.11.6/go.mod h1:LysEHSvpvDySVdC2f87zGWf6CIKJcAvqab1ZaiQtds4= +github.com/bytedance/sonic/loader v0.1.1 h1:c+e5Pt1k/cy5wMveRDyk2X4B9hF4g7an8N3zCYjJFNM= +github.com/bytedance/sonic/loader v0.1.1/go.mod h1:ncP89zfokxS5LZrJxl5z0UJcsk4M4yY2JpfqGeCtNLU= +github.com/cespare/xxhash/v2 v2.1.2 h1:YRXhKfTDauu4ajMg1TPgFO5jnlC2HCbmLXMcTG5cbYE= +github.com/cespare/xxhash/v2 v2.1.2/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs= +github.com/clbanning/mxj/v2 v2.5.5 h1:oT81vUeEiQQ/DcHbzSytRngP6Ky9O+L+0Bw0zSJag9E= +github.com/clbanning/mxj/v2 v2.5.5/go.mod h1:hNiWqW14h+kc+MdF9C6/YoRfjEJoR3ou6tn/Qo+ve2s= +github.com/cloudwego/base64x v0.1.4 h1:jwCgWpFanWmN8xoIUHa2rtzmkd5J2plF/dnLS6Xd/0Y= +github.com/cloudwego/base64x v0.1.4/go.mod h1:0zlkT4Wn5C6NdauXdJRhSKRlJvmclQ1hhJgA0rcu/8w= +github.com/cloudwego/iasm v0.2.0 h1:1KNIy1I1H9hNNFEEH3DVnI4UujN+1zjpuk6gwHLTssg= +github.com/cloudwego/iasm v0.2.0/go.mod h1:8rXZaNYT2n95jn+zTI1sDr+IgcD2GVs0nlbbQPiEFhY= +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM= +github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f h1:lO4WD4F/rVNCu3HqELle0jiPLLBs70cWOduZpkS1E78= +github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f/go.mod h1:cuUVRXasLTGF7a8hSLbxyZXjz+1KgoB3wDUb6vlszIc= +github.com/disintegration/imaging v1.6.2 h1:w1LecBlG2Lnp8B3jk5zSuNqd7b4DXhcjwek1ei82L+c= +github.com/disintegration/imaging v1.6.2/go.mod h1:44/5580QXChDfwIclfc/PCwrr44amcmDAg8hxG0Ewe4= +github.com/facebookarchive/grace v0.0.0-20180706040059-75cf19382434 h1:AFIATPhFj7mrISc4z9zEpfm4a8UfwsCWzJ+Je5jA5Rs= +github.com/facebookarchive/grace v0.0.0-20180706040059-75cf19382434/go.mod h1:PY9iiFMrFjTzegsqfKCcb+Ekk5/j0ch0sMdBwlT6atI= +github.com/facebookgo/clock v0.0.0-20150410010913-600d898af40a h1:yDWHCSQ40h88yih2JAcL6Ls/kVkSE8GFACTGVnMPruw= +github.com/facebookgo/clock v0.0.0-20150410010913-600d898af40a/go.mod h1:7Ga40egUymuWXxAe151lTNnCv97MddSOVsjpPPkityA= +github.com/facebookgo/ensure v0.0.0-20200202191622-63f1cf65ac4c h1:8ISkoahWXwZR41ois5lSJBSVw4D0OV19Ht/JSTzvSv0= +github.com/facebookgo/ensure v0.0.0-20200202191622-63f1cf65ac4c/go.mod h1:Yg+htXGokKKdzcwhuNDwVvN+uBxDGXJ7G/VN1d8fa64= +github.com/facebookgo/freeport v0.0.0-20150612182905-d4adf43b75b9 h1:wWke/RUCl7VRjQhwPlR/v0glZXNYzBHdNUzf/Am2Nmg= +github.com/facebookgo/freeport v0.0.0-20150612182905-d4adf43b75b9/go.mod h1:uPmAp6Sws4L7+Q/OokbWDAK1ibXYhB3PXFP1kol5hPg= +github.com/facebookgo/grace v0.0.0-20180706040059-75cf19382434 h1:mOp33BLbcbJ8fvTAmZacbBiOASfxN+MLcLxymZCIrGE= +github.com/facebookgo/grace v0.0.0-20180706040059-75cf19382434/go.mod h1:KigFdumBXUPSwzLDbeuzyt0elrL7+CP7TKuhrhT4bcU= +github.com/facebookgo/httpdown v0.0.0-20180706035922-5979d39b15c2 h1:nXeeRHmgNgjLxi+7dY9l9aDvSS1uwVlNLqUWIY4Ath0= +github.com/facebookgo/httpdown v0.0.0-20180706035922-5979d39b15c2/go.mod h1:TUV/fX3XrTtBQb5+ttSUJzcFgLNpILONFTKmBuk5RSw= +github.com/facebookgo/stack v0.0.0-20160209184415-751773369052 h1:JWuenKqqX8nojtoVVWjGfOF9635RETekkoH6Cc9SX0A= +github.com/facebookgo/stack v0.0.0-20160209184415-751773369052/go.mod h1:UbMTZqLaRiH3MsBH8va0n7s1pQYcu3uTb8G4tygF4Zg= +github.com/facebookgo/stats v0.0.0-20151006221625-1b76add642e4 h1:0YtRCqIZs2+Tz49QuH6cJVw/IFqzo39gEqZ0iYLxD2M= +github.com/facebookgo/stats v0.0.0-20151006221625-1b76add642e4/go.mod h1:vsJz7uE339KUCpBXx3JAJzSRH7Uk4iGGyJzR529qDIA= +github.com/facebookgo/subset v0.0.0-20200203212716-c811ad88dec4 h1:7HZCaLC5+BZpmbhCOZJ293Lz68O7PYrF2EzeiFMwCLk= +github.com/facebookgo/subset v0.0.0-20200203212716-c811ad88dec4/go.mod h1:5tD+neXqOorC30/tWg0LCSkrqj/AR6gu8yY8/fpw1q0= +github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8= +github.com/fsnotify/fsnotify v1.7.0 h1:8JEhPFa5W2WU7YfeZzPNqzMP6Lwt7L2715Ggo0nosvA= +github.com/fsnotify/fsnotify v1.7.0/go.mod h1:40Bi/Hjc2AVfZrqy+aj+yEI+/bRxZnMJyTJwOpGvigM= +github.com/gabriel-vasile/mimetype v1.4.3 h1:in2uUcidCuFcDKtdcBxlR0rJ1+fsokWf+uqxgUFjbI0= +github.com/gabriel-vasile/mimetype v1.4.3/go.mod h1:d8uq/6HKRL6CGdk+aubisF/M5GcPfT7nKyLpA0lbSSk= +github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE= +github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI= +github.com/gin-gonic/gin v1.10.0 h1:nTuyha1TYqgedzytsKYqna+DfLos46nTv2ygFy86HFU= +github.com/gin-gonic/gin v1.10.0/go.mod h1:4PMNQiOhvDRa013RKVbsiNwoyezlm2rm0uX/T7kzp5Y= +github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s= +github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA= +github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY= +github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY= +github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY= +github.com/go-playground/validator/v10 v10.22.0 h1:k6HsTZ0sTnROkhS//R0O+55JgM8C4Bx7ia+JlgcnOao= +github.com/go-playground/validator/v10 v10.22.0/go.mod h1:dbuPbCMFw/DrkbEynArYaCwl3amGuJotoKCe95atGMM= +github.com/go-redis/redis/v8 v8.11.5 h1:AcZZR7igkdvfVmQTPnu9WE37LRrO/YrBH5zWyjDC0oI= +github.com/go-redis/redis/v8 v8.11.5/go.mod h1:gREzHqY1hg6oD9ngVRbLStwAWKhA0FEgq8Jd4h5lpwo= +github.com/go-sql-driver/mysql v1.7.0 h1:ueSltNNllEqE3qcWBTD0iQd3IpL/6U+mJxLkazJ7YPc= +github.com/go-sql-driver/mysql v1.7.0/go.mod h1:OXbVy3sEdcQ2Doequ6Z5BW6fXNQTmx+9S1MCJN5yJMI= +github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU= +github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I= +github.com/golang-jwt/jwt/v5 v5.2.1 h1:OuVbFODueb089Lh128TAcimifWaLhJwVflnrgM17wHk= +github.com/golang-jwt/jwt/v5 v5.2.1/go.mod h1:pqrtFR0X4osieyHYxtmOUWsAWrfe1Q5UVIyoH402zdk= +github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 h1:DACJavvAHhabrF08vX0COfcOBJRhZ8lUbR+ZWIs0Y5g= +github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0/go.mod h1:E/TSTwGwJL78qG/PmXZO1EjYhfJinVAhrmmHX6Z8B9k= +github.com/google/go-cmp v0.5.9 h1:O2Tfq5qg4qc4AmwVlvv0oLiVAGB7enBSJ2x2DqQFi38= +github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= +github.com/gopherjs/gopherjs v0.0.0-20181017120253-0766667cb4d1/go.mod h1:wJfORRmW1u3UXTncJ5qlYoELFm8eSnnEO6hX4iZ3EWY= +github.com/gopherjs/gopherjs v0.0.0-20200217142428-fce0ec30dd00/go.mod h1:wJfORRmW1u3UXTncJ5qlYoELFm8eSnnEO6hX4iZ3EWY= +github.com/hashicorp/hcl v1.0.0 h1:0Anlzjpi4vEasTeNFn2mLJgTSwt0+6sfsiTG8qcWGx4= +github.com/hashicorp/hcl v1.0.0/go.mod h1:E5yfLk+7swimpb2L/Alb/PJmXilQ/rhwaUYs4T20WEQ= +github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E= +github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc= +github.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ= +github.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8= +github.com/json-iterator/go v1.1.10/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4= +github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM= +github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo= +github.com/jtolds/gls v4.20.0+incompatible/go.mod h1:QJZ7F/aHp+rZTRtaJ1ow/lLfFfVYBRgL+9YlvaHOwJU= +github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg= +github.com/klauspost/cpuid/v2 v2.2.7 h1:ZWSB3igEs+d0qvnxR/ZBzXVmxkgt8DdzP6m9pfuVLDM= +github.com/klauspost/cpuid/v2 v2.2.7/go.mod h1:Lcz8mBdAVJIBVzewtcLocK12l3Y+JytZYpaMropDUws= +github.com/knz/go-libedit v1.10.1/go.mod h1:MZTVkCWyz0oBc7JOWP3wNAzd002ZbM/5hgShxwh4x8M= +github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE= +github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= +github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= +github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY= +github.com/leodido/go-urn v1.4.0 h1:WT9HwE9SGECu3lg4d/dIA+jxlljEa1/ffXKmRjqdmIQ= +github.com/leodido/go-urn v1.4.0/go.mod h1:bvxc+MVxLKB4z00jd1z+Dvzr47oO32F/QSNjSBOlFxI= +github.com/magiconair/properties v1.8.7 h1:IeQXZAiQcpL9mgcAe1Nu6cX9LLw6ExEHKjN0VQdvPDY= +github.com/magiconair/properties v1.8.7/go.mod h1:Dhd985XPs7jluiymwWYZ0G4Z61jb3vdS329zhj2hYo0= +github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY= +github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= +github.com/mitchellh/mapstructure v1.5.0 h1:jeMsZIYE/09sWLaz43PL7Gy6RuMjD2eJVyuac5Z2hdY= +github.com/mitchellh/mapstructure v1.5.0/go.mod h1:bFUtVrKA4DC2yAKiSyO/QUcy7e+RRV2QTWOzhPopBRo= +github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= +github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg= +github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= +github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0= +github.com/modern-go/reflect2 v1.0.1/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0= +github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M= +github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk= +github.com/mohae/deepcopy v0.0.0-20170929034955-c48cc78d4826 h1:RWengNIwukTxcDr9M+97sNutRR1RKhG96O6jWumTTnw= +github.com/mohae/deepcopy v0.0.0-20170929034955-c48cc78d4826/go.mod h1:TaXosZuwdSHYgviHp1DAtfrULt5eUgsSMsZf+YrPgl8= +github.com/mojocn/base64Captcha v1.3.6 h1:gZEKu1nsKpttuIAQgWHO+4Mhhls8cAKyiV2Ew03H+Tw= +github.com/mojocn/base64Captcha v1.3.6/go.mod h1:i5CtHvm+oMbj1UzEPXaA8IH/xHFZ3DGY3Wh3dBpZ28E= +github.com/niemeyer/pretty v0.0.0-20200227124842-a10e7caefd8e h1:fD57ERR4JtEqsWbfPhv4DMiApHyliiK5xCTNVSPiaAs= +github.com/niemeyer/pretty v0.0.0-20200227124842-a10e7caefd8e/go.mod h1:zD1mROLANZcx1PVRCS0qkT7pwLkGfwJo4zjcN/Tysno= +github.com/nxadm/tail v1.4.8 h1:nPr65rt6Y5JFSKQO7qToXr7pePgD6Gwiw05lkbyAQTE= +github.com/onsi/ginkgo v1.16.5 h1:8xi0RTUf59SOSfEtZMvwTvXYMzG4gV23XVHOZiXNtnE= +github.com/onsi/gomega v1.18.1 h1:M1GfJqGRrBrrGGsbxzV5dqM2U2ApXefZCQpkukxYRLE= +github.com/pelletier/go-toml/v2 v2.2.2 h1:aYUidT7k73Pcl9nb2gScu7NSrKCSHIDE89b3+6Wq+LM= +github.com/pelletier/go-toml/v2 v2.2.2/go.mod h1:1t835xjRzz80PqgE6HHgN2JOsmgYu/h4qDAS4n929Rs= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 h1:Jamvg5psRIccs7FGNTlIRMkT8wgtp5eCXdBlqhYGL6U= +github.com/richardlehane/mscfb v1.0.4 h1:WULscsljNPConisD5hR0+OyZjwK46Pfyr6mPu5ZawpM= +github.com/richardlehane/mscfb v1.0.4/go.mod h1:YzVpcZg9czvAuhk9T+a3avCpcFPMUWm7gK3DypaEsUk= +github.com/richardlehane/msoleps v1.0.1/go.mod h1:BWev5JBpU9Ko2WAgmZEuiz4/u3ZYTKbjLycmwiWUfWg= +github.com/richardlehane/msoleps v1.0.3 h1:aznSZzrwYRl3rLKRT3gUk9am7T/mLNSnJINvN0AQoVM= +github.com/richardlehane/msoleps v1.0.3/go.mod h1:BWev5JBpU9Ko2WAgmZEuiz4/u3ZYTKbjLycmwiWUfWg= +github.com/rogpeppe/go-internal v1.9.0 h1:73kH8U+JUqXU8lRuOHeVHaa/SZPifC7BkcraZVejAe8= +github.com/sagikazarmark/locafero v0.4.0 h1:HApY1R9zGo4DBgr7dqsTH/JJxLTTsOt7u6keLGt6kNQ= +github.com/sagikazarmark/locafero v0.4.0/go.mod h1:Pe1W6UlPYUk/+wc/6KFhbORCfqzgYEpgQ3O5fPuL3H4= +github.com/sagikazarmark/slog-shim v0.1.0 h1:diDBnUNK9N/354PgrxMywXnAwEr1QZcOr6gto+ugjYE= +github.com/sagikazarmark/slog-shim v0.1.0/go.mod h1:SrcSrq8aKtyuqEI1uvTDTK1arOWRIczQRv+GVI1AkeQ= +github.com/sirupsen/logrus v1.9.3 h1:dueUQJ1C2q9oE3F7wvmSGAaVtTmUizReu6fjN8uqzbQ= +github.com/sirupsen/logrus v1.9.3/go.mod h1:naHLuLoDiP4jHNo9R0sCBMtWGeIprob74mVsIT4qYEQ= +github.com/smartystreets/assertions v0.0.0-20180927180507-b2de0cb4f26d/go.mod h1:OnSkiWE9lh6wB0YB77sQom3nweQdgAjqCqsofrRNTgc= +github.com/smartystreets/assertions v1.1.0/go.mod h1:tcbTF8ujkAEcZ8TElKY+i30BzYlVhC/LOxJk7iOWnoo= +github.com/smartystreets/goconvey v1.6.4/go.mod h1:syvi0/a8iFYH4r/RixwvyeAJjdLS9QV7WQ/tjFTllLA= +github.com/sourcegraph/conc v0.3.0 h1:OQTbbt6P72L20UqAkXXuLOj79LfEanQ+YQFNpLA9ySo= +github.com/sourcegraph/conc v0.3.0/go.mod h1:Sdozi7LEKbFPqYX2/J+iBAM6HpqSLTASQIKqDmF7Mt0= +github.com/spf13/afero v1.11.0 h1:WJQKhtpdm3v2IzqG8VMqrr6Rf3UYpEF239Jy9wNepM8= +github.com/spf13/afero v1.11.0/go.mod h1:GH9Y3pIexgf1MTIWtNGyogA5MwRIDXGUr+hbWNoBjkY= +github.com/spf13/cast v1.6.0 h1:GEiTHELF+vaR5dhz3VqZfFSzZjYbgeKDpBxQVS4GYJ0= +github.com/spf13/cast v1.6.0/go.mod h1:ancEpBxwJDODSW/UG4rDrAqiKolqNNh2DX3mk86cAdo= +github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA= +github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg= +github.com/spf13/viper v1.19.0 h1:RWq5SEjt8o25SROyN3z2OrDB9l7RPd3lwTWU8EcEdcI= +github.com/spf13/viper v1.19.0/go.mod h1:GQUN9bilAbhU/jgc1bKs99f/suXKeUMct8Adx5+Ntkg= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/objx v0.2.0/go.mod h1:qt09Ya8vawLte6SNmTgCsAVtYtaKzEcn8ATUoHMkEqE= +github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw= +github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo= +github.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA= +github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= +github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA= +github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= +github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= +github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo= +github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg= +github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= +github.com/subosito/gotenv v1.6.0 h1:9NlTDc1FTs4qu0DDq7AEtTPNw6SVm7uBMsUCUjABIf8= +github.com/subosito/gotenv v1.6.0/go.mod h1:Dk4QP5c2W3ibzajGcXpNraDfq2IrhjMIvMSWPKKo0FU= +github.com/tjfoc/gmsm v1.3.2 h1:7JVkAn5bvUJ7HtU08iW6UiD+UTmJTIToHCfeFzkcCxM= +github.com/tjfoc/gmsm v1.3.2/go.mod h1:HaUcFuY0auTiaHB9MHFGCPx5IaLhTUd2atbCFBQXn9w= +github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI= +github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08= +github.com/ugorji/go/codec v1.2.12 h1:9LC83zGrHhuUA9l16C9AHXAqEV/2wBQ4nkvumAE65EE= +github.com/ugorji/go/codec v1.2.12/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg= +github.com/wechatpay-apiv3/wechatpay-go v0.2.20 h1:gS8oFn1bHGnyapR2Zb4aqTV6l4kJWgbtqjCq6k1L9DQ= +github.com/wechatpay-apiv3/wechatpay-go v0.2.20/go.mod h1:A254AUBVB6R+EqQFo3yTgeh7HtyqRRtN2w9hQSOrd4Q= +github.com/xuri/efp v0.0.0-20231025114914-d1ff6096ae53 h1:Chd9DkqERQQuHpXjR/HSV1jLZA6uaoiwwH3vSuF3IW0= +github.com/xuri/efp v0.0.0-20231025114914-d1ff6096ae53/go.mod h1:ybY/Jr0T0GTCnYjKqmdwxyxn2BQf2RcQIIvex5QldPI= +github.com/xuri/excelize/v2 v2.8.1 h1:pZLMEwK8ep+CLIUWpWmvW8IWE/yxqG0I1xcN6cVMGuQ= +github.com/xuri/excelize/v2 v2.8.1/go.mod h1:oli1E4C3Pa5RXg1TBXn4ENCXDV5JUMlBluUhG7c+CEE= +github.com/xuri/nfp v0.0.0-20230919160717-d98342af3f05 h1:qhbILQo1K3mphbwKh1vNm4oGezE1eF9fQWmNiIpSfI4= +github.com/xuri/nfp v0.0.0-20230919160717-d98342af3f05/go.mod h1:WwHg+CVyzlv/TX9xqBFXEZAuxOPxn2k1GNHwG41IIUQ= +github.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74= +github.com/yuin/goldmark v1.1.30/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74= +github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY= +go.uber.org/atomic v1.9.0 h1:ECmE8Bn/WFTYwEW/bpKD3M8VtR/zQVbavAoalC1PYyE= +go.uber.org/atomic v1.9.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc= +go.uber.org/multierr v1.9.0 h1:7fIwc/ZtS0q++VgcfqFDxSBZVv/Xo49/SYnDFupUwlI= +go.uber.org/multierr v1.9.0/go.mod h1:X2jQV1h+kxSjClGpnseKVIxpmcjrj7MNnI0bnlfKTVQ= +golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8= +golang.org/x/arch v0.8.0 h1:3wRIsP3pM4yUptoR96otTUOXI367OS0+c9eeRi9doIc= +golang.org/x/arch v0.8.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys= +golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= +golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI= +golang.org/x/crypto v0.0.0-20191219195013-becbf705a915/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= +golang.org/x/crypto v0.0.0-20200510223506-06a226fb4e37/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= +golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc= +golang.org/x/crypto v0.10.0/go.mod h1:o4eNf7Ede1fv+hwOwZsTHl9EsPFO6q6ZvYR8vYfY45I= +golang.org/x/crypto v0.18.0/go.mod h1:R0j02AL6hcrfOiy9T4ZYp/rcWeMxM3L6QYxlOuEG1mg= +golang.org/x/crypto v0.23.0 h1:dIJU/v2J8Mdglj/8rJ6UUOM3Zc9zLZxVZwwxMooUSAI= +golang.org/x/crypto v0.23.0/go.mod h1:CKFgDieR+mRhux2Lsu27y0fO304Db0wZe70UKqHu0v8= +golang.org/x/exp v0.0.0-20230905200255-921286631fa9 h1:GoHiUyI/Tp2nVkLI2mCxVkOjsbSXD66ic0XW0js0R9g= +golang.org/x/exp v0.0.0-20230905200255-921286631fa9/go.mod h1:S2oDrQGGwySpoQPVqRShND87VCbxmc6bL1Yd2oYrm6k= +golang.org/x/image v0.0.0-20191009234506-e7c1f5e7dbb8/go.mod h1:FeLwcggjj3mMvU+oOTbSwawSJRM1uh48EjtB4UJZlP0= +golang.org/x/image v0.13.0/go.mod h1:6mmbMOeV28HuMTgA6OSRkdXKYw/t5W9Uwn2Yv1r3Yxk= +golang.org/x/image v0.14.0 h1:tNgSxAFe3jC4uYqvZdTr84SZoM1KfwdC9SKIFrLjFn4= +golang.org/x/image v0.14.0/go.mod h1:HUYqC05R2ZcZ3ejNQsIHQDQiwWM4JBqmm6MKANTp4LE= +golang.org/x/mod v0.2.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA= +golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4= +golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs= +golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= +golang.org/x/net v0.0.0-20200226121028-0de0cce0169b/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= +golang.org/x/net v0.0.0-20200506145744-7e3656a0809f/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A= +golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg= +golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c= +golang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs= +golang.org/x/net v0.7.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs= +golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg= +golang.org/x/net v0.11.0/go.mod h1:2L/ixqYpgIVXmeoSA/4Lu7BzTG4KIyPIryS4IsOd1oQ= +golang.org/x/net v0.20.0/go.mod h1:z8BVo6PvndSri0LbOE3hAn0apkU+1YvI6E70E9jsnvY= +golang.org/x/net v0.25.0 h1:d/OCCoBEUq33pjydKrGQhw7IlUPI2Oylr+8qLx49kac= +golang.org/x/net v0.25.0/go.mod h1:JkAGAh7GEvH74S6FOH42FLoXpXbE/aqXSrIQjXgsiwM= +golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20200317015054-43a5402ce75a/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200323222414-85ca7c5b95cd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200509044756-6aff5f38e54f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.9.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.16.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= +golang.org/x/sys v0.20.0 h1:Od9JTbYCk261bKm4M/mw7AklTlFYIa0bIp9BgSm1S8Y= +golang.org/x/sys v0.20.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= +golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= +golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8= +golang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k= +golang.org/x/term v0.8.0/go.mod h1:xPskH00ivmX89bAKVGSKKtLOWNx2+17Eiy94tnKShWo= +golang.org/x/term v0.9.0/go.mod h1:M6DEAAIenWoTxdKrOltXcmDY3rSplQUkrvaDU5FcQyo= +golang.org/x/term v0.16.0/go.mod h1:yn7UURbUtPyrVJPGPq404EukNFxcm/foM+bV/bfcDsY= +golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= +golang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk= +golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= +golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ= +golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8= +golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8= +golang.org/x/text v0.10.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE= +golang.org/x/text v0.13.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE= +golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU= +golang.org/x/text v0.15.0 h1:h1V/4gjBv8v9cjcR6+AR5+/cIYK5N/WAgiv4xlsEtAk= +golang.org/x/text v0.15.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU= +golang.org/x/time v0.5.0 h1:o7cqy6amK/52YcAKIPlM3a+Fpj35zvRj2TP+e1xFSfk= +golang.org/x/time v0.5.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/tools v0.0.0-20190328211700-ab21143f2384/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= +golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= +golang.org/x/tools v0.0.0-20200509030707-2212a7e161a5/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE= +golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc= +golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU= +golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +google.golang.org/protobuf v1.34.1 h1:9ddQBjfCyZPOHPUiPxpYESBLc+T8P3E+Vo4IbKZgFWg= +google.golang.org/protobuf v1.34.1/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/check.v1 v1.0.0-20200227125254-8fa46927fb4f h1:BLraFXnmrev5lT+xlilqcH8XK9/i0At2xKjWk4p6zsU= +gopkg.in/check.v1 v1.0.0-20200227125254-8fa46927fb4f/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/ini.v1 v1.56.0/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k= +gopkg.in/ini.v1 v1.67.0 h1:Dgnx+6+nfE+IfzjUEISNeydPJh9AXNNsWbGP9KzCsOA= +gopkg.in/ini.v1 v1.67.0/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k= +gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 h1:uRGJdciOHaEIrze2W8Q3AKkepLTh2hOroT7a+7czfdQ= +gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= +gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gorm.io/driver/mysql v1.5.7 h1:MndhOPYOfEp2rHKgkZIhJ16eVUIRf2HmzgoPmh7FCWo= +gorm.io/driver/mysql v1.5.7/go.mod h1:sEtPWMiqiN1N1cMXoXmBbd8C6/l+TESwriotuRRpkDM= +gorm.io/gorm v1.25.7/go.mod h1:hbnx/Oo0ChWMn1BIhpy1oYozzpM15i4YPuHDmfYtwg8= +gorm.io/gorm v1.25.11 h1:/Wfyg1B/je1hnDx3sMkX+gAlxrlZpn6X0BXRlwXlvHg= +gorm.io/gorm v1.25.11/go.mod h1:xh7N7RHfYlNc5EmcI/El95gXusucDrQnHXe0+CgWcLQ= +nullprogram.com/x/optparse v1.0.0/go.mod h1:KdyPE+Igbe0jQUrVfMqDMeJQIJZEuyV7pjYmp6pbG50= +rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4= diff --git a/main.go b/main.go new file mode 100644 index 0000000..600509c --- /dev/null +++ b/main.go @@ -0,0 +1,47 @@ +package main + +import "C" +import ( + "fmt" + "github.com/facebookarchive/grace/gracehttp" + "net/http" + "strconv" + "vote-admin-video-api/api/router" + "vote-admin-video-api/config" + "vote-admin-video-api/core" +) + +func main() { + // 加载配置文件 + core.Viper() + + // 加载日志 + core.Logrus() + + // 加载数据库 + core.Mysql() + + // 加载redis缓存 + core.Redis() + + // 加载验证器 + core.Validator() + + // 加载雪花算法 + core.Snowflake() + + // 初始化路由-加载中间件 + r := router.Init() + + // 启动 HTTP 服务器 + server := &http.Server{ + Addr: ":" + strconv.Itoa(config.C.Port), // 设置服务器监听的端口号 + Handler: r, + } + + // 使用 grace 运行服务器 + err := gracehttp.Serve(server) + if err != nil { + fmt.Printf("启动失败:%v\n\n", err) + } +} diff --git a/resource/3061726102564.png b/resource/3061726102564.png new file mode 100644 index 0000000..86fbbe7 Binary files /dev/null and b/resource/3061726102564.png differ diff --git a/utils/aes.go b/utils/aes.go new file mode 100644 index 0000000..4f6ba0c --- /dev/null +++ b/utils/aes.go @@ -0,0 +1,101 @@ +package utils + +import ( + "crypto/aes" + "crypto/cipher" + "crypto/hmac" + "crypto/rand" + "crypto/sha256" + "encoding/base64" + "encoding/hex" + "io" +) + +// Encrypt 加密函数 +func Encrypt(text, key string) (string, error) { + plaintext := []byte(text) + hashedKey := hashKey(key) + + // 创建一个新的 AES cipher.Block + block, err := aes.NewCipher(hashedKey) + if err != nil { + return "", err + } + + // 使用 GCM 模式加密 + aesGCM, err := cipher.NewGCM(block) + if err != nil { + return "", err + } + + nonce := make([]byte, aesGCM.NonceSize()) + if _, err = io.ReadFull(rand.Reader, nonce); err != nil { + return "", err + } + + // 加密数据 + ciphertext := aesGCM.Seal(nonce, nonce, plaintext, nil) + return base64.StdEncoding.EncodeToString(ciphertext), nil +} + +// Decrypt 解密函数 +func Decrypt(cryptoText, key string) (string, error) { + ciphertext, _ := base64.StdEncoding.DecodeString(cryptoText) + hashedKey := hashKey(key) + + // 创建一个新的 AES cipher.Block + block, err := aes.NewCipher(hashedKey) + if err != nil { + return "", err + } + + // 使用 GCM 模式解密 + aesGCM, err := cipher.NewGCM(block) + if err != nil { + return "", err + } + + nonceSize := aesGCM.NonceSize() + nonce, ciphertext := ciphertext[:nonceSize], ciphertext[nonceSize:] + + // 解密数据 + plaintext, err := aesGCM.Open(nil, nonce, ciphertext, nil) + if err != nil { + return "", err + } + + return string(plaintext), nil +} + +// HashString 使用SHA256算法对字符串进行哈希加密 +func HashString(s string) string { + // 创建一个新的SHA256哈希实例 + hasher := sha256.New() + + // 写入要哈希的字符串的字节 + hasher.Write([]byte(s)) + + // 获取哈希结果的字节切片 + hashBytes := hasher.Sum(nil) + + // 将字节切片转换成十六进制字符串表示 + s = hex.EncodeToString(hashBytes) + + return s +} + +// HmacSHA256 计算 HMAC-SHA256 签名 +func HmacSHA256(data string, key string) string { + // 创建新的 HMAC 使用 SHA256 哈希函数 + h := hmac.New(sha256.New, []byte(key)) + // 写入数据 + h.Write([]byte(data)) + // 计算 HMAC-SHA256 并返回其字节切片 + return hex.EncodeToString(h.Sum(nil)) +} + +// 将任意长度的密钥转换为适合 AES 的长度（32 字节） +func hashKey(key string) []byte { + hash := sha256.Sum256([]byte(key)) + return hash[:] +} diff --git a/utils/captcha.go b/utils/captcha.go new file mode 100644 index 0000000..6f502f4 --- /dev/null +++ b/utils/captcha.go @@ -0,0 +1,51 @@ +package utils + +import ( + "github.com/mojocn/base64Captcha" + "image/color" + "time" +) + +// GenerateCaptcha 生成验证码-base64 +func GenerateCaptcha() (id, b64s string, err error) { + var driver *base64Captcha.DriverString + // 配置验证码的参数 + driverString := &base64Captcha.DriverString{ + Height: 40, + Width: 100, + NoiseCount: 0, + ShowLineOptions: 0, + Length: 4, + Source: "1234567890", + BgColor: &color.RGBA{R: 3, G: 102, B: 214, A: 125}, + Fonts: []string{"wqy-microhei.ttc"}, + } + // ConvertFonts 按名称加载字体 + driver = driverString.ConvertFonts() + + base64Captcha.Expiration = 30 * time.Minute + store := base64Captcha.DefaultMemStore + + captcha := base64Captcha.NewCaptcha(driver, store) + id, b64s, _, err = captcha.Generate() + if err != nil { + return "", "", err + } + return id, b64s, nil +} + +// VerifyCaptcha 验证验证码 +func VerifyCaptcha(id, answer string) bool { + // 创建验证码实例 + base64Captcha.Expiration = 30 * time.Minute + store := base64Captcha.DefaultMemStore + captcha := base64Captcha.NewCaptcha(nil, store) + + // 验证验证码 + isValid := captcha.Verify(id, answer, true) + if !isValid { + return false + } + + return true +} diff --git a/utils/compute.go b/utils/compute.go new file mode 100644 index 0000000..4d4dc3d --- /dev/null +++ b/utils/compute.go @@ -0,0 +1,64 @@ +package utils + +import ( + "time" +) + +// 一些计算 + +// ComputeIndividualIncomeTax 计算个人所得税 +func ComputeIndividualIncomeTax(income float64) float64 { + if income <= 800 { + return 0 + } + + if income <= 4000 { + income = income - 800 + } + + // 实际纳税金额 + if income > 4000 { + income = income * 0.8 + } + + // 税率、速算扣除数 + var taxRate, quickDeduction float64 + + if income <= 20000 { + taxRate = 0.2 + quickDeduction = 0 + } else if income <= 50000 { + taxRate = 0.3 + quickDeduction = 2000 + } else { + taxRate = 0.4 + quickDeduction = 7000 + } + + incomeTax := income*taxRate - quickDeduction + + return incomeTax +} + +// CalculateAge 计算年龄 +func CalculateAge(birthdate string) (int, error) { + // 解析出生日期字符串 + layout := "2006-01-02" + birthTime, err := time.Parse(layout, birthdate) + if err != nil { + return 0, err + } + + // 获取当前时间 + now := time.Now() + + // 计算年龄 + years := now.Year() - birthTime.Year() + + // 如果今年的生日还没到，年龄减一 + if now.Month() < birthTime.Month() || (now.Month() == birthTime.Month() && now.Day() < birthTime.Day()) { + years-- + } + + return years, nil +} diff --git a/utils/directory.go b/utils/directory.go new file mode 100644 index 0000000..dcb0a5a --- /dev/null +++ b/utils/directory.go @@ -0,0 +1,21 @@ +package utils + +import ( + "errors" + "os" +) + +// PathExists 文件是否存在 +func PathExists(path string) (bool, error) { + fi, err := os.Stat(path) + if err == nil { + if fi.IsDir() { + return true, nil + } + return false, errors.New("存在同名文件") + } + if os.IsNotExist(err) { + return false, nil + } + return false, err +} diff --git a/utils/export.go b/utils/export.go new file mode 100644 index 0000000..a726458 --- /dev/null +++ b/utils/export.go @@ -0,0 +1,298 @@ +package utils + +import ( + "bytes" + "fmt" + "github.com/xuri/excelize/v2" + "reflect" + "strconv" +) + +// func Export(widths []int) (bool, error) { +// f := excelize.NewFile() +// defer func() { +// _ = f.Close() +// }() +// +// // 创建一个工作表 +// index, err := f.NewSheet("Sheet1") +// if err != nil { +// return false, err +// } +// +// // 设置工作簿的默认工作表 +// f.SetActiveSheet(index) +// +// // 单元格对齐样式 +// alignment := &excelize.Alignment{ +// Horizontal: "center", +// Vertical: "center", +// } +// +// // 单元格颜色填充样式 +// fill := excelize.Fill{ +// Type: "pattern", +// Pattern: 1, +// Color: []string{"#c9daf8"}, +// Shading: 0, +// } +// +// // 第一行的左、右、下边框 +// border := []excelize.Border{ +// { +// Type: "left,right,bottom", +// Color: "", +// Style: 1, +// }, +// } +// +// // 工作表样式 +// style, err := f.NewStyle( +// &excelize.Style{ +// Fill: fill, +// Alignment: alignment, +// Border: border, +// }, +// ) +// if err != nil { +// return false, err +// } +// +// // 依次设置每一列的列宽 +// widths = []int{18, 18, 18, 18, 18, 20, 23, 46, 18, 30, 30, 18, 18, 30, 18, 30} +// for col, width := range widths { +// // 获取列名 +// colName, err := excelize.ColumnNumberToName(col + 1) +// if err != nil { +// return false, err +// } +// +// // 设置列宽 +// err = f.SetColWidth("Sheet1", colName, colName, float64(width)) +// if err != nil { +// return false, err +// } +// +// // 设置列背景颜色 +// err = f.SetCellStyle("Sheet1", colName+"1", colName+"1", style) +// if err != nil { +// return false, err +// } +// } +// +// // 设置行高 +// err = f.SetRowStyle("Sheet1", 1, 10, style) +// if err != nil { +// return false, err +// } +// +// if err := f.SaveAs("output.xlsx"); err != nil { +// return false, err +// } +// +// return true, nil +// } + +// HeaderCellData 表头内容 +type HeaderCellData struct { + Value string // 值 + CellType string // 类型 + NumberFmt string // 格式化方式 + ColWidth int // 列宽 +} + +func Export(header []HeaderCellData, data []interface{}) (*bytes.Buffer, error) { + sheetName := "Sheet1" + + f := excelize.NewFile() + defer func() { + _ = f.Close() + }() + + // 创建一个工作表 + index, err := f.NewSheet(sheetName) + if err != nil { + return nil, err + } + + // 设置工作簿的默认工作表 + f.SetActiveSheet(index) + + // 设置工作表默认字体 + err = f.SetDefaultFont("宋体") + if err != nil { + return nil, err + } + + // 统一单元格对齐样式 + alignment := &excelize.Alignment{ + Horizontal: "center", + Vertical: "center", + } + + // 设置行高 35-第一行 + err = f.SetRowHeight(sheetName, 1, 35) + if err != nil { + return nil, err + } + + // 处理工作表表头 + for c, cell := range header { + // 获取列名 + colName, err := excelize.ColumnNumberToName(c + 1) + if err != nil { + return nil, err + } + + // 添加单元格的值 + err = f.SetCellValue(sheetName, colName+"1", cell.Value) + if err != nil { + return nil, err + } + + // 单元格颜色填充样式 + fill := excelize.Fill{ + Type: "pattern", + Pattern: 1, + Color: []string{"#c9daf8"}, + Shading: 0, + } + + // 第一行的左、右、下边框 + border := []excelize.Border{ + { + Type: "left", + Color: "#000000", + Style: 1, + }, + { + Type: "right", + Color: "#000000", + Style: 1, + }, + { + Type: "bottom", + Color: "#000000", + Style: 1, + }, + } + + // 设置单元格值类型和格式 + style, _ := f.NewStyle(&excelize.Style{ + Alignment: alignment, // 字体居中 + Fill: fill, // 背景颜色 + Border: border, // 边框 + }) + err = f.SetCellStyle(sheetName, colName+"1", colName+"1", style) + if err != nil { + return nil, err + } + + // 设置列宽 + err = f.SetColWidth(sheetName, colName, colName, float64(cell.ColWidth)) + if err != nil { + return nil, err + } + } + + // 设置单元格格式 + row := len(data) + for i, cell := range header { + // 获取列名 + colName, err := excelize.ColumnNumberToName(i + 1) + if err != nil { + return nil, err + } + + // 字体居中 + style := &excelize.Style{} + style = &excelize.Style{ + Alignment: alignment, + } + + if cell.CellType == "float" { + style.NumFmt = 2 + customNumFmt := "0.000" + style.CustomNumFmt = &customNumFmt + } + + if cell.CellType == "date" { + style.NumFmt = 22 + customNumFmt := "yyyy-mm-dd hh:mm:ss" + style.CustomNumFmt = &customNumFmt + } + + newStyle, _ := f.NewStyle(style) + err = f.SetCellStyle(sheetName, colName+"2", colName+strconv.Itoa(row), newStyle) + if err != nil { + return nil, err + } + } + + // 填充数据 + for r, rowData := range data { + rv := reflect.ValueOf(rowData) + for c := 0; c < rv.NumField(); c++ { + cellValue := rv.Field(c).Interface() + // 获取列名 + colName, err := excelize.ColumnNumberToName(c + 1) + if err != nil { + return nil, err + } + + axis := colName + fmt.Sprintf("%d", r+2) + + // 设置单元格值 + err = f.SetCellValue(sheetName, axis, cellValue) + if err != nil { + return nil, err + } + + // 设置单元格值类型 + cellType := header[c].CellType + + // 字体居中 + style := &excelize.Style{} + style = &excelize.Style{ + Alignment: alignment, + } + + if cellType == "float" { + style.NumFmt = 2 + customNumFmt := "0.000" + style.CustomNumFmt = &customNumFmt + } + + if cellType == "date" { + style.NumFmt = 22 + customNumFmt := "yyyy-mm-dd hh:mm:ss" + style.CustomNumFmt = &customNumFmt + } + + newStyle, _ := f.NewStyle(style) + err = f.SetCellStyle(sheetName, axis, axis, newStyle) + if err != nil { + return nil, err + } + + // 设置行高 35-第一行 + err = f.SetRowHeight(sheetName, r+2, 35) + if err != nil { + return nil, err + } + } + } + + buffer, err := f.WriteToBuffer() + if err != nil { + return nil, err + } + + return buffer, nil + + // 保存文件 + // if err := f.SaveAs("output.xlsx"); err != nil { + // return nil, err + // } + // return nil, errors.New("已导出文件") +} diff --git a/utils/idCard.go b/utils/idCard.go new file mode 100644 index 0000000..72cdafd --- /dev/null +++ b/utils/idCard.go @@ -0,0 +1,124 @@ +// Package utils 身份证处理 +package utils + +import ( + "errors" + "fmt" + "regexp" + "strconv" + "strings" + "time" + "unicode/utf8" +) + +// GetCardAge 获取身份证年龄 +func GetCardAge(cardNum string) (int, error) { + // 获取当前时间 + now := time.Now() + + // 解析身份证号中的出生日期 + birthDateStr := cardNum[6:14] + birthYear, err := strconv.Atoi(birthDateStr[0:4]) + if err != nil { + return 0, err + } + birthMonth, err := strconv.Atoi(birthDateStr[4:6]) + if err != nil { + return 0, err + } + + // 计算年龄 + age := now.Year() - birthYear + if now.Month() < time.Month(birthMonth) { + age-- + } + + return age, nil +} + +// CheckCardNum 检测身份证号 +func CheckCardNum(cardNum string) (bool, error) { + fmt.Println(cardNum) + regex := `^(?:1[1-5]|2[1-3]|3[1-7]|4[1-6]|5[0-4]|6[1-5])\d{4}(?:1[89]|20)\d{2}(?:0[1-9]|1[0-2])(?:0[1-9]|[12]\d|3[01])\d{3}[\dxX]$` + match, err := regexp.MatchString(regex, cardNum) + fmt.Println(match) + if !match || err != nil { + return false, errors.New("身份证号错误") + } + return true, nil +} + +// GetCardSex 获取身份证性别 +func GetCardSex(cardNum string) (int, error) { + genderStr := cardNum[len(cardNum)-2 : len(cardNum)-1] + genderNum, err := strconv.Atoi(genderStr) + if err != nil { + return 0, err + } + + // 判断性别 + if genderNum%2 == 0 { + return 2, nil + } else { + return 1, nil + } +} + +// GetMaskCardNum 身份证号码脱敏 +func GetMaskCardNum(cardNum string) string { + if len(cardNum) != 18 { + return cardNum + } + + // 获取身份证号前后部分 + frontPart := cardNum[0:6] + backPart := cardNum[14:] + + // 替换中间部分数字为 * + middlePart := "****" + + // 拼接新的身份证号 + maskedIDCard := frontPart + middlePart + backPart + + return maskedIDCard +} + +// GetMaskCardName 身份证名字脱敏 +func GetMaskCardName(cardName string) string { + // 判断姓名长度 + length := utf8.RuneCountInString(cardName) + + // 判断是否是英文姓名 + isEnglish := strings.ContainsAny(cardName, "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ") + + if length == 2 { + // 两个字符的姓名 + if isEnglish { + // 英文姓名 + return string(cardName[0]) + "*" + } else { + // 中文姓名 + return string([]rune(cardName)[0]) + "*" + } + } else if length == 3 { + // 三个字符的姓名 + if isEnglish { + // 英文姓名 + return string(cardName[0]) + "*" + string(cardName[2]) + } else { + // 中文姓名 + return string([]rune(cardName)[0]) + "*" + string([]rune(cardName)[2]) + } + } else if length >= 4 { + // 四个及以上字符的姓名 + if isEnglish { + // 英文姓名 + return string(cardName[0]) + "*" + string(cardName[length-1]) + } else { + // 中文姓名 + return string([]rune(cardName)[0]) + "*" + string([]rune(cardName)[length-1]) + } + } + + return cardName +} diff --git a/utils/image.go b/utils/image.go new file mode 100644 index 0000000..a033161 --- /dev/null +++ b/utils/image.go @@ -0,0 +1,43 @@ +package utils + +import ( + "bytes" + "fmt" + "github.com/disintegration/imaging" + "image" +) + +// AddWatermarkToImage 添加图片水印 +func AddWatermarkToImage(fileBytes []byte, watermarkPath string) ([]byte, error) { + // 将字节切片转为io.Reader + reader := bytes.NewReader(fileBytes) + + // 从字节流中解码图片 + srcImage, err := imaging.Decode(reader) + if err != nil { + return nil, fmt.Errorf("图片解码失败: %w", err) + } + + // 打开水印图片 + watermarkImage, err := imaging.Open(watermarkPath) + if err != nil { + return nil, fmt.Errorf("水印图片加载失败: %w", err) + } + + // 获取原图片的宽高，用于水印的位置计算 + srcBounds := srcImage.Bounds() + watermarkBounds := watermarkImage.Bounds() + + // 将水印添加到原图片的右下角 + offset := image.Pt(srcBounds.Max.X-watermarkBounds.Max.X, srcBounds.Max.Y-watermarkBounds.Max.Y) + dstImage := imaging.Overlay(srcImage, watermarkImage, offset, 1.0) + + // 将处理后的图片保存到字节切片中 + var outputBuffer bytes.Buffer + err = imaging.Encode(&outputBuffer, dstImage, imaging.PNG) + if err != nil { + return nil, fmt.Errorf("图片编码失败: %w", err) + } + + return outputBuffer.Bytes(), nil +} diff --git a/utils/intToString.go b/utils/intToString.go new file mode 100644 index 0000000..e730d41 --- /dev/null +++ b/utils/intToString.go @@ -0,0 +1,77 @@ +package utils + +// OrderCancelReasonToString 订单取消原因（1:主动取消 2:后台取消 3:支付超时取消） +func OrderCancelReasonToString(i int) string { + switch i { + case 1: + return "主动取消" + case 2: + return "后台取消" + case 3: + return "支付超时取消" + default: + return "取消" + } +} + +// OrderSingleStatusToString 订单状态（1:待支付 2:已完成 3:已取消） +func OrderSingleStatusToString(i int) string { + switch i { + case 1: + return "待支付" + case 2: + return "已完成" + case 3: + return "已取消" + default: + return "未知" + } +} + +// OrderSingleRefundStatusToString 订单退款状态（0:无退款 1:申请退款 2:退款中 3:退款成功 4:拒绝退款 5:退款关闭 6:退款异常 7:部分退款） +func OrderSingleRefundStatusToString(i int) string { + switch i { + case 0: + return "无退款" + case 1: + return "申请退款" + case 2: + return "退款中" + case 3: + return "退款成功" + case 4: + return "拒绝退款" + case 5: + return "退款关闭" + case 6: + return "退款异常" + case 7: + return "部分退款" + default: + return "未知" + } +} + +// OrderSinglePayStatusToString 订单支付状态（1:未支付 2:已支付 3:支付中 4:支付失败 5:支付超时 6:支付关闭 7:已撤销 8:转入退款） +func OrderSinglePayStatusToString(i int) string { + switch i { + case 1: + return "未支付" + case 2: + return "已支付" + case 3: + return "支付中" + case 4: + return "支付失败" + case 5: + return "支付超时" + case 6: + return "支付关闭" + case 7: + return "已撤销" + case 8: + return "转入退款" + default: + return "未知" + } +} diff --git a/utils/jwt.go b/utils/jwt.go new file mode 100644 index 0000000..84353b3 --- /dev/null +++ b/utils/jwt.go @@ -0,0 +1,40 @@ +package utils + +import ( + "github.com/golang-jwt/jwt/v5" + "time" + "vote-admin-video-api/config" +) + +type Token struct { + UserId string `json:"user_id"` // 用户id + jwt.RegisteredClaims // v5版本新加的方法 +} + +// NewJWT GenerateJWT 生成JWT +func (t Token) NewJWT() (string, error) { + ttl := time.Duration(config.C.Jwt.Ttl) + + t.RegisteredClaims.ExpiresAt = jwt.NewNumericDate(time.Now().Add(ttl * time.Hour)) // 过期时间24小时 + t.RegisteredClaims.IssuedAt = jwt.NewNumericDate(time.Now()) // 签发时间 + t.RegisteredClaims.NotBefore = jwt.NewNumericDate(time.Now()) // 生效时间 + + // 使用HS256签名算法 + token := jwt.NewWithClaims(jwt.SigningMethodHS256, t) + s, err := token.SignedString([]byte(config.C.Jwt.SignKey)) + + return s, err +} + +// ParseJwt 解析JWT +func ParseJwt(authorization string) (*Token, error) { + t, err := jwt.ParseWithClaims(authorization, &Token{}, func(token *jwt.Token) (interface{}, error) { + return []byte(config.C.Jwt.SignKey), nil + }) + + if claims, ok := t.Claims.(*Token); ok && t.Valid { + return claims, nil + } else { + return nil, err + } +} diff --git a/utils/logger.go b/utils/logger.go new file mode 100644 index 0000000..65834d2 --- /dev/null +++ b/utils/logger.go @@ -0,0 +1,28 @@ +package utils + +import ( + "encoding/json" + "fmt" + "github.com/sirupsen/logrus" + "vote-admin-video-api/global" +) + +func LogJsonInfo(msg string, v interface{}) { + jsonData, err := json.Marshal(v) + if err != nil { + fmt.Println("Error marshaling struct to JSON:", err) + return + } + + jsonString := string(jsonData) + + global.Logger.WithFields(logrus.Fields{ + "data": jsonString, + }).Info(msg) +} + +func LogJsonErr(msg string, v interface{}) { + global.Logger.WithFields(logrus.Fields{ + "data": v, + }).Errorf(msg) +} diff --git a/utils/mask.go b/utils/mask.go new file mode 100644 index 0000000..33aacfc --- /dev/null +++ b/utils/mask.go @@ -0,0 +1,90 @@ +package utils + +import ( + "regexp" + "unicode/utf8" +) + +// MaskNameStr 用户名掩码 +// maskType 加密类型(1:三个字符或三个字符以上掐头取尾，中间用*代替 首字母保留，后两位用*代替) +func MaskNameStr(str string, maskType int) string { + if str == "" { + return str + } + + // 使用正则表达式判断是否包含中文字符 + chinesePattern := "[\u4e00-\u9fa5]+" + isChinese, _ := regexp.MatchString(chinesePattern, str) + + // 使用正则表达式判断是否包含英文字母 + englishPattern := "[A-Za-z]+" + isEnglish, _ := regexp.MatchString(englishPattern, str) + + // 判断是否包含中文字符 + if isChinese { + // 按照中文字符计算长度 + strLen := utf8.RuneCountInString(str) + + if strLen >= 3 { + if maskType == 1 { + // 三个字符或三个字符以上掐头取尾，中间用*代替 + firstChar, _ := utf8.DecodeRuneInString(str) + lastChar, _ := utf8.DecodeLastRuneInString(str) + str = string(firstChar) + "*" + string(lastChar) + } else { + // 首字母保留，后两位用*代替 + firstChar, _ := utf8.DecodeRuneInString(str) + str = string(firstChar) + "**" + } + } else if strLen == 2 { + // 两个字符 + firstChar, _ := utf8.DecodeRuneInString(str) + str = string(firstChar) + "*" + } + } else if isEnglish { + // 按照英文字串计算长度 + strLen := utf8.RuneCountInString(str) + + if strLen >= 3 { + if maskType == 1 { + // 三个字符或三个字符以上掐头取尾，中间用*代替 + firstChar, _ := utf8.DecodeRuneInString(str) + lastChar, _ := utf8.DecodeLastRuneInString(str) + str = string(firstChar) + "*" + string(lastChar) + } else { + // 首字母保留，后两位用*代替 + firstChar, _ := utf8.DecodeRuneInString(str) + str = string(firstChar) + "**" + } + } else if strLen == 2 { + // 两个字符 + firstChar, _ := utf8.DecodeRuneInString(str) + str = string(firstChar) + "*" + } + } + + return str +} + +// MaskPhoneStr 手机号、固话加密 +// 固话：0510-89754815 0510-8****815 +// 手机号：18221234158 18*******58 +func MaskPhoneStr(phone string) string { + if phone == "" { + return phone + } + + // 使用正则表达式匹配固定电话 + phonePattern := `(0[0-9]{2,3}[\-]?[2-9][0-9]{6,7}[\-]?[0-9]?)` + isFixedLine := regexp.MustCompile(phonePattern).MatchString(phone) + + if isFixedLine { + // 匹配到固定电话 + // 替换匹配的部分为固定格式 + return regexp.MustCompile(`(0[0-9]{2,3}[\-]?[2-9])[0-9]{3,4}([0-9]{3}[\-]?[0-9]?)`).ReplaceAllString(phone, "$1****$2") + } else { + // 匹配到手机号 + // 替换匹配的部分为固定格式 + return regexp.MustCompile(`(1[0-9]{1})[0-9]{7}([0-9]{2})`).ReplaceAllString(phone, "$1*******$2") + } +} diff --git a/utils/regular.go b/utils/regular.go new file mode 100644 index 0000000..467f4a5 --- /dev/null +++ b/utils/regular.go @@ -0,0 +1,12 @@ +package utils + +import "regexp" + +// RegexpMobile 手机号匹配 +func RegexpMobile(mobile string) bool { + ok, err := regexp.MatchString(`^1(3\d|4[5-9]|5[0-35-9]|6[2567]|7[0-8]|8\d|9[0-35-9])\d{8}$`, mobile) + if err != nil { + return false + } + return ok +} diff --git a/utils/replace.go b/utils/replace.go new file mode 100644 index 0000000..6510c4f --- /dev/null +++ b/utils/replace.go @@ -0,0 +1,32 @@ +package utils + +import ( + "strings" + "vote-admin-video-api/config" +) + +// RemoveOssDomain 去除oss地址中的前缀 +func RemoveOssDomain(url string) string { + if url != "" { + url = strings.Replace(url, config.C.Oss.OssCustomDomainName, "", 1) + } + return url +} + +// AddOssDomain 增加oss地址中的前缀 +func AddOssDomain(url string) string { + if url == "" { + return "" + } + return config.C.Oss.OssCustomDomainName + url +} + +// GetExtension 解析并返回字符串中 `.` 后面的部分 +func GetExtension(s string) string { + // 使用 strings.Split 分割字符串 + parts := strings.Split(s, ".") + if len(parts) > 1 { + return parts[len(parts)-1] + } + return "" +} diff --git a/utils/type.go b/utils/type.go new file mode 100644 index 0000000..eeb37fb --- /dev/null +++ b/utils/type.go @@ -0,0 +1,28 @@ +package utils + +import ( + "strconv" +) + +// StrToFloat32 字符串转float32 +func StrToFloat32(s string) (f *float32, err error) { + f64, err := strconv.ParseFloat(s, 32) + if err != nil { + return nil, err + } + + // 将float64转换为float32 + f32 := float32(f64) + + return &f32, nil +} + +// StrToFloat64 字符串转float64 +func StrToFloat64(s string) (f *float64, err error) { + f64, err := strconv.ParseFloat(s, 64) + if err != nil { + return nil, err + } + + return &f64, nil +} diff --git a/utils/tz.go b/utils/tz.go new file mode 100644 index 0000000..6c6d245 --- /dev/null +++ b/utils/tz.go @@ -0,0 +1,50 @@ +package utils + +import ( + "fmt" + "time" + "vote-admin-video-api/api/model" +) + +// StrToLocalTime 字符串时间转localtime +func StrToLocalTime(s string) (*model.LocalTime, error) { + // 获取本地时区 + location, err := time.LoadLocation("Local") + if err != nil { + return nil, err + } + + timeParse, err := time.ParseInLocation("2006-01-02 15:04:05", s, location) + if err != nil { + return nil, err + } + + t := model.LocalTime(timeParse) + + return &t, nil +} + +// TimeToAgoString 根据给定的时间返回一个描述性的字符串 +func TimeToAgoString(t time.Time) string { + now := time.Now() + diffTime := now.Sub(t) + + switch { + case diffTime < time.Minute: + return "1分钟前" + case diffTime < time.Hour: + minutes := int(diffTime.Minutes()) + return fmt.Sprintf("%d分钟前", minutes) + case diffTime < 24*time.Hour: + return "一小时前" + case diffTime < 48*time.Hour: + return "两天前" + case diffTime < 72*time.Hour: + return "三天前" + case diffTime < 10*time.Hour*24: + days := int(diffTime.Hours()) / 24 + return fmt.Sprintf("%d天前", days) + default: + return "十天前" + } +} diff --git a/utils/validator.go b/utils/validator.go new file mode 100644 index 0000000..d8acbe6 --- /dev/null +++ b/utils/validator.go @@ -0,0 +1,17 @@ +package utils + +import ( + "errors" + "github.com/go-playground/validator/v10" + "vote-admin-video-api/global" +) + +// Translate 检验并返回检验错误信息 +func Translate(err error) (errMsg string) { + var errs validator.ValidationErrors + errors.As(err, &errs) + for _, err := range errs { + errMsg = err.Translate(global.Trans) + } + return +}