mysql - Golang API gorilla MySQL 查询

Question

所以...我终于完成了包含 super Nerd 桌面游戏 Warhammer 的副项目，我在其中创建了一个数据库 MySQL，下一步是创建一个 API。

我现在有三张 table ……“tyranids”、“greyknghts”和“deathguard”。我想进行动态查询以选择目标表。我能够做到这一点，但随着表格的增长，我需要使这个动态化。

func getTyranids(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-type", "application/json")

var units []Unit

result, err := db.Query("SELECT * FROM tyranids")

if err != nil {
    panic(err.Error)
}

defer result.Close()

for result.Next() {
    var unit Unit
    err := result.Scan(&unit.ID, &unit.Name, &unit.Type, &unit.Movement, &unit.WeaponsSkill, &unit.BallisticSkill, &unit.Strength, &unit.Toughness, &unit.Wounds, &unit.Attacks, &unit.Leadership, &unit.Initiate, &unit.Points)
    if err != nil {
        panic(err.Error)
    }
    units = append(units, unit)
}
json.NewEncoder(w).Encode(units)

}

我怎样才能写这个，所以我不需要每个表的函数？

我已经使它与每个单独单元的 mux.Vars 一起工作。

func getTyranidUnit(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")

params := mux.Vars(r)

result, err := db.Query("SELECT * FROM tyranids WHERE name = ?", params["name"])
if err != nil {
    panic(err.Error())
}
defer result.Close()

var unit Unit

for result.Next() {
    err := result.Scan(&unit.ID, &unit.Name, &unit.Type, &unit.Movement, &unit.WeaponsSkill, &unit.BallisticSkill, &unit.Strength, &unit.Toughness, &unit.Wounds, &unit.Attacks, &unit.Leadership, &unit.Initiate, &unit.Points)

    if err != nil {
        panic(err.Error())
    }
}
json.NewEncoder(w).Encode(unit)

}

func main() {

db, err = sql.Open("mysql", "xx:xx@tcp(xxx)/Warhammer")

if err != nil {
    panic(err.Error())
}

defer db.Close()

router := mux.NewRouter().StrictSlash(true)

router.HandleFunc("/tyranids", getTyranids).Methods("GET")
router.HandleFunc("/tyranids/{name}", getTyranidUnit).Methods("GET")    

http.ListenAndServe(":8001", router)

}

谢谢你。

Answer 1

你可以做一个map将表与键配对:

speciesUnitMap:= map[string]string{
    "Norn-Queen": "tyranids",
    "Hive Tyrant": "tyranids",
    "Rippers ": "tyranids",
    "Hive Ship": "tyranids",
‎    "Ork Boyz": "orcs",
    "‎Waaagh!": "orcs",
‎    "Warboss": "orcs",
    "‎Blood Axes" "orcs",

}


// This is dirty I would construct the string before 
// and do verifications that the received param is conform
result, err := db.Query("SELECT * FROM " + 
       speciesUnitMap[params["name"]] + " WHERE name = ?", params["name"])