-
Notifications
You must be signed in to change notification settings - Fork 134
/
functions.go
90 lines (78 loc) · 1.92 KB
/
functions.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
package database
import (
"context"
"database/sql"
"fmt"
"strings"
"time"
"github.com/google/uuid"
"github.com/jackc/pgx/v4"
"github.com/jackc/pgx/v4/pgxpool"
"github.com/pkg/errors"
"github.com/armadaproject/armada/internal/armada/configuration"
)
func CreateConnectionString(values map[string]string) string {
// https://www.postgresql.org/docs/10/libpq-connect.html#id-1.7.3.8.3.5
result := ""
replacer := strings.NewReplacer(`\`, `\\`, `'`, `\'`)
for k, v := range values {
result += k + "='" + replacer.Replace(v) + "'"
}
return result
}
func OpenPgxConn(config configuration.PostgresConfig) (*pgx.Conn, error) {
db, err := pgx.Connect(context.Background(), CreateConnectionString(config.Connection))
if err != nil {
return nil, err
}
err = db.Ping(context.Background())
return db, err
}
func OpenPgxPool(config configuration.PostgresConfig) (*pgxpool.Pool, error) {
db, err := pgxpool.Connect(context.Background(), CreateConnectionString(config.Connection))
if err != nil {
return nil, err
}
err = db.Ping(context.Background())
return db, err
}
func UniqueTableName(table string) string {
suffix := strings.ReplaceAll(uuid.New().String(), "-", "")
return fmt.Sprintf("%s_tmp_%s", table, suffix)
}
func ParseNullStringDefault(nullString sql.NullString) string {
if !nullString.Valid {
return ""
}
return nullString.String
}
func ParseNullString(nullString sql.NullString) *string {
if !nullString.Valid {
return nil
}
return &nullString.String
}
func ParseNullTime(nullTime sql.NullTime) *time.Time {
if !nullTime.Valid {
return nil
}
return &nullTime.Time
}
func ParseNullInt32(nullInt sql.NullInt32) *int32 {
if !nullInt.Valid {
return nil
}
return &nullInt.Int32
}
func ReadInt(rows pgx.Rows) (int, error) {
defer rows.Close()
var val int
for rows.Next() {
err := rows.Scan(&val)
if err != nil {
return -1, err
}
return val, nil
}
return -1, errors.New("no rows found")
}