BlotDB数据库例子

        一直想在Go系中找个类似PHP+Sqlite的组合，查了下，发现了BlotDB数据库正合用。 这个库是Go写的,且在GitHub

上star数很高，看来我是孤陋寡闻了 。

      研究了下，它是个Key/Value数据库，会在磁盘上生成一个db文件，支持事务处理,支持只读打开..... 不过它不

支持SQL语句，需通过相关函数和代码来手工实现。

     弄了个例子:

/*
BlotDB 练习

Author: xcl
Date: 2-15-11-25
*/
package main

import (
	"bytes"
	"fmt"
	"github.com/boltdb/bolt"
	"log"
)

func main() {
	arrBucket := []string{"a", "b"}
	mgr, _ := NewBoltManager("my.db", arrBucket)
	mgr.Add("a", []byte("11"))
	mgr.Add("a", []byte("22"))
	id33, _ := mgr.Add("a", []byte("33"))
	mgr.Add("a", []byte("22"))

	mgr.Add("b", []byte("11"))
	mgr.Add("b", []byte("22"))

	log.Println("Select a=>>>>>>>>>>>>>>>>>>>")
	mgr.Select("a")
	log.Println("Select b=>>>>>>>>>>>>>>>>>>>")
	mgr.Select("b")

	log.Println("RemoveID a=>>>>>>>>>>>>>>>>>>>")
	mgr.RemoveID("a", []byte(fmt.Sprintf("%d", id33)))

	log.Println("RemoveValTransaction a=>>>>>>>>>>>>>>>>>>>")
	mgr.RemoveValTransaction("a", []byte("22"))
	log.Println("Select a=>>>>>>>>>>>>>>>>>>>")
	mgr.Select("a")

	//清理Bucket
	for _, v := range arrBucket {
		mgr.RemoveBucket(v)
	}
	mgr.Close()

}

//BlotDB的管理类
type BoltManager struct {
	db *bolt.DB "数据库类"
}

//创建库管理,并生成Bucket
func NewBoltManager(dbPath string, bucket []string) (*BoltManager, error) {
	//  bolt.Open("my.db", 0600, &bolt.Options{Timeout: 1 * time.Second,ReadOnly: true})
	db, err := bolt.Open(dbPath, 0644, nil)
	if err != nil {
		return nil, err
	}
	err = db.Update(func(tx *bolt.Tx) error {
		for _, v := range bucket {
			_, err := tx.CreateBucketIfNotExists([]byte(v))
			if err != nil {
				return err
			}
		}
		return nil
	})

	if err != nil {
		return nil, err
	}
	return &BoltManager{db}, nil
}

//关库数据库
func (m *BoltManager) Close() error {
	return m.db.Close()
}

//移除Bucket
func (m *BoltManager) RemoveBucket(bucketName string) (err error) {
	err = m.db.Update(func(tx *bolt.Tx) error {
		return tx.DeleteBucket([]byte(bucketName))
	})
	return err
}

//组Bucket增加值
func (m *BoltManager) Add(bucketName string, val []byte) (id uint64, err error) {

	err = m.db.Update(func(tx *bolt.Tx) error {
		b := tx.Bucket([]byte(bucketName))
		id, _ = b.NextSequence() //sequence uint64
		bBuf := fmt.Sprintf("%d", id)
		return b.Put([]byte(bBuf), val)
	})
	return
}

//遍历
func (m *BoltManager) Select(bucketName string) (err error) {
	m.db.View(func(tx *bolt.Tx) error {
		b := tx.Bucket([]byte(bucketName))
		b.ForEach(func(k, v []byte) error {
			log.Printf("key=%s, value=%s\n", string(k), v)
			return nil
		})
		return nil
	})
	return nil
}

//移除指定Bucket中指定ID
func (m *BoltManager) RemoveID(bucketName string, id []byte) error {
	err := m.db.Update(func(tx *bolt.Tx) error {
		b := tx.Bucket([]byte(bucketName))
		return b.Delete(id)
	})
	return err
}

//移除指定Bucket中指定Val
func (m *BoltManager) RemoveVal(bucketName string, val []byte) (err error) {
	var arrID []string
	arrID = make([]string, 1)
	err = m.db.View(func(tx *bolt.Tx) error {
		b := tx.Bucket([]byte(bucketName))
		c := b.Cursor()
		for k, v := c.First(); k != nil; k, v = c.Next() {
			log.Printf("key=%s, value=%s\n", k, string(v))
			if bytes.Compare(v, val) == 0 {
				arrID = append(arrID, string(k))
			}
		}
		return nil
	})

	err = m.db.Update(func(tx *bolt.Tx) error {
		b := tx.Bucket([]byte(bucketName))
		for _, v := range arrID {
			b.Delete([]byte(v))
			log.Println("Del k:", v)
		}
		return nil
	})

	return err
}

//查找指定值
func (m *BoltManager) SelectVal(bucketName string, val []byte) (
	arr []string,
	err error,
) {
	arr = make([]string, 0, 1)
	err = m.db.View(func(tx *bolt.Tx) error {
		c := tx.Bucket([]byte(bucketName)).Cursor()
		for k, v := c.First(); k != nil; k, v = c.Next() {
			if bytes.Compare(v, val) == 0 {
				arr = append(arr, string(k))
			}
		}
		return nil
	})
	return arr, err
}

//在事务中，移除指定Bucket中指定Val
func (m *BoltManager) RemoveValTransaction(bucketName string, val []byte) (
	count int,
	err error,
) {
	arrID, err1 := m.SelectVal(bucketName, val)
	if err1 != nil {
		return 0, err1
	}
	count = len(arrID)
	if count == 0 {
		return count, nil
	}

	tx, err1 := m.db.Begin(true)
	if err1 != nil {
		return count, err1
	}
	b := tx.Bucket([]byte(bucketName))
	for _, v := range arrID {
		if err = b.Delete([]byte(v)); err != nil {
			log.Printf("删除ID(%s)失败! 执行回滚. err:%s \r\n", v, err)
			tx.Rollback()
			return
		}
		log.Println("删除ID(", v, ")成功!")
	}
	err = tx.Commit()
	return
}

/*

2015/11/25  15:55    <DIR>          github.com
2015/11/25  21:46             4,992 main.go
2015/11/25  21:45            32,768 my.db

运行结果:
2015/11/25 21:45:51 Select a=>>>>>>>>>>>>>>>>>>>
2015/11/25 21:45:51 key=1, value=11
2015/11/25 21:45:51 key=2, value=22
2015/11/25 21:45:51 key=3, value=33
2015/11/25 21:45:51 key=4, value=22
2015/11/25 21:45:51 Select b=>>>>>>>>>>>>>>>>>>>
2015/11/25 21:45:51 key=1, value=11
2015/11/25 21:45:51 key=2, value=22
2015/11/25 21:45:51 RemoveID a=>>>>>>>>>>>>>>>>>>>
2015/11/25 21:45:51 RemoveValTransaction a=>>>>>>>>>>>>>>>>>>>
2015/11/25 21:45:51 删除ID( 2 )成功!
2015/11/25 21:45:51 删除ID( 4 )成功!
2015/11/25 21:45:51 Select a=>>>>>>>>>>>>>>>>>>>
2015/11/25 21:45:51 key=1, value=11
*/

BLOG: http://blog.csdn.net/xcl168