对 MySQL 数据库进行动态 SQL 查询

Question

我以前从未在 mysql 中使用过 golang，所以我是第一次阅读这些内容。我想做这样的事情：

if userId && gender && age
db.QueryRow("SELECT name FROM users WHERE userId=? AND gender=? AND age=?", userId,gender,age)
else if gender && age
db.QueryRow("SELECT name FROM users WHERE gender=? AND age=?", gender, age)
else if userId && gender
db.QueryRow("SELECT name FROM users WHERE userId=? AND gender=?", userId,gender)
else if userId && age
db.QueryRow("SELECT name FROM users WHERE userId=? AND age=?", userId, age)
else if gender
db.QueryRow("SELECT name FROM users WHERE gender=?", gender)
else if userId
db.QueryRow("SELECT name FROM users WHERE userId=?", userId)
else if age
db.QueryRow("SELECT name FROM users WHERE age=?", age)

这是太多的输入，尤其是如果我有十几个变量要添加到 WHERE 条件中。

如果这是 PHP，我会这样做：

$sql = "SELECT name FROM users ";
$where = array();
foreach(explode(",","userId,gender,age,name,height,weight,ethnicity" as $field)
{
    if(isset($arrayOfValues[$field]))
    {
        $where[count($where)] = $field." = ?".$field
        $db->bind("?".$field,$arrayOfValues[$field]);
    }
}

if(count($where)>0)
$sql = $sql . " WHERE ".implode(" AND ",$where);
$db->query($sql);

通过使用 foreach 循环，我可以动态生成查询并根据需要动态绑定尽可能多的变量。

golang 和 mysql 有类似的选项吗？或者是否有替代方法可以不为查询输入每个变量组合？

Answer 1

如果您有这样的字段名称和值的地图：

m := map[string]interface{}{"UserID": 1234, "Age": 18}

然后您可以像这样构建查询：

var values []interface{}
var where []string
for _, k := range []string{"userId", "gender", "age", "name", "height", "weight", "ethnicity"} {
    if v, ok := m[k]; ok {
        values = append(values, v)
        where = append(where, fmt.Sprintf("%s = ?", k))
    }
}
r, err := db.QueryRow("SELECT name FROM users WHERE " + strings.Join(where, " AND "), values...)

这不易受到 SQL 注入的影响，因为占位符用于应用程序直接控制之外的部分查询。

如果已知映射键是允许的字段名称，则使用此：

var values []interface{}
var where []string
for k, v := range m {
    values = append(values, v)
    where = append(where, fmt.Sprintf("%s = ?", k))
}
r, err := db.QueryRow("SELECT name FROM users WHERE " + strings.Join(where, " AND "), values...)