trantor/lib/database/database.go

package database

import (
	"gopkg.in/pg.v4"
)

type DB interface {
	Close() error
	AddBook(book Book) error
	GetBooks(query string, length int, start int) (books []Book, num int, err error)
	GetNewBooks(query string, length int, start int) (books []Book, num int, err error)
	GetBookId(id string) (Book, error)
	DeleteBook(id string) error
	UpdateBook(id string, data map[string]interface{}) error
	FlagBadQuality(id string, user string) error
	ActiveBook(id string) error
	IsBookActive(id string) bool
	AddUser(name string, pass string) error
	GetRole(name string) (string, error)
	SetPassword(name string, pass string) error
	ValidPassword(name string, pass string) bool
	AddNews(text string) error
	GetNews(num int, days int) (news []New, err error)
	AddStats(stats interface{}) error
	GetVisitedBooks() (books []Book, err error)
	UpdateMostVisited() error
	GetDownloadedBooks() (books []Book, err error)
	UpdateDownloadedBooks() error
	GetTags() ([]string, error)
	UpdateTags() error
	GetVisits(visitType VisitType) ([]Visits, error)
	UpdateHourVisits() error
	UpdateDayVisits() error
	UpdateMonthVisits() error
	UpdateHourDownloads() error
	UpdateDayDownloads() error
	UpdateMonthDownloads() error
}

type pgDB struct {
	sql *pg.DB
}

// Options for the database
type Options struct {
	Addr     string
	User     string
	Password string
	Name     string
}

// Init the database connection
func Init(options Options) (DB, error) {
	sql := pg.Connect(&pg.Options{
		Addr:     options.Addr,
		User:     options.User,
		Password: options.Password,
		Database: options.Name,
	})
	// TODO: create db
	return &pgDB{sql}, nil
}

// Close the database connection
func (db pgDB) Close() error {
	return db.sql.Close()
}

func RO(db DB) DB {
	return &roDB{db}
}