93364dac77
List Torrent delete log
Torrent edit log
Comment delete log
And every other logged activities
Can be filtered out by a filter tag ("edit" or "delete" supported)
Pages navigation
Can be accessed by /activities
Added some translation string
Fixed hidden username on api request
Fixed comments username on modpanel
New Activity model
New Activity handler
New Activity Service
Fixed some updating issue for ES when moderating torrents
Be aware deleting torrents and comments return the model now!
103 lignes
2,3 Kio
Go
103 lignes
2,3 Kio
Go
package db
|
|
|
|
import (
|
|
"github.com/NyaaPantsu/nyaa/config"
|
|
"github.com/NyaaPantsu/nyaa/model"
|
|
"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 = "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
|
|
var ElasticSearchClient *elastic.Client
|
|
|
|
// IsSqlite : Variable to know if we are in sqlite or postgres
|
|
var IsSqlite bool
|
|
|
|
func ElasticSearchInit() (*elastic.Client, error) {
|
|
client, err := elastic.NewClient()
|
|
if err != nil {
|
|
log.Errorf("Unable to create elasticsearch client: %s", err)
|
|
return nil, err
|
|
} else {
|
|
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(&model.User{}, &model.UserFollows{}, &model.UserUploadsOld{}, &model.Notification{}, &model.Activity{})
|
|
if db.Error != nil {
|
|
return db, db.Error
|
|
}
|
|
db.AutoMigrate(&model.Torrent{}, &model.TorrentReport{})
|
|
if db.Error != nil {
|
|
return db, db.Error
|
|
}
|
|
db.AutoMigrate(&model.File{})
|
|
if db.Error != nil {
|
|
return db, db.Error
|
|
}
|
|
db.AutoMigrate(&model.Comment{}, &model.OldComment{})
|
|
if db.Error != nil {
|
|
return db, db.Error
|
|
}
|
|
|
|
return db, nil
|
|
}
|