111 lignes
2,7 Kio
Go
111 lignes
2,7 Kio
Go
package database
|
|
|
|
import (
|
|
"github.com/NyaaPantsu/nyaa/common"
|
|
"github.com/NyaaPantsu/nyaa/config"
|
|
"github.com/NyaaPantsu/nyaa/database/postgres"
|
|
//"github.com/NyaaPantsu/nyaa/db/sqlite"
|
|
"github.com/NyaaPantsu/nyaa/model"
|
|
"github.com/NyaaPantsu/nyaa/util/log"
|
|
|
|
"errors"
|
|
)
|
|
|
|
// Database obstraction layer
|
|
type Database interface {
|
|
|
|
// Initialize internal state
|
|
Init() error
|
|
|
|
// return true if we need to call MigrateNext again
|
|
NeedsMigrate() (bool, error)
|
|
// migrate to next database revision
|
|
MigrateNext() error
|
|
|
|
// get torrents given parameters
|
|
GetTorrentsWhere(param *common.TorrentParam) ([]model.Torrent, error)
|
|
|
|
// insert new comment
|
|
InsertComment(comment *model.Comment) error
|
|
// new torrent report
|
|
InsertTorrentReport(report *model.TorrentReport) error
|
|
|
|
// check if user A follows B (by id)
|
|
UserFollows(a, b uint32) (bool, error)
|
|
|
|
// delete reports given params
|
|
DeleteTorrentReportsWhere(param *common.ReportParam) (uint32, error)
|
|
|
|
// get reports given params
|
|
GetTorrentReportsWhere(param *common.ReportParam) ([]model.TorrentReport, error)
|
|
|
|
// bulk record scrape events in 1 transaction
|
|
RecordScrapes(scrapes []common.ScrapeResult) error
|
|
|
|
// insert new user
|
|
InsertUser(u *model.User) error
|
|
|
|
// update existing user info
|
|
UpdateUser(u *model.User) error
|
|
|
|
// get users given paramteters
|
|
GetUsersWhere(param *common.UserParam) ([]model.User, error)
|
|
|
|
// delete many users given parameters
|
|
DeleteUsersWhere(param *common.UserParam) (uint32, error)
|
|
|
|
// get comments by given parameters
|
|
GetCommentsWhere(param *common.CommentParam) ([]model.Comment, error)
|
|
|
|
// delete comment by given parameters
|
|
DeleteCommentsWhere(param *common.CommentParam) (uint32, error)
|
|
|
|
// add user A following B
|
|
AddUserFollowing(a, b uint32) error
|
|
|
|
// delete user A following B
|
|
DeleteUserFollowing(a, b uint32) (bool, error)
|
|
|
|
// insert/update torrent
|
|
UpsertTorrent(t *model.Torrent) error
|
|
|
|
// XXX: add more as needed
|
|
}
|
|
|
|
var ErrInvalidDatabaseDialect = errors.New("invalid database dialect")
|
|
var ErrSqliteSucksAss = errors.New("sqlite3 sucks ass so it's not supported yet")
|
|
|
|
var Impl Database
|
|
|
|
func Configure(conf *config.Config) (err error) {
|
|
switch conf.DBType {
|
|
case "postgres":
|
|
Impl, err = postgres.New(conf.DBParams)
|
|
break
|
|
case "sqlite3":
|
|
err = ErrSqliteSucksAss
|
|
// Impl, err = sqlite.New(conf.DBParams)
|
|
break
|
|
default:
|
|
err = ErrInvalidDatabaseDialect
|
|
}
|
|
if err == nil {
|
|
log.Infof("Init %s database", conf.DBType)
|
|
err = Impl.Init()
|
|
}
|
|
return
|
|
}
|
|
|
|
// Migrate migrates the database to latest revision, call after Configure
|
|
func Migrate() (err error) {
|
|
next := true
|
|
for err == nil && next {
|
|
next, err = Impl.NeedsMigrate()
|
|
if next {
|
|
err = Impl.MigrateNext()
|
|
}
|
|
|
|
}
|
|
return
|
|
}
|