go-mastodon-stats/internal/gomastodonstats/postgresql.go

package gomastodonstats

import (
	"database/sql"
	"fmt"
	"time"

	_ "github.com/lib/pq"
)

// Returns valid DSL for PSQL
func getPostgresConnectionString(schema string) string {
	return fmt.Sprintf(
		"postgresql://%s:%s@%s/%s?sslmode=disable",
		POSTGRESQL_USER,
		POSTGRESQL_PASS,
		POSTGRESQL_HOST,
		schema,
	)
}

// Returns DB connection
func getPostgresConnection(schema string) (*sql.DB, error) {
	return sql.Open("postgres", getPostgresConnectionString(schema))
}

// insertMetric to stats DB
func insertValues(m metric) error {
	db, err := getPostgresConnection(POSTGRESQL_STATS_DB)
	if err != nil {
		return err
	}

	_, err = db.Exec(fmt.Sprintf(
		"INSERT into %s (service,metric_name,metric_time,metric_value) "+
			"VALUES ($1, $2, $3, $4)",
		POSTGRESQL_STATS_TABLE,
	), m.Service, m.MetricName, m.MetricTime, m.MetricValue)
	return err
}

func runIntQuery(schema string, q string) (int, error) {
	var res int
	db, err := getPostgresConnection(schema)
	if err != nil {
		return res, err
	}

	rows, err := db.Query(q)
	if err != nil {
		return res, err
	}
	defer rows.Close()

	for rows.Next() {
		rows.Scan(&res)
	}

	return res, err
}

func runIntQueryWithTime(schema string, q string, t time.Time) (int, error) {
	var res int
	db, err := getPostgresConnection(schema)
	if err != nil {
		return res, err
	}

	rows, err := db.Query(q, t)
	if err != nil {
		return res, err
	}
	defer rows.Close()

	for rows.Next() {
		rows.Scan(&res)
	}

	return res, err
}