Как получить ответ в виде массива из базы данных?

Question

[VaneS Ri_Lax]

Добрый день!
Пытаюсь разобраться как в golang работать с базой данных.
В качестве СУБД у меня postgres
Написал вот такой код:

func main() {
	db, err := sql.Open("postgres", config.PostgresqlConfig)
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()

	rows, err := db.Query("SELECT * FROM tablename")
	if err != nil {
		log.Fatal(err)
	}
	defer rows.Close()

	for rows.Next() {
		var a, b string

		err := rows.Scan(&a, &b)

		if err != nil {
			fmt.Println(err)
			continue
		}
		fmt.Println(a, b)
	}
}

Данные я записываю в переменные вот так:
err := rows.Scan(&a, &b)
Все работает, но мне так не совсем удобно, могу ли я как то получить ответ в виде массива или слайса?

Comments

[devalone]

Советую go-pg

Answer 1

[Дмитрий Шицков]

Удобнее всего работать с базой данных в golang с использованием ORM. Например, gorm.

Answer 2

[Andrey Tsvetkov]

Внутри цикла просто вставляйте свои переменные в слайс, а после цикла занимайтесь обработкой полученного слайса

Comments

[VaneS Ri_Lax]

Это не универсально, я хочу написать функцию, где входные данные это SQL, а на выходе я хочу получать слайс с данными. А в вашем варианте под каждый запрос я должен переписывать 1 код в разных участках программы под разные нужды.

[Дмитрий Шицков]

VaneS Ri_Lax,

под каждый запрос я должен переписывать 1 код в разных участках программы под разные нужды.

Что-то не так вы понимаете. Как раз ответ подразумевает, что полученные данные складываются в слайс.
Делаете запрос к базе, в результате узнаете из респонза число полей и, если нужно, их названия(чтобы, скажем, собрать мапы вместо слайсов). Дальше дело техники - создавать слайс нужного размера и скидывать в него значения.

[VaneS Ri_Lax]

Дмитрий Шицков, Может я чего и не понимаю, вот смотрите я хочу написать что то типа

func dbclient(reqSQL string)[][] string{
	var res [][]string

	db, err := sql.Open("postgres", config.PostgresqlConfig)
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()

	rows, err := db.Query(reqSQL)
	if err != nil {
		log.Fatal(err)
	}
	defer rows.Close()

	for rows.Next() {
		var item []string

		err := rows.Scan(&item) //как мне тут данные сложить в слайс?
		if err != nil {
			fmt.Println(err)
			continue
		}
		res = append(res, item) //тут думаю все понятно
	}

	return res
}

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

[Дмитрий Шицков]

VaneS Ri_Lax, вот готовый пример https://stackoverflow.com/questions/14477941/read-...

[Andrey Tsvetkov]

Сегодня вы хотите слайс с динамическим числом полей из базы. Завтра вы захотите добавить в ответ одну чиселку, всего одну и вам покажется, что можно передавать строкой из базы, а после в нужном месте конвертнуть в число через strconv. Послезавтра чиселок станет больше, а так же добавятся булеан типы и, боже упаси, децималы с NULL-able и вы решите что пора переходить на сторону зла с массивом интерфейсов и рефлексией. Испытывая ежедневное страдание и боль от работы с вашей чудо-функцией вы когда-нибудь придёте к мысли, что ваша идея написать универсальное решение для запросов в базу с треском провалилась и вы начнёте использовать как все нормальные люди под каждый запрос свою структуру, если конечно кто-нибудь не переманит вас на совсем скользкий путь орээмов

Answer 3

[Папа Стифлера]

С драйвером "из коробки" не сможете.
Для постгри есть альтернатива - go-pg (https://github.com/go-pg/pg). Он позволяет делать то, что вам нужно. И еще кучу всего.