1
0
Fork 0
burlesque/storage.go

50 lines
780 B
Go
Raw Normal View History

2014-07-10 12:19:39 +00:00
package main
import (
2014-07-12 10:42:26 +00:00
"strconv"
"strings"
2014-07-31 08:23:41 +00:00
"bitbucket.org/ww/cabinet"
2014-07-10 12:19:39 +00:00
)
type (
2014-09-09 08:20:40 +00:00
message []byte
key []byte
2014-07-10 12:19:39 +00:00
)
var (
2014-07-16 17:50:21 +00:00
storage = cabinet.New()
2014-07-10 12:19:39 +00:00
)
2014-09-09 08:20:40 +00:00
func newKey(queue string, index uint) key {
2014-07-12 10:42:26 +00:00
istr := strconv.FormatUint(uint64(index), 10)
2014-09-09 08:20:40 +00:00
k := strings.Join([]string{queue, istr}, "_")
return key(k)
2014-07-12 10:42:26 +00:00
}
2014-09-09 08:20:40 +00:00
func setupStorage() {
err := storage.Open(config.storage, cabinet.KCOWRITER|cabinet.KCOCREATE)
2014-07-10 12:19:39 +00:00
if err != nil {
2014-09-09 08:20:40 +00:00
alert(err, "Failed to open database '%s'", config.storage)
2014-07-10 12:19:39 +00:00
}
}
2014-09-09 08:20:40 +00:00
func closeStorage() {
2014-07-16 17:50:21 +00:00
var err error
2014-07-10 12:19:39 +00:00
2014-07-16 17:50:21 +00:00
err = storage.Sync(true)
if err != nil {
2014-09-09 08:20:40 +00:00
alert(err, "Failed to sync storage (hard)")
2014-07-16 17:50:21 +00:00
} else {
2014-09-09 08:20:40 +00:00
log("Storage synchronized")
2014-07-16 17:50:21 +00:00
}
2014-07-12 10:42:04 +00:00
2014-07-16 17:50:21 +00:00
err = storage.Close()
if err != nil {
2014-09-09 08:20:40 +00:00
alert(err, "Failed to close storage")
2014-07-16 17:50:21 +00:00
} else {
2014-09-09 08:20:40 +00:00
log("Storage closed")
2014-07-10 12:19:39 +00:00
}
}