85 lines
1.6 KiB
Go
85 lines
1.6 KiB
Go
package mdb
|
|
|
|
import (
|
|
"net"
|
|
"os"
|
|
"sync"
|
|
|
|
"git.crumpington.com/private/mdb/kvstore"
|
|
)
|
|
|
|
type Database struct {
|
|
root string
|
|
kv *kvstore.KV
|
|
|
|
collections map[string]dbCollection
|
|
}
|
|
|
|
func NewPrimary(root string) *Database {
|
|
must(os.MkdirAll(root, 0700))
|
|
db := &Database{
|
|
root: root,
|
|
collections: map[string]dbCollection{},
|
|
}
|
|
db.kv = kvstore.NewPrimary(root)
|
|
return db
|
|
}
|
|
|
|
func NewSecondary(root string) *Database {
|
|
must(os.MkdirAll(root, 0700))
|
|
db := &Database{
|
|
root: root,
|
|
collections: map[string]dbCollection{},
|
|
}
|
|
db.kv = kvstore.NewSecondary(root, db.onStore, db.onDelete)
|
|
return db
|
|
}
|
|
|
|
func (db *Database) Start() {
|
|
wg := sync.WaitGroup{}
|
|
for _, c := range db.collections {
|
|
wg.Add(1)
|
|
go func(c dbCollection) {
|
|
defer wg.Done()
|
|
c.loadData()
|
|
}(c)
|
|
}
|
|
wg.Wait()
|
|
}
|
|
|
|
func (db *Database) WALStatus() (ws WALStatus) {
|
|
ws.MaxID = db.kv.WALMaxSeqNum()
|
|
ws.MaxAppliedID = db.kv.MaxSeqNum()
|
|
return
|
|
}
|
|
|
|
func (db *Database) Close() {
|
|
db.kv.Close()
|
|
}
|
|
|
|
// ----------------------------------------------------------------------------
|
|
|
|
func (db *Database) onStore(collection string, id uint64, data []byte) {
|
|
c, ok := db.collections[collection]
|
|
if ok {
|
|
c.onStore(collection, id, data)
|
|
}
|
|
}
|
|
|
|
func (db *Database) onDelete(collection string, id uint64) {
|
|
c, ok := db.collections[collection]
|
|
if ok {
|
|
c.onDelete(collection, id)
|
|
}
|
|
}
|
|
|
|
// ----------------------------------------------------------------------------
|
|
|
|
func (db *Database) SyncSend(conn net.Conn) {
|
|
db.kv.SyncSend(conn)
|
|
}
|
|
|
|
func (db *Database) SyncRecv(conn net.Conn) {
|
|
db.kv.SyncRecv(conn)
|
|
}
|