向lib/pq传递足够多的参数

Question

为了简化代码，防止重复传递类似的参数，我为一些结构创建了一个方法Put，如下所示。

例如，此处不使用UID的原因是UID被设置为自动增加值。

type User struct {
    UID      int
    Name     string
    Username string
    Password string
}

func (user *User) Put(db *sql.DB, query string) (sql.Result, error) {
    return db.Exec(query, u.UID, u.Name, u.Username, u.Password)
}

func main() {
    db := ...
    u := &User{...}

    u.Put(db, `INSERT INTO user(name, username, password) VALUES ($2, $3, $4);`)
}

但是我得到了下面的错误

pq: could not determine data type of parameter $1

(SQL驱动程序是https://github.com/lib/pq)

Answer 1

在查询中，$2, $3...$N是作为额外参数传入db.Exec()的参数的索引，它们必须从$1开始(如果使用database/sql包)。

我认为在您的情况下，您可以更改Put方法签名(添加struct values extractor)

func (user *User) Put(db *sql.DB, query string, extractor func(user User)[]interface{}) (sql.Result, error) {
    return db.Exec(query, extractor(*user)...)
}

然后用一个类似的

u.Put(db,`INSERT INTO user(name, username, password) VALUES ($1, $2, $3);`, func(u User) []interface{} {
    return []interface{}{u.Name, u.Username, u.Password}
})