gfast/app/model/admin/plug_ad/plug_ad.go

package plug_ad

import (
	"github.com/gogf/gf/errors/gerror"
	"github.com/gogf/gf/frame/g"
	"github.com/gogf/gf/os/gtime"
)

// AddReq 用于存储新增广告请求的请求参数
type AddReq struct {
	AdName     string `p:"adName"`     // 广告名称
	AdAdtypeid int    `p:"adAdtypeid"` // 所属位置
	AdCheckid  int    `p:"adCheckid"`  // 1=图片 2=JS
	AdJs       string `p:"adJs"`       // JS代码
	AdPic      string `p:"adPic"`      // 广告图片URL
	AdUrl      string `p:"adUrl"`      // 广告链接
	AdContent  string `p:"adContent"`  // 广告文字内容
	AdSort     int    `p:"adSort"`     // 排序
	AdOpen     int    `p:"adOpen"`     // 1=审核  0=未审核
}

// EditReq 用于存储修改广告位请求参数
type EditReq struct {
	PlugAdID int64 `p:"plugAdID"`
	AddReq
}

// SelectPageReq 用于存储分页查询广告的请求参数
type SelectPageReq struct {
	AdName   string `p:"adName"`   // 广告名称
	PageNo   int64  `p:"pageNo"`   // 当前页
	PageSize int64  `p:"pageSize"` // 每页显示记录数
}

// GetPlugAdByID 根据ID查询广告记录
func GetPlugAdByID(id int64) (*Entity, error) {
	entity, err := Model.FindOne("ad_id", id)
	if err != nil {
		g.Log().Error(err)
		err = gerror.New("根据ID查询广告记录出错")
	}
	if entity == nil {
		err = gerror.New("根据ID未能查询到广告记录")
	}
	return entity, nil
}

// AddSave 添加广告
func AddSave(req *AddReq) error {
	var entity Entity
	entity.AdName = req.AdName                // 名称
	entity.AdAdtypeid = req.AdAdtypeid        // 位置
	entity.AdCheckid = req.AdCheckid          // 1=图片 2=JS
	entity.AdJs = req.AdJs                    // JS代码
	entity.AdPic = req.AdPic                  // 广告图片URL
	entity.AdUrl = req.AdUrl                  // 广告链接
	entity.AdContent = req.AdContent          // 广告文字内容
	entity.AdAddtime = int(gtime.Timestamp()) // 时间戳
	entity.AdSort = req.AdSort                // 排序
	entity.AdOpen = req.AdOpen                // 1=审核  0=未审核
	// 保存实体
	_, err := entity.Insert()
	if err != nil {
		g.Log().Error(err)
		err = gerror.New("保存失败")
		return err
	}
	return nil
}

// 删除广告
func DeleteByIDs(Ids []int) error {
	_, err := Model.Delete("ad_id in(?)", Ids)
	if err != nil {
		g.Log().Error(err)
		return gerror.New("删除广告失败")
	}
	return nil
}

// 根据广告ID来修改广告位信息
func EditSave(editReq *EditReq) error {
	// 先根据ID来查询要修改的广告记录
	entity, err := GetPlugAdByID(editReq.PlugAdID)
	if err != nil {
		return err
	}
	// 修改实体
	entity.AdName = editReq.AdName
	entity.AdAdtypeid = editReq.AdAdtypeid
	entity.AdCheckid = editReq.AdCheckid
	entity.AdJs = editReq.AdJs
	entity.AdPic = editReq.AdPic
	entity.AdUrl = editReq.AdUrl
	entity.AdContent = editReq.AdContent
	entity.AdSort = editReq.AdSort
	entity.AdOpen = editReq.AdOpen
	_, err = entity.Update()
	if err != nil {
		g.Log().Error(err)
		return gerror.New("修改广告位失败")
	}
	return nil
}

// 分页查询,返回值total总记录数,page当前页
func SelectListByPage(req *SelectPageReq) (total int, page int64, list []*Entity, err error) {
	model := Model
	if req != nil {
		if req.AdName != "" {
			model = model.Where("ad_name like ?", "%"+req.AdName+"%")
		}
	}
	// 查询广告位总记录数(总行数)
	total, err = model.Count()
	if err != nil {
		g.Log().Error(err)
		err = gerror.New("获取总记录数失败")
		return 0, 0, nil, err
	}
	if req.PageNo == 0 {
		req.PageNo = 1
	}
	page = req.PageNo
	if req.PageSize == 0 {
		req.PageSize = 10
	}
	// 分页排序查询
	list, err = model.Page(int(page), int(req.PageSize)).Order("ad_sort asc,ad_id asc").All()
	if err != nil {
		g.Log().Error(err)
		err = gerror.New("分页查询广告失败")
		return 0, 0, nil, err
	}
	return total, page, list, nil
}