66 lines
1.1 KiB
Go
66 lines
1.1 KiB
Go
|
package tiledb
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"mapserver/coords"
|
||
|
|
|
||
|
|
"github.com/dgraph-io/badger"
|
||
|
|
)
|
||
|
|
|
||
|
|
func New(path string) (*TileDB, error) {
|
||
|
|
opts := badger.DefaultOptions
|
||
|
|
opts.Dir = path
|
||
|
|
opts.ValueDir = path
|
||
|
|
db, err := badger.Open(opts)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return &TileDB{
|
||
|
|
db: db,
|
||
|
|
}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
type TileDB struct {
|
||
|
|
db *badger.DB
|
||
|
|
}
|
||
|
|
|
||
|
|
func getKey(pos *coords.TileCoords) []byte {
|
||
|
|
return []byte(fmt.Sprintf("%d/%d/%d/%d", pos.X, pos.Y, pos.Zoom, pos.LayerId))
|
||
|
|
}
|
||
|
|
|
||
|
|
func (this *TileDB) GetTile(pos *coords.TileCoords) ([]byte, error) {
|
||
|
|
var tile []byte
|
||
|
|
err := this.db.View(func(txn *badger.Txn) error {
|
||
|
|
item, err := txn.Get(getKey(pos))
|
||
|
|
if item != nil {
|
||
|
|
tile, err = item.ValueCopy(nil)
|
||
|
|
}
|
||
|
|
return err
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
return tile, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (this *TileDB) SetTile(pos *coords.TileCoords, tile []byte) error {
|
||
|
|
err := this.db.Update(func(txn *badger.Txn) error {
|
||
|
|
err := txn.Set(getKey(pos), tile)
|
||
|
|
return err
|
||
|
|
})
|
||
|
|
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (this *TileDB) RemoveTile(pos *coords.TileCoords) error {
|
||
|
|
err := this.db.Update(func(txn *badger.Txn) error {
|
||
|
|
err := txn.Delete(getKey(pos))
|
||
|
|
return err
|
||
|
|
})
|
||
|
|
|
||
|
|
return err
|
||
|
|
}
|