132 lines
2.9 KiB
Go
132 lines
2.9 KiB
Go
package gime
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/br0xen/boltease"
|
|
)
|
|
|
|
type GimeDB struct {
|
|
db *boltease.DB
|
|
archDb *boltease.DB
|
|
dbOpened int
|
|
dbArchOpened int
|
|
path string
|
|
filename, arch string
|
|
}
|
|
|
|
const (
|
|
TypeCurrent = "current"
|
|
TypeRecent = "recent"
|
|
TypeArchive = "archive"
|
|
)
|
|
|
|
// Load Database returns a database loaded off the files given
|
|
// name and archName located in path
|
|
func LoadDatabase(path, name, archName string) (*GimeDB, error) {
|
|
if path[len(path)-1] != '/' {
|
|
path = path + "/"
|
|
}
|
|
gdb := GimeDB{
|
|
path: path,
|
|
filename: name,
|
|
arch: archName,
|
|
}
|
|
gdb.initDatabase()
|
|
return &gdb, nil
|
|
}
|
|
|
|
func (gdb *GimeDB) openDatabase() error {
|
|
gdb.dbOpened += 1
|
|
if gdb.dbOpened == 1 {
|
|
var err error
|
|
gdb.db, err = boltease.Create(gdb.path+gdb.filename, 0600, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (gdb *GimeDB) closeDatabase() error {
|
|
gdb.dbOpened -= 1
|
|
if gdb.dbOpened == 0 {
|
|
return gdb.db.CloseDB()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (gdb *GimeDB) openArchiveDatabase() error {
|
|
gdb.dbArchOpened += 1
|
|
if gdb.dbArchOpened == 1 {
|
|
var err error
|
|
gdb.archDb, err = boltease.Create(gdb.path+gdb.arch, 0600, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (gdb *GimeDB) closeArchiveDatabase() error {
|
|
gdb.dbArchOpened -= 1
|
|
if gdb.dbArchOpened == 0 {
|
|
return gdb.archDb.CloseDB()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (gdb *GimeDB) initDatabase() error {
|
|
// Initialize the current/recent/settings database
|
|
var err error
|
|
if err = gdb.openDatabase(); err != nil {
|
|
return err
|
|
}
|
|
defer gdb.closeDatabase()
|
|
// Create the path to the bucket to store application settings
|
|
if err := gdb.db.MkBucketPath([]string{"settings"}); err != nil {
|
|
return err
|
|
}
|
|
// Create the path to the bucket to store the current time entry
|
|
if err := gdb.db.MkBucketPath([]string{TypeCurrent}); err != nil {
|
|
return err
|
|
}
|
|
// Create the path to the bucket to store recent time entries
|
|
if err := gdb.db.MkBucketPath([]string{TypeRecent}); err != nil {
|
|
return err
|
|
}
|
|
|
|
// Now initialize the Archive Database
|
|
if err = gdb.openArchiveDatabase(); err != nil {
|
|
return err
|
|
}
|
|
defer gdb.closeArchiveDatabase()
|
|
// Create the path to the bucket to store archived time entries
|
|
return gdb.archDb.MkBucketPath([]string{TypeArchive})
|
|
}
|
|
|
|
func (gdb *GimeDB) openDBType(tp string) (*boltease.DB, error) {
|
|
var err error
|
|
if tp == TypeCurrent || tp == TypeRecent {
|
|
if err = gdb.openDatabase(); err != nil {
|
|
return nil, err
|
|
}
|
|
return gdb.db, err
|
|
} else if tp == TypeArchive {
|
|
if err = gdb.openArchiveDatabase(); err != nil {
|
|
return nil, err
|
|
}
|
|
return gdb.archDb, err
|
|
}
|
|
return nil, errors.New("Invalid database type: " + tp)
|
|
}
|
|
|
|
func (gdb *GimeDB) closeDBType(tp string) error {
|
|
if tp == TypeCurrent || tp == TypeRecent {
|
|
return gdb.closeDatabase()
|
|
} else if tp == TypeArchive {
|
|
return gdb.closeArchiveDatabase()
|
|
}
|
|
return errors.New("Invalid database type: " + tp)
|
|
}
|