:art:后台接口终于磨磨蹭蹭的写完了
This commit is contained in:
84
service/navigation.go
Normal file
84
service/navigation.go
Normal file
@@ -0,0 +1,84 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"github.com/badoux/goscraper"
|
||||
"github.com/spf13/cast"
|
||||
"gorm.io/gorm"
|
||||
"website-nav/global/client"
|
||||
"website-nav/http/param"
|
||||
"website-nav/http/vo"
|
||||
"website-nav/model"
|
||||
)
|
||||
|
||||
type navigation struct {
|
||||
*gorm.DB
|
||||
}
|
||||
|
||||
func Navigation() navigation { return navigation{DB: client.DB} }
|
||||
|
||||
// List
|
||||
// @description: 导航分页列表
|
||||
// @receiver navigation
|
||||
// @param p
|
||||
// @return data
|
||||
// @return total
|
||||
// @return err
|
||||
func (s navigation) List(p param.NavigationList) (data []vo.NavigationItem, total int64, err error) {
|
||||
sql := s.Table("t_navigation as tn").
|
||||
Scopes(Paginate(p.Current, p.Size)).
|
||||
Joins("LEFT JOIN t_navigation_type as tnt ON tnt.id = tn.type_id").
|
||||
Select("tn.id,tn.title,tn.url,tn.description,tn.icon,tn.enabled,tn.created_at,tn.updated_at,tnt.id as type_id,tnt.name as type")
|
||||
|
||||
if p.Type != "" {
|
||||
sql.Where("tn.type_id = ?", p.Type)
|
||||
}
|
||||
if p.Title != "" {
|
||||
sql.Where("tn.title LIKE ?", "%"+p.Title+"%")
|
||||
}
|
||||
if p.Link != "" {
|
||||
sql.Where("tn.url LIKE ?", "%"+p.Link+"%")
|
||||
}
|
||||
|
||||
err = sql.Find(&data).Offset(-1).Limit(-1).Count(&total).Error
|
||||
return
|
||||
}
|
||||
|
||||
// SaveNavigation
|
||||
// @description: 新增/编辑导航
|
||||
// @receiver s
|
||||
// @param p
|
||||
// @return error
|
||||
func (s navigation) SaveNavigation(p param.SaveNavigation) error {
|
||||
if p.Id != "" {
|
||||
return s.Model(&model.Navigation{}).Where("id = ?", p.Id).Updates(map[string]any{
|
||||
"title": p.Title,
|
||||
"url": p.Url,
|
||||
"icon": p.Icon,
|
||||
"enabled": p.Enabled,
|
||||
"description": p.Description,
|
||||
}).Error
|
||||
}
|
||||
|
||||
websiteInfo, err := goscraper.Scrape(p.Url, 5)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return s.Create(&model.Navigation{
|
||||
Title: p.Title,
|
||||
Url: p.Url,
|
||||
Icon: fmt.Sprintf("%s%s", p.Url, websiteInfo.Preview.Icon),
|
||||
Enabled: cast.ToBool(p.Enabled),
|
||||
Description: p.Description,
|
||||
}).Error
|
||||
}
|
||||
|
||||
// DeleteNavigation
|
||||
// @description: 删除导航
|
||||
// @receiver s
|
||||
// @param id
|
||||
// @return error
|
||||
func (s navigation) DeleteNavigation(id string) error {
|
||||
return s.Where("id = ?", id).Delete(&model.Navigation{}).Error
|
||||
}
|
||||
Reference in New Issue
Block a user