forked from free-lancers/electricity_bill_calc_service
feat(park):完成更新园区状态功能。
This commit is contained in:
@@ -1,8 +1,11 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"electricity_bill_calc/exceptions"
|
||||
"electricity_bill_calc/global"
|
||||
"electricity_bill_calc/model"
|
||||
|
||||
"xorm.io/builder"
|
||||
)
|
||||
|
||||
type _ParkService struct{}
|
||||
@@ -18,9 +21,31 @@ func (_ParkService) SaveNewPark(park model.Park) error {
|
||||
}
|
||||
|
||||
func (_ParkService) UpdateParkInfo(park *model.Park) error {
|
||||
_, err := global.DBConn.ID(park.Id).Cols("name", "abbr", "region", "address", "contact", "phone", "capacity", "tenement_quantity", "category", "meter_04kv_type").Update(park)
|
||||
rows, err := global.DBConn.
|
||||
Where(builder.Eq{"id": park.Id, "user_id": park.UserId}).
|
||||
Cols("name", "abbr", "region", "address", "contact", "phone", "capacity", "tenement_quantity", "category", "meter_04kv_type").
|
||||
Update(park)
|
||||
if err != nil {
|
||||
return err
|
||||
if rows == 0 {
|
||||
return exceptions.NewNotFoundError("未能找到符合条件的园区。")
|
||||
} else {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (_ParkService) ChangeParkState(uid, pid string, state bool) error {
|
||||
rows, err := global.DBConn.
|
||||
Table(&model.Park{}).
|
||||
Where(builder.Eq{"id": pid, "user_id": uid}).
|
||||
Update(map[string]interface{}{"enabled": state})
|
||||
if err != nil {
|
||||
if rows == 0 {
|
||||
return exceptions.NewNotFoundError("未能找到符合条件的园区。")
|
||||
} else {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
Reference in New Issue
Block a user