db

Documentation

Index

• type DB
  • func New(cfg configuration.PostgresDB) (*DB, error)
  • func (db *DB) RemoveAll() error
• type PGCovidStore
  • func NewCovidStore(db *DB) *PGCovidStore
  • func (store *PGCovidStore) Add(entries []models.CountryEntry) error
  • func (store *PGCovidStore) CountEntriesByTime(from, to time.Time) ([]TimestampCount, error)
  • func (store *PGCovidStore) GetAllCountryNames() (names []string, err error)
  • func (store *PGCovidStore) GetAllForCountryName(countryName string) ([]models.CountryEntry, error)
  • func (store *PGCovidStore) GetAllForRange(from, to time.Time) ([]models.CountryEntry, error)
  • func (store *PGCovidStore) GetLatestForCountries(endTime time.Time) (map[string]models.CountryEntry, error)
  • func (store *PGCovidStore) GetTotalsPerDay() ([]models.CountryEntry, error)
  • func (store *PGCovidStore) Rows() (int, error)
• type PGPopulationStore
  • func NewPopulationStore(db *DB) *PGPopulationStore
  • func (store *PGPopulationStore) Add(code string, pop int64) error
  • func (store *PGPopulationStore) List() (map[string]int64, error)
• type TimestampCount

Types

type DB

type DB struct {
	Collector prometheus.Collector

	Handle *sqlx.DB
	// contains filtered or unexported fields
}

DB hold the handle to a database. Provides a Prometheus DBStatsCollector to monitor DB connections

func New

func New(cfg configuration.PostgresDB) (*DB, error)

New creates a new DB connector

func (*DB) RemoveAll

func (db *DB) RemoveAll() error

RemoveAll deletes all database tables

type PGCovidStore

type PGCovidStore struct {
	DB *DB
}

PGCovidStore implements CovidStore for Postgres databases

func NewCovidStore

func NewCovidStore(db *DB) *PGCovidStore

NewCovidStore creates a new PGCovidStore and initializes the database, if necessary

func (*PGCovidStore) Add

func (store *PGCovidStore) Add(entries []models.CountryEntry) error

Add inserts new entries in the database

func (*PGCovidStore) CountEntriesByTime

func (store *PGCovidStore) CountEntriesByTime(from, to time.Time) ([]TimestampCount, error)

CountEntriesByTime counts updates per timestamp

func (*PGCovidStore) GetAllCountryNames

func (store *PGCovidStore) GetAllCountryNames() (names []string, err error)

GetAllCountryNames gets all unique country names from the database

func (*PGCovidStore) GetAllForCountryName

func (store *PGCovidStore) GetAllForCountryName(countryName string) ([]models.CountryEntry, error)

GetAllForCountryName returns all entries in the database, sorted by timestamp

func (*PGCovidStore) GetAllForRange

func (store *PGCovidStore) GetAllForRange(from, to time.Time) ([]models.CountryEntry, error)

GetAllForRange returns all entries in the database, sorted by timestamp

func (*PGCovidStore) GetLatestForCountries

func (store *PGCovidStore) GetLatestForCountries(endTime time.Time) (map[string]models.CountryEntry, error)

GetLatestForCountries gets the last entries for each country up the specified endTime. If endTime is time.Time{}, it will get the latest entries up to the current time.

func (*PGCovidStore) GetTotalsPerDay

func (store *PGCovidStore) GetTotalsPerDay() ([]models.CountryEntry, error)

GetTotalsPerDay returns the total new cases per day across all countries

func (*PGCovidStore) Rows

func (store *PGCovidStore) Rows() (int, error)

Rows returns the number of rows in the store

type PGPopulationStore

type PGPopulationStore struct {
	DB *DB
}

PGPopulationStore implements PopulationStore for Postgres databases

func NewPopulationStore

func NewPopulationStore(db *DB) *PGPopulationStore

NewPopulationStore creates a new PostgresDB object

func (*PGPopulationStore) Add

func (store *PGPopulationStore) Add(code string, pop int64) error

Add to Population database table. If a record for the specified country code already exists, it will be updated

func (*PGPopulationStore) List

func (store *PGPopulationStore) List() (map[string]int64, error)

List all records from the Population table

type TimestampCount

type TimestampCount struct {
	Timestamp time.Time
	Count     int
}