bolt.go

package sse

import (
	"errors"
	"github.com/boltdb/bolt"
)

type BoltDB struct {
	Conn *bolt.DB
}

func BoltDBOpen() (BoltDB, error) {
	db, err := bolt.Open("my.db", 0600, nil)
	if err != nil {
		return BoltDB{}, err
	}
	conn := BoltDB{Conn: db}
	return conn, nil
}

func (db BoltDB) Init() error {
	err := db.Conn.Update(func(tx *bolt.Tx) error {
		_, err := tx.CreateBucketIfNotExists([]byte(DOCUMENTS))
		if err != nil {
			return err
		}
		_, err = tx.CreateBucketIfNotExists([]byte(COUNTS))
		if err != nil {
			return err
		}
		_, err = tx.CreateBucketIfNotExists([]byte(INDEX))
		if err != nil {
			return err
		}
		return nil
	})

	return err
}

func (db BoltDB) Get(bucket string, id []byte) ([]byte, error) {
	var value []byte
	// Use View() to enforce read-only access to BoltDB.
	err := db.Conn.View(func(tx *bolt.Tx) error {
		b := tx.Bucket([]byte(bucket))
		if b == nil {
			return errors.New("BoltDB: bucket not found")
		}
		temp := b.Get(id)
		value = make([]byte, len(temp))
		copy(value, temp)
		return nil
	})

	return value, err
}

func (db BoltDB) Put(bucket string, id, value []byte) error {
	// Use Update() to enforce read-write access to BoltDB.
	err := db.Conn.Update(func(tx *bolt.Tx) error {
		// TODO: Put logic
		b := tx.Bucket([]byte(bucket))
		if b == nil {
			return errors.New("BoltDB: bucket not found")
		}
		err := b.Put(id, value)
		return err
	})

	return err
}

func (db BoltDB) Delete(bucket string, id []byte) error {
	// Use Update() to enforce read-write access to BoltDB.
	err := db.Conn.Update(func(tx *bolt.Tx) error {
		b := tx.Bucket([]byte(bucket))
		if b == nil {
			return errors.New("BoltDB: bucket not found")
		}
		err := b.Delete(id)
		return err
	})

	return err
}

func (db BoltDB) Close() {
	db.Conn.Close()
}