完善季度

repository/peak-valley/peak_valley_quarter.go

@@ -0,0 +1,43 @@
+package peak_valley
+
+import (
+	"energy-management-system/form"
+	"energy-management-system/global"
+	peak_valley_model "energy-management-system/model/peak-valley"
+)
+
+// 季度谷峰规则
+
+// 开始时间
+// 结束时间
+// 谷峰规则ID
+
+func (r *PeakValley) GetPeakValleyQuarterPage(req *form.PeakValleyQuarterListReq) (count int64, list []*peak_valley_model.PeakValleyQuarter, err error) {
+	db := global.Db.Model(&peak_valley_model.PeakValleyQuarter{})
+
+	//for key, value := range qr {
+	//	db = db.Where(key, value)
+	//}
+	err = db.Count(&count).Error
+	if err != nil {
+		return
+	}
+	err = db.Offset((req.Page.GetPageIndex() - 1) * req.Page.GetPageSize()).Limit(req.Page.GetPageSize()).Order("id desc").Preload("Rule").Find(&list).Error
+	return
+}
+
+func (r *PeakValley) CreatePeakValleyQuarter(d *peak_valley_model.PeakValleyQuarter) error {
+	return global.Db.Create(d).Error
+}
+func (r *PeakValley) UpdatePeakValleyQuarter(d *peak_valley_model.PeakValleyQuarter) error {
+	return global.Db.Save(d).Error
+}
+
+func (r *PeakValley) GetOnePeakValleyQuarter(qr map[string]interface{}) (d *peak_valley_model.PeakValleyQuarter, err error) {
+	db := global.Db
+	for key, value := range qr {
+		db = db.Where(key, value)
+	}
+	err = db.First(&d).Error
+	return
+}