Удобное сканирование в структуры в связке Go/PgX. Решение проблемы сканирования в PgX. Golang28.03.2025 14:31

3b9b8e0e48e26fbff11ef07bc869a0d3

Работая с базами данных в Go, многие из нас сталкивались с библиотекой pgx — высокопроизводительным драйвером для PostgreSQL. Однако при всей ее мощи есть одна неудобная особенность: сканирование результатов запросов в структуры может быть довольно громоздким, особенно когда речь идет о вложенных структурах или списках.

Проблема неудобного сканирования в go/pgx

Основная сложность заключается в том, что при выполнении SQL-запросов и получении результатов нам часто приходится вручную сканировать каждое поле и сопоставлять его с соответствующим полем в структуре Go. Если структура простая, это не вызывает больших затруднений. Но когда мы имеем дело с вложенными структурами или списками, код становится менее читаемым и более подверженным ошибкам.

Пример стандартного подхода: 

type User struct {
    ID    int
    Name  string
    Email string
}

row := conn.QueryRow(context.Background(), "SELECT id, name, email FROM users WHERE id=$1", userID)
var user User
err := row.Scan(&user.ID, &user.Name, &user.Email)
if err != nil {
    // обработка ошибки
}

А теперь представьте, что у нас есть поле с массивом или вложенными структурами. Код быстрого превращается в громоздкий и сложный для поддержки.

Решение: использование pgxWrappy

Недавно я наткнулся на библиотеку pgxWrappy, которая существенно упрощает этот процесс. Она представляет собой обертку над pgx, позволяющую удобно сканировать результаты запросов непосредственно в вложенные структуры и срезы.

Ключевые возможности pgxWrappy

  • Легкое сканирование во вложенные структуры: Автоматическое сопоставление результатов SQL-запросов со структурами Go, включая вложенные структуры.

  • Удобная работа со срезами: Поддержка сканирования нескольких строк в срезы структур или указателей на структуры.

  • Поддержка транзакций: Предоставляет обертки для работы с транзакциями, включая методы для начала, фиксации и отката транзакций.

  • Интеграция с pgx: Построена поверх высокопроизводительного драйвера pgx, используя его надежные функции.

Как начать использовать pgxWrappy

Установка

Установите пакет с помощью go get: 

go get github.com/Arlandaren/pgxWrappy

Пример использования

Инициализация

import (
    "context"
    "github.com/Arlandaren/pgxWrappy"
)

func main() {
    db, err := pgxwrappy.Connect(context.Background(), "postgres://username:password@localhost:5432/database")
    if err != nil {
        // обработка ошибки
    }
    defer db.Close()
}

Сканирование в структуры

Допустим, у нас есть следующие структуры: 

type Profile struct {
    Bio string `db:"bio"`
}

type User struct {
    ID      int     `db:"id"`
    Name    string  `db:"name"`
    Email   string  `db:"email"`
    Profile Profile `db:"profile"`
}

Теперь мы можем выполнить запрос и автоматически сканировать результат: 

var user User
err := db.Get(context.Background(), &user, "SELECT id, name, email, profile FROM users WHERE id=$1", userID)
if err != nil {
    // обработка ошибки
}

Работа со срезами

var users []User
err := db.Select(context.Background(), &users, "SELECT id, name, email, profile FROM users")
if err != nil {
    // обработка ошибки
}

Почему стоит выбрать pgx и pgxWrappy

pgx уже известен своей производительностью и гибкостью по сравнению с другими драйверами PostgreSQL для Go. Использование pgxWrappy поверх pgx позволяет сохранить все преимущества pgx, одновременно устраняя проблему неудобного сканирования результатов запросов.

Краткое сравнение с другими драйверами

  • database/sql: Стандартная библиотека Go, но не предоставляет всех возможностей pgx и может быть менее эффективной.

  • pq: Популярный драйвер, но не такой производительный и гибкий, как pgx.

  • pgx: Высокая производительность, поддержка современных возможностей PostgreSQL, отсутствие промежуточных преобразований.

Заключение

Если вы ищете способ упростить работу с результатами запросов PostgreSQL в Go, рекомендую обратить внимание на pgxWrappy. Эта библиотека позволяет сконцентрироваться на логике приложения, избавляя от рутины ручного сканирования данных.

Ссылки:

  • pgxWrappy на GitHub

  • Документация pgxWrappy

  • pgx на GitHub

С помощью pgxWrappy вы сможете сделать ваш код более чистым, поддерживаемым и сконцентрироваться на решении бизнес-задач, а не на технических деталях взаимодействия с базой данных.

© Habrahabr.ru