95 lines
2.0 KiB
Go
95 lines
2.0 KiB
Go
|
package bolt
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/boltdb/bolt"
|
||
|
|
"github.com/gogo/protobuf/proto"
|
||
|
|
)
|
||
|
|
|
||
|
|
// changeIntervalToDuration
|
||
|
|
// Before, we supported queries that included `GROUP BY :interval:`
|
||
|
|
// After, we only support queries with `GROUP BY time(:interval:)`
|
||
|
|
// thereby allowing non_negative_derivative(_____, :interval)
|
||
|
|
var changeIntervalToDuration = Migration{
|
||
|
|
ID: "59b0cda4fc7909ff84ee5c4f9cb4b655b6a26620",
|
||
|
|
Up: up,
|
||
|
|
Down: down,
|
||
|
|
}
|
||
|
|
|
||
|
|
var up = func(db bolt.DB) error {
|
||
|
|
err := db.Update(func(tx *bolt.Tx) error {
|
||
|
|
bucket := tx.Bucket(DashboardsBucket)
|
||
|
|
|
||
|
|
fmt.Println("We're in the transaction")
|
||
|
|
|
||
|
|
err := bucket.ForEach(func(id, data []byte) error {
|
||
|
|
board := &Dashboard{}
|
||
|
|
|
||
|
|
fmt.Println("New dashboard!")
|
||
|
|
|
||
|
|
err := proto.Unmarshal(data, board)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal("unmarshaling error: ", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
fmt.Println("Parsed the dashboard!")
|
||
|
|
|
||
|
|
for i, cell := range board.cells {
|
||
|
|
for i, query := range cell.queries {
|
||
|
|
query.Command = strings.Replace(query.Command, ":interval:", "time(:interval:)", -1)
|
||
|
|
cell.queries[i] = query
|
||
|
|
}
|
||
|
|
board.cells[i] = cell
|
||
|
|
}
|
||
|
|
|
||
|
|
fmt.Println("Updated the dashboard!")
|
||
|
|
|
||
|
|
data, err = proto.Marshal(board)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal("marshaling error: ", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
fmt.Println("marshaled the dashboard!")
|
||
|
|
|
||
|
|
err = bucket.Put(id, data)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal("error updating dashboard: ", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
fmt.Println("Updated the dashboard!")
|
||
|
|
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal("error updating dashboards: ", err)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
var down = func(ctx context.Context, client bolt.Client) error {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
var DashboardsBucket = []byte("Dashoard")
|
||
|
|
|
||
|
|
type Dashboard struct {
|
||
|
|
Cells []*DashboardCell `protobuf:"bytes,3,rep,name=cells" json:"cells,omitempty"`
|
||
|
|
}
|
||
|
|
type DashboardCell struct {
|
||
|
|
Queries []*Query `protobuf:"bytes,5,rep,name=queries" json:"queries,omitempty"`
|
||
|
|
}
|
||
|
|
type Query struct {
|
||
|
|
Command string `protobuf:"bytes,1,opt,name=Command,proto3" json:"Command,omitempty"`
|
||
|
|
}
|