Go + sqlx에서 슬라이스를 사용하여 쿼리에 여러 변수를 전달하는 방법

10746 단어 5PostgreSQLSQLsqlx

소개



Go + sqlx에서 슬라이스를 사용하여 쿼리에 여러 변수를 전달하는 방법을 잘 모르기 때문에 시도해 보았습니다.

환경



이것으로 만든 환경을 사용하고 있다.
Docker를 사용하여 PostgreSQL+Go의 개발 환경을 만들어 보았다 - Qiita

데이터는 이런 식으로 들어 있다.


코드



간단한 방법



슬라이스를 그대로 낸다.
package main

import (
    "fmt"
    "log"

    "github.com/jmoiron/sqlx"
    _ "github.com/lib/pq"
)

// Employee は構造体です
type Employee struct {
    id     string
    number string
}

func main() {
    db, err := sqlx.Connect("postgres", "host=db port=5432 user=root sslmode=disable")
    if err != nil {
        log.Fatalln(err)
    }

    var nums = []int{4445, 5432}
    query, args, err := sqlx.In("SELECT * FROM employee WHERE emp_number IN (?);", nums)
    query = db.Rebind(query)
    rows, err := db.Query(query, args...)
    if err != nil {
        fmt.Println(err)
    }
    var es []Employee
    for rows.Next() {
        var e Employee
        rows.Scan(&e.id, &e.number)
        es = append(es, e)
    }
    fmt.Println(es)
}

map을 사용한 방법



여기라면, 복수의 변수(nums와 ids라든지)를 건네줄 수 있다.
고뇨고뇨하고 슬라이스를 전개하고 있다.
package main

import (
    "fmt"
    "log"

    "github.com/jmoiron/sqlx"
    _ "github.com/lib/pq"
)

// Employee は構造体です
type Employee struct {
    id     string
    number string
}

func main() {
    db, err := sqlx.Connect("postgres", "host=db port=5432 user=root sslmode=disable")
    if err != nil {
        log.Fatalln(err)
    }

    arg := map[string]interface{}{
        "nums": []int{4445, 5432},
    }

    query, args, err := sqlx.Named(`SELECT * FROM employee WHERE emp_number IN (:nums);`, arg)
    fmt.Println(query)
    query, args, err = sqlx.In(query, args...)
    fmt.Println(query)
    query = db.Rebind(query)
    rows, err := db.Query(query, args...)
    if err != nil {
        fmt.Println(err)
    }
    var es []Employee
    for rows.Next() {
        var e Employee
        rows.Scan(&e.id, &e.number)
        es = append(es, e)
    }
    fmt.Println(es)
}

결론



sqlx를 사용했지만 sql 패키지로도 할 수 있거나 괜찮습니까? 잘 모르겠어요.
구조체로 전달하는 방법도 있지만, 그 곳은 slice를 넣는 법을 찾지 못했다. 할 수 있을까? 잘 모르겠어요.

참고



jmoiron/sqlx: general purpose extensions to golang's database/sql
Illustrated Guide to SQLX
Named parameters don't work with IN statement · Issue #485 · jmoiron/sqlx

좋은 웹페이지 즐겨찾기