87 lines
1.4 KiB
Go
87 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"github.com/google/uuid"
|
|
"go.mills.io/bitcask/v2"
|
|
"log"
|
|
)
|
|
|
|
type Database struct {
|
|
connection *bitcask.Bitcask
|
|
Handle *bitcask.Collection
|
|
}
|
|
|
|
func NewDatabase(path string) *Database {
|
|
db, err := bitcask.Open(path)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
return &Database{
|
|
connection: db,
|
|
Handle: db.Collection("noodles"),
|
|
}
|
|
}
|
|
|
|
func (db *Database) MakeID() string {
|
|
id, err := uuid.NewUUID()
|
|
if err != nil {
|
|
log.Print(err)
|
|
return ""
|
|
}
|
|
return id.String()
|
|
}
|
|
|
|
func (db *Database) GetAll() []Noodle {
|
|
var data []Noodle
|
|
err := db.Handle.List(&data)
|
|
if err != nil {
|
|
log.Print(err)
|
|
return nil
|
|
}
|
|
return data
|
|
}
|
|
|
|
func (db *Database) GetAllGeneric() []interface{} {
|
|
var data []interface{}
|
|
err := db.Handle.List(&data)
|
|
if err != nil {
|
|
log.Print(err)
|
|
return nil
|
|
}
|
|
return data
|
|
}
|
|
|
|
func (db *Database) Get(id string) Noodle {
|
|
var noodle Noodle
|
|
log.Printf("Looking up noodle key='%s'", id)
|
|
log.Printf("key='%s' exists=%b", id, db.Handle.Has(id))
|
|
err := db.Handle.Get(id, &noodle)
|
|
if err != nil {
|
|
log.Print(err)
|
|
return Noodle{}
|
|
}
|
|
return noodle
|
|
}
|
|
|
|
func (db *Database) Add(noodle Noodle) error {
|
|
err := db.Handle.Add(noodle.Id, noodle)
|
|
if err != nil {
|
|
log.Print(err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (db *Database) Delete(id string) error {
|
|
err := db.Handle.Delete(id)
|
|
if err != nil {
|
|
log.Print(err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (db *Database) Close() error {
|
|
return db.connection.Close()
|
|
}
|