70 lines
1.7 KiB
Go
70 lines
1.7 KiB
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
|
|
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}
|
|
}
|