watsonb8/groupme
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix data storage and other things

Browse Source
This commit is contained in:
Tulir Asokan
2018-08-16 19:20:07 +03:00
parent 9c48eeb534
commit 141eba644b
13 changed files with 158 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
database/database.go
View File

@ -24,7 +24,7 @@ import (
type Database struct {
*sql.DB
log *log.Sublogger
log log.Logger
User *UserQuery
Portal *PortalQuery
@ -39,23 +39,29 @@ func New(file string) (*Database, error) {
db := &Database{
DB: conn,
log: log.CreateSublogger("Database", log.LevelDebug),
log: log.Sub("Database"),
}
db.User = &UserQuery{
db: db,
log: log.CreateSublogger("Database/User", log.LevelDebug),
log: db.log.Sub("User"),
}
db.Portal = &PortalQuery{
db: db,
log: log.CreateSublogger("Database/Portal", log.LevelDebug),
log: db.log.Sub("Portal"),
}
db.Puppet = &PuppetQuery{
db: db,
log: log.CreateSublogger("Database/Puppet", log.LevelDebug),
log: db.log.Sub("Puppet"),
}
return db, nil
}
func (db *Database) CreateTables() {
db.User.CreateTable()
db.Portal.CreateTable()
db.Puppet.CreateTable()
}
type Scannable interface {
Scan(...interface{}) error
}