23 lines
491 B
Go
23 lines
491 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"gorm.io/driver/sqlite"
|
||
|
|
"gorm.io/gorm"
|
||
|
|
)
|
||
|
|
|
||
|
|
// NewDatabase creates a SQLite database object from a file name
|
||
|
|
func NewDatabase(filename string) (*gorm.DB, error) {
|
||
|
|
return gorm.Open(sqlite.Open(filename), &gorm.Config{})
|
||
|
|
}
|
||
|
|
|
||
|
|
// CreateDatabaseObjects creates database relations
|
||
|
|
func CreateDatabaseObjects(db *gorm.DB) error {
|
||
|
|
if err := db.AutoMigrate(&Miner{}); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
if err := db.AutoMigrate(&Pool{}); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|