forked from facesea/banshee
/
db.go
101 lines (93 loc) · 2.12 KB
/
db.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
// Copyright 2015 Eleme Inc. All rights reserved.
// Package storage implements persistence storage based on leveldb.
package storage
import (
"fmt"
"github.com/eleme/banshee/storage/admindb"
"github.com/eleme/banshee/storage/indexdb"
"github.com/eleme/banshee/storage/metricdb"
"github.com/eleme/banshee/storage/statedb"
"github.com/eleme/banshee/util/log"
"os"
"path"
)
// DB file mode.
const filemode = 0755
// Child db filename.
const (
admindbFileName = "admin"
indexdbFileName = "index"
metricdbFileName = "metric"
statedbFileName = "state"
)
// Options is for db opening.
type Options struct {
// statedb
NumGrid int
GridLen int
}
// DB handles the storage on leveldb.
type DB struct {
// Child db
Admin *admindb.DB
Index *indexdb.DB
Metric *metricdb.DB
State *statedb.DB
}
// Open a DB by fileName and options.
func Open(fileName string, options *Options) (*DB, error) {
// Create if not exist
_, err := os.Stat(fileName)
if os.IsNotExist(err) {
log.Debug("create dir %s", fileName)
err := os.Mkdir(fileName, filemode)
if err != nil {
return nil, err
}
}
// Admindb.
db := new(DB)
db.Admin, err = admindb.Open(path.Join(fileName, admindbFileName))
if err != nil {
return nil, err
}
// Indexdb.
db.Index, err = indexdb.Open(path.Join(fileName, indexdbFileName))
if err != nil {
return nil, err
}
// Metricdb.
db.Metric, err = metricdb.Open(path.Join(fileName, metricdbFileName))
if err != nil {
return nil, err
}
name := fmt.Sprintf("%s-%dx%d", statedbFileName, options.NumGrid, options.GridLen)
opts := &statedb.Options{NumGrid: options.NumGrid, GridLen: options.GridLen}
// Statedb.
db.State, err = statedb.Open(path.Join(fileName, name), opts)
if err != nil {
return nil, err
}
log.Debug("storage is opened successfully")
return db, nil
}
// Close a DB.
func (db *DB) Close() error {
// Admindb.
if err := db.Admin.Close(); err != nil {
return err
}
// Indexdb.
if err := db.Index.Close(); err != nil {
return err
}
// Metricdb.
if err := db.Metric.Close(); err != nil {
return err
}
// Statedb.
if err := db.State.Close(); err != nil {
return err
}
return nil
}