Showing how we can remove services, preventing cyclic imports and lessing the number of imports. Now db is in models. Db and models are highly tightened, according to go standards, you should put them in the same package. In models, there are folders separating the different methods used to modify the models. For example, if you want to create a user, you have to use /models (for the user struct) and /models/user (for creating a user. However, if you want to delete a torrent, you just have to import /models and do torrent.Delete(definitely bool). By the way packages in models are the plural name of a model. For example, you have torrent.go for a torrent model and its package torrents for db stuff related functions (Find, Create, Some helpers)
106 lignes
2,3 Kio
Go
106 lignes
2,3 Kio
Go
package models
|
|
|
|
import (
|
|
"github.com/NyaaPantsu/nyaa/config"
|
|
"github.com/NyaaPantsu/nyaa/util/log"
|
|
"github.com/azhao12345/gorm"
|
|
_ "github.com/jinzhu/gorm/dialects/postgres" // Need for postgres support
|
|
_ "github.com/jinzhu/gorm/dialects/sqlite" // Need for sqlite
|
|
elastic "gopkg.in/olivere/elastic.v5"
|
|
)
|
|
|
|
const (
|
|
// SqliteType : name of the sqlite type in gorm
|
|
SqliteType = "sqlite3"
|
|
)
|
|
|
|
// Logger interface
|
|
type Logger interface {
|
|
Print(v ...interface{})
|
|
}
|
|
|
|
// DefaultLogger : use the default gorm logger that prints to stdout
|
|
var DefaultLogger Logger
|
|
|
|
// ORM : Variable for interacting with database
|
|
var ORM *gorm.DB
|
|
|
|
// ElasticSearchClient : Client for Elastic search
|
|
var ElasticSearchClient *elastic.Client
|
|
|
|
// IsSqlite : Variable to know if we are in sqlite or postgres
|
|
var IsSqlite bool
|
|
|
|
// ElasticSearchInit : Initialization of ES client
|
|
func ElasticSearchInit() (*elastic.Client, error) {
|
|
client, err := elastic.NewClient()
|
|
if err != nil {
|
|
log.Errorf("Unable to create elasticsearch client: %s", err)
|
|
return nil, err
|
|
}
|
|
log.Infof("Using elasticsearch client")
|
|
return client, nil
|
|
}
|
|
|
|
// GormInit init gorm ORM.
|
|
func GormInit(conf *config.Config, logger Logger) (*gorm.DB, error) {
|
|
|
|
db, openErr := gorm.Open(conf.DBType, conf.DBParams)
|
|
if openErr != nil {
|
|
log.CheckError(openErr)
|
|
return nil, openErr
|
|
}
|
|
|
|
IsSqlite = conf.DBType == SqliteType
|
|
|
|
connectionErr := db.DB().Ping()
|
|
if connectionErr != nil {
|
|
log.CheckError(connectionErr)
|
|
return nil, connectionErr
|
|
}
|
|
|
|
// Negative MaxIdleConns means don't retain any idle connection
|
|
maxIdleConns := -1
|
|
if IsSqlite {
|
|
// sqlite doesn't like having a negative maxIdleConns
|
|
maxIdleConns = 10
|
|
}
|
|
|
|
db.DB().SetMaxIdleConns(maxIdleConns)
|
|
db.DB().SetMaxOpenConns(400)
|
|
|
|
if config.Conf.Environment == "DEVELOPMENT" {
|
|
db.LogMode(true)
|
|
}
|
|
|
|
switch conf.DBLogMode {
|
|
case "detailed":
|
|
db.LogMode(true)
|
|
case "silent":
|
|
db.LogMode(false)
|
|
}
|
|
|
|
if logger != nil {
|
|
db.SetLogger(logger)
|
|
}
|
|
|
|
db.AutoMigrate(&User{}, &UserFollows{}, &UserUploadsOld{}, &Notification{}, &Activity{})
|
|
if db.Error != nil {
|
|
return db, db.Error
|
|
}
|
|
db.AutoMigrate(&Torrent{}, &TorrentReport{}, &Scrape{})
|
|
if db.Error != nil {
|
|
return db, db.Error
|
|
}
|
|
db.AutoMigrate(&File{})
|
|
if db.Error != nil {
|
|
return db, db.Error
|
|
}
|
|
db.AutoMigrate(&Comment{}, &OldComment{})
|
|
if db.Error != nil {
|
|
return db, db.Error
|
|
}
|
|
|
|
return db, nil
|
|
}
|