передать массив в запрос go-pg - PullRequest
Новая
       12

передать массив в запрос go-pg

0 голосов
/ 03 июля 2018

Я использую Go-pg, и когда я использую следующий способ выполнить SQL-запрос: 

db.Query(&result, sqlQuery, params)

где params является структурой, подобной следующей: 

type MyParams struct {
    Codes []int
}

и sqlQuery - это 

SELECT id FROM bar WHERE code_id IN (?codes)

в реальном запросе SQL я получаю запрос, подобный этому: 

SELECT id FROM bar WHERE code_id IN ('[7,45]')

Можно ли правильно передать заполнитель массива int для запроса: 

SELECT id FROM bar WHERE code_id IN (7,45)

Ответы [ 3 ]

0 голосов
/ 03 июля 2018

Вы можете попробовать сделать что-то вроде этого: 

// Slice with your integer IDs as strings.
ids := []string{}

// Convert integer ID to string ID.
for _, i := range []int{7, 45} {
    ids = append(ids, strconv.Itoa(i))
}

Теперь вы можете использовать Join func, и окончательный запрос будет выглядеть довольно просто, например: 

in := strings.Join(ids, ",")
query := "SELECT id FROM bar WHERE code_id IN (" + in + ")"
0 голосов
/ 03 июля 2018

Есть пара вещей, которые вы можете сделать:

  1. Продолжайте использовать in (...) и используйте pg.In.
  2. Используйте = any(array) в вашем SQL вместо in (list) и pg.Array в Go, чтобы отправить правильный массив в PostgreSQL.

Первый выглядит так: 

db.Query(&result, `SELECT id FROM bar WHERE code_id IN (?)`, pg.In(params.Codes))

второй выглядит так: 

db.Query(&result, `SELECT id FROM bar WHERE code_id = ANY (?)`, pg.Array(params.Codes))
0 голосов
/ 03 июля 2018

Можно создать строку с разделителем, необходимым для создания строки, а затем передать ее в db.query для получения результата. 

package main

import (
    "fmt"
    "strconv"
    "strings"
)

func main() {
    a := []int{7,45,32}
    str := ConvertToString(a, ",")
    query := `Select * from table1 where ID IN(`+ str +`)`
    fmt.Println(query)
}

func ConvertToString(a []int, delim string) string{
    var b string
    for _, v := range a{
        b += strconv.Itoa(v)
        b += ","
    }
    return strings.Trim(b,",")
}

Рабочий код на Детская площадка .

Отредактировано : -

Вы можете использовать Golang sqlx пакет для вашего требования.

database/sql пакет не проверяет ваш запрос и передает ваш аргументы непосредственно к драйверу, это делает работу с запросами с В пунктах сложно: 

SELECT * FROM users WHERE level IN (?);

Когда это будет подготовлено как утверждение на бэкэнде, bindvar? будет соответствовать только одному аргументу, но то, что часто желательно для этого, чтобы быть переменным числом аргументов в зависимости от длина некоторого среза, например: 

var levels = []int{4, 6, 7}
rows, err := db.Query("SELECT * FROM users WHERE level IN (?);", levels)
...