sub.go 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. package pick
  2. import (
  3. "zhiyuan/models"
  4. "zhiyuan/pkg/app"
  5. "zhiyuan/pkg/db"
  6. "zhiyuan/services/form"
  7. )
  8. var MatPick models.MatPick
  9. func CountPick(where map[string]interface{}) (int64, error) {
  10. return db.Count(MatPick.TableName(), where)
  11. }
  12. func GetPicks(where map[string]interface{}, fields []string, page app.Page, retVal interface{}) ([]*models.MatPick, error) {
  13. if page.PageNum > 0 && page.PageSize > 0 {
  14. where["_limit"] = db.GetOffset(uint(page.PageNum), uint(page.PageSize))
  15. }
  16. return MatPick.GetMulti(where, fields, retVal)
  17. }
  18. func GetPick(where map[string]interface{}, fields []string, dest interface{}) (*models.MatPick, error) {
  19. return MatPick.GetOne(where, fields, dest)
  20. }
  21. func getPickMap(form form.MaterialPickAdd) map[string]interface{} {
  22. pickMap := map[string]interface{}{
  23. "pick_name": form.PickName,
  24. //"pkg_id": form.PkgID,
  25. "room_type": form.RoomType,
  26. "standard": form.Standard,
  27. "confirm_type": form.ConfirmType,
  28. "cycle": form.Cycle,
  29. "level": form.Level,
  30. }
  31. return pickMap
  32. }
  33. func AddPick(form form.MaterialPickAdd) (int64, error) {
  34. return db.InsertOne(MatPick.TableName(), getPickMap(form))
  35. }
  36. func EditPick(form form.MaterialPickAdd, id int) (int64, error) {
  37. return db.Update(MatPick.TableName(), map[string]interface{}{"id": id}, getPickMap(form))
  38. }
  39. func DelPick(id int) (int64, error) {
  40. return db.Delete(MatPick.TableName(), map[string]interface{}{"id": id})
  41. }