zhangqian
2023-11-01 530fed8ec225453572d57b15c200ab062c335457
model/city.go
@@ -39,6 +39,9 @@
   if slf.Name != "" {
      db = db.Where("name = ?", slf.Name)
   }
   if slf.ProvinceId != 0 {
      db = db.Where("province_id = ?", slf.ProvinceId)
   }
   return db
}
@@ -71,6 +74,11 @@
   return slf
}
func (slf *CitySearch) SetProvinceId(provinceId int) *CitySearch {
   slf.ProvinceId = provinceId
   return slf
}
func (slf *CitySearch) SetName(name string) *CitySearch {
   slf.Name = name
   return slf
@@ -100,3 +108,24 @@
   err := db.Find(&list).Error
   return list, err
}
// InitDefaultCity 初始化默认城市
func InitDefaultCity() error {
   records := []*City{
      {
         Name: "北京市",
      },
   }
   var db = mysqlx.GetDB()
   for _, record := range records {
      var city City
      if err := db.Where("name = ?", record.Name).First(&city).Error; err != nil {
         if err := db.Create(record).Error; err != nil {
            return err
         }
      }
   }
   return nil
}