Как использовать предложения 'where id in' с jackc/pgx?

Предлагает ли pgx поддержку предложений where? Я обнаружил в другом потоке stackru, что нужно использовать конкатенацию строк для создания запроса вручную. IMO, хотя это немного подвержено ошибкам, так как вы должны позаботиться о экранировании / sql-инъекции и тому подобном самостоятельно.

Я также попытался понять это самостоятельно:

const updatePurgedRecordingsStmt = "update recordings set status = 'DELETED', deleted = now() where status <> 'DELETED' and id in ($1);"

func (r *Repository) DeleteRecordings() error {
    pool, err := r.connPool()
    if err != nil {
        return errors.Wrap(err, "cannot establish connection")
    }
    pgRecIds := &pgtype.Int4Array{}

    if err := pgRecIds.Set([]int32{int32(1), int32(2)}); err != nil {
        return errors.Wrap(err, "id conversion failed")
    }
    if _, err = pool.Exec(updatePurgedRecordingsStmt, pgRecIds); err != nil {
        return errors.Wrap(err, "update stmt failed")
    }
    return nil
}

Когда я выполняю этот код, я получаю следующую ошибку:

ОШИБКА: неверный двоичный формат данных в параметре связывания 1 (SQLSTATE 22P03)

Версии, которые я использую:

Postgres:

db=> SELECT version();
                                              version
-----------------------------------------------------------------------------------------------------------
 PostgreSQL 9.6.11 on x86_64-pc-linux-gnu, compiled by gcc (GCC) 4.8.2 20140120 (Red Hat 4.8.2-16), 64-bit
(1 row)

PGX:

github.com/jackc/fake v0.0.0-20150926172116-812a484cc733 h1:vr3AYkKovP8uR8AvSGGUK1IDqRa5lAAvEkZG1LKaCRc=
github.com/jackc/fake v0.0.0-20150926172116-812a484cc733/go.mod h1:WrMFNQdiFJ80sQsxDoMokWK1W5TQtxBFNpzWTD84ibQ=
github.com/jackc/pgx v3.3.0+incompatible h1:Wa90/+qsITBAPkAZjiByeIGHFcj3Ztu+VzrrIpHjL90=
github.com/jackc/pgx v3.3.0+incompatible/go.mod h1:0ZGrqGqkRlliWnWB4zKnWtjbSWbGkVEFm4TeybAXq+I=
github.com/lib/pq v1.0.0 h1:X5PMW56eZitiTeO7tKzZxFCSpbFZJtkMMooicw2us9A=
github.com/lib/pq v1.0.0/go.mod h1:5WUZQaWbwv1U+lTReE5YruASi9Al49XbQIvNi/34Woo=
Источник

2 ответа

Решение

Как ты уже знаешь IN ожидает список скалярных выражений, а не массив, однако pgtype.Int4Array представляет массив, а не список скалярных выражений.

"IMO, это немного подвержено ошибкам, так как вы должны позаботиться о том, чтобы избежать /sql-инъекций и т.п. самостоятельно".

Необязательно, вы можете зациклить свой массив, создать строку ссылок на параметры, объединить это с запросом и затем выполнить его, передав в массив с ...,

var paramrefs string
ids := []int{1,2,3,4}
for i, _ := range ids {
    paramrefs += `$` + strconv.Itoa(i+1) + `,`
}
paramrefs = paramrefs[:len(paramrefs)-1] // remove last ","

query := `UPDATE ... WHERE id IN (` + paramrefs + `)`
pool.Exec(query, ids...)

В качестве альтернативы вы можете использовать ANY вместо IN,

ids := &pgtype.Int4Array{}
ids.Set([]int{1,2,3,4})
query := `UPDATE ... WHERE id = ANY ($1)`
pool.Exec(query, ids)

(здесь вам может потребоваться привести ссылку на параметр к соответствующему типу массива, я не уверен, попробуйте без преобразования, если не в порядке, попробуйте с приведением)

Источник 
      func prepareWhereINString(count int) string {
    var paramrefs string
    for i := 0; i < count; i++ {
        paramrefs += `$` + strconv.Itoa(i+1) + `,`
    }
    paramrefs = paramrefs[:len(paramrefs)-1] // remove last ","
    return paramrefs
}
Источник
Другие вопросы по тегам