upload_dao.go 715 B

12345678910111213141516171819202122232425262728293031323334353637
  1. /*
  2. * @description: 安全巡检数据库操作相关
  3. * @Author: LanJianRong
  4. * @Date: 2020-11-20
  5. * @FilePath: \design_quantity\dao\upload_dao.go
  6. */
  7. package dao
  8. import (
  9. "github.com/go-xorm/xorm"
  10. "go.mod/models"
  11. )
  12. // 数据库操作引擎
  13. type UploadDao struct {
  14. engine *xorm.Engine
  15. }
  16. // 获得一个DAO对象
  17. func NewUploadDao(engine *xorm.Engine) *UploadDao {
  18. return &UploadDao{
  19. engine: engine,
  20. }
  21. }
  22. func (d *UploadDao) Add(upload *models.CmUpload) (int64, error) {
  23. id, err := d.engine.Insert(upload)
  24. return id, err
  25. }
  26. // 获取文件
  27. func (d *UploadDao) GetFile(id int) (*models.CmUpload, error) {
  28. file := &models.CmUpload{}
  29. _, err := d.engine.Where("id = ?", id).Get(file)
  30. return file, err
  31. }