优化代码 var modelMap = map[string]interface{}{ "comb": &models.PsComb{}, "conduit": &models.PsConduit{}, "feature": &models.Feature{}, "discharger": &models.PsDischarger{}, "gate": &models.PsGate{}, "interception": &models.PsInterception{}, "lake": &models.PsLake{}, "manhole": &models.PsManhole{}, "outfall": &models.PsOutfall{}, "pipe": &models.PsPipe{}, "pumpStation": &models.PsPumpStation{}, "retention": &models.PsRetention{}, "river": &models.PsRiver{}, "wwtp": &models.PsWWTP{}, "psSystem": &models.PsSystem{}, "landUse": &models.LandUse{}, "publicity": &models.Publicity{}, } func Batch(ctx *gin.Context) { var requ Request if err := ctx.ShouldBind(&requ); err != nil { ctx.JSON(400, models.Response{ Code: 400, Msg: "参数有误", Data: err.Error(), }) return } resp := models.Response{ Code: 0, Msg: "success", } for _, item := range requ.Post { if reqType, ok := modelMap[item.Type]; ok { if err := mapstructure.Decode(item.Value, reqType); err != nil { panic(err) } res := common.DB.Create(reqType) if res.Error != nil { resp.Code = 500 resp.Msg = "创建失败" resp.Data = res.Error ctx.JSON(500, resp) return } ctx.JSON(200, resp)
时间: 2023-07-23 20:15:56 浏览: 157
这段代码看起来比较简单,但可以从以下几个方面进行优化:
1. 将 `modelMap` 改为在 `init()` 函数中初始化,而不是在每次调用 `Batch()` 函数时初始化。这样可以避免重复初始化,提高代码的性能。
2. 针对循环中的每个 `item`,可以采用批量插入(bulk insert)的方式,而不是逐个插入。这样可以减少数据库交互的次数,提高代码的性能。例如,可以将所有待插入的数据构造成一个切片,然后通过 `common.DB.CreateInBatches()` 方法进行批量插入。
3. 当出现错误时,应该立即返回错误信息,而不是继续进行循环。这样可以避免不必要的数据库交互和数据错误。
下面是一个优化后的示例代码:
```
var modelMap = map[string]interface{}{
"comb": &models.PsComb{},
"conduit": &models.PsConduit{},
"feature": &models.Feature{},
"discharger": &models.PsDischarger{},
"gate": &models.PsGate{},
"interception": &models.PsInterception{},
"lake": &models.PsLake{},
"manhole": &models.PsManhole{},
"outfall": &models.PsOutfall{},
"pipe": &models.PsPipe{},
"pumpStation": &models.PsPumpStation{},
"retention": &models.PsRetention{},
"river": &models.PsRiver{},
"wwtp": &models.PsWWTP{},
"psSystem": &models.PsSystem{},
"landUse": &models.LandUse{},
"publicity": &models.Publicity{},
}
func init() {
for _, model := range modelMap {
common.DB.AutoMigrate(model)
}
}
func Batch(ctx *gin.Context) {
var requ Request
if err := ctx.ShouldBind(&requ); err != nil {
ctx.JSON(400, models.Response{
Code: 400,
Msg: "参数有误",
Data: err.Error(),
})
return
}
resp := models.Response{
Code: 0,
Msg: "success",
}
var values []interface{}
for _, item := range requ.Post {
if reqType, ok := modelMap[item.Type]; ok {
if err := mapstructure.Decode(item.Value, reqType); err != nil {
ctx.JSON(500, models.Response{
Code: 500,
Msg: "创建失败",
Data: err.Error(),
})
return
}
values = append(values, reqType)
}
}
if len(values) > 0 {
res := common.DB.CreateInBatches(values, len(values))
if res.Error != nil {
ctx.JSON(500, models.Response{
Code: 500,
Msg: "创建失败",
Data: res.Error,
})
return
}
}
ctx.JSON(200, resp)
}
```
注意,这只是一种示例代码,具体实现可能需要根据您的实际情况进行调整。
阅读全文