You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
74 lines
1.7 KiB
74 lines
1.7 KiB
package tiedot
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/HouzuoGuo/tiedot/db"
|
|
)
|
|
|
|
func NewDatabase(dirname string) (*Database, error) {
|
|
data := Database{}
|
|
data.setdbdir(dirname)
|
|
if err := data.setDBhandle(); err != nil {
|
|
return &Database{}, err
|
|
}
|
|
return &data, nil
|
|
}
|
|
|
|
func (data *Database) setdbdir(dirname string) {
|
|
data.DBdir = dirname
|
|
}
|
|
|
|
func (data *Database) setDBhandle() error {
|
|
var err error
|
|
data.DBhandle, err = db.OpenDB(data.DBdir)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (data *Database) InitCollection(colls ...string) {
|
|
for _, str := range colls {
|
|
data.DBhandle.Create(str)
|
|
}
|
|
}
|
|
|
|
func (data *Database) GetAllCollections() []string {
|
|
var colls []string
|
|
for _, name := range data.DBhandle.AllCols() {
|
|
colls = append(colls, name)
|
|
}
|
|
return colls
|
|
}
|
|
|
|
func (data *Database) getCollectionHandle(coll string) *db.Col {
|
|
return data.DBhandle.Use(coll)
|
|
}
|
|
|
|
func (data *Database) InsertToCollection(coll string, model map[string]interface{}) (docid int, err error) {
|
|
collHandle := data.getCollectionHandle(coll)
|
|
return collHandle.Insert(model)
|
|
}
|
|
|
|
func (data *Database) InsertIndexToCollection(coll string, indexes []string) error {
|
|
collHandle := data.getCollectionHandle(coll)
|
|
return collHandle.Index(indexes)
|
|
}
|
|
|
|
func (data *Database) QueryDB(coll string, queryByte []byte) (map[int]struct{}, error) {
|
|
if len(queryByte) <= 0 {
|
|
return nil, fmt.Errorf("Invalid query string")
|
|
}
|
|
|
|
var query interface{}
|
|
json.Unmarshal(queryByte, &query)
|
|
|
|
queryResult := make(map[int]struct{}) // query result (document IDs) goes into map keys
|
|
err := db.EvalQuery(query, data.getCollectionHandle(coll), &queryResult)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return queryResult, nil
|
|
}
|
|
|