Influxdata
/
chronograf
mirror of https://github.com/influxdata/chronograf.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Runnable migration file -- missing an update

pull/3120/head
Luke Morris 2018-03-07 23:09:56 -08:00
parent 601250f1d5
commit e37066a109
3 changed files with 77 additions and 41 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
bolt/base.go
View File

@ -1,9 +1,13 @@
package bolt
import "github.com/boltdb/bolt"
import (
"fmt"
"github.com/boltdb/bolt"
)
// SchemaVersionBucket stores ids of completed migrations
const SchemaVersionBucket = []byte("SchemaVersions")
var SchemaVersionBucket = []byte("SchemaVersions")
// IsMigrationComplete checks for the presence of a particular migration id
func IsMigrationComplete(db bolt.DB, id string) (bool, error) {
@ -54,15 +58,16 @@ type Migration struct {
// Migrate runs one migration's Up() function, if it has not already been run
func (m Migration) Migrate(db bolt.DB) error {
complete, err := IsMigrationComplete(db, m.ID)
_, err := IsMigrationComplete(db, m.ID)
if err != nil {
return err
}
if complete {
return nil
}
// if complete {
// return nil
// }
client.logger.Info("Running migration (", m.ID, ")")
fmt.Println("Running a migration!")
// client.logger.Info("Running migration (", m.ID, ")")
if err = m.Up(db); err != nil {
return err

97
bolt/change_interval_to_duration.go
View File

@ -1,7 +1,6 @@
package bolt
import (
"context"
"fmt"
"log"
"strings"
@ -20,47 +19,41 @@ var changeIntervalToDuration = Migration{
Down: down,
}
func updateDashboard(board *Dashboard) {
for _, cell := range board.Cells {
for _, query := range cell.Queries {
query.Command = strings.Replace(query.Command, ":interval:", "time(:interval:)", -1)
}
}
}
var up = func(db bolt.DB) error {
// For each dashboard
err := db.Update(func(tx *bolt.Tx) error {
bucket := tx.Bucket(DashboardsBucket)
fmt.Println("We're in the transaction")
bucket := tx.Bucket(dashboardBucket)
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!")
fmt.Println(board)
// Migrate the dashboard
updateDashboard(board)
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(board)
fmt.Println("Updated the dashboard!")
// data, err = proto.Marshal(board)
// if err != nil {
// log.Fatal("marshaling error: ", err)
// }
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!")
// err = bucket.Put(id, data)
// if err != nil {
// log.Fatal("error updating dashboard: ", err)
// }
return nil
})
@ -68,6 +61,8 @@ var up = func(db bolt.DB) error {
if err != nil {
log.Fatal("error updating dashboards: ", err)
}
return nil
})
if err != nil {
@ -77,18 +72,54 @@ var up = func(db bolt.DB) error {
return nil
}
var down = func(ctx context.Context, client bolt.Client) error {
var down = func(db bolt.DB) error {
return nil
}
var DashboardsBucket = []byte("Dashoard")
var dashboardBucket = []byte("Dashoard")
type Dashboard struct {
Cells []*DashboardCell `protobuf:"bytes,3,rep,name=cells" json:"cells,omitempty"`
ID int64 `protobuf:"varint,1,opt,name=ID,proto3" json:"ID,omitempty"`
Name string `protobuf:"bytes,2,opt,name=Name,proto3" json:"Name,omitempty"`
Cells []*DashboardCell `protobuf:"bytes,3,rep,name=cells" json:"cells,omitempty"`
Templates string `protobuf:"bytes,4,rep,name=templates" json:"templates,omitempty"`
Organization string `protobuf:"bytes,5,opt,name=Organization,proto3" json:"Organization,omitempty"`
}
func (*Dashboard) ProtoMessage() {}
func (m *Dashboard) Reset() { *m = Dashboard{} }
func (m *Dashboard) String() string { return proto.CompactTextString(m) }
type DashboardCell struct {
X int32 `protobuf:"varint,1,opt,name=x,proto3" json:"x,omitempty"`
Y int32 `protobuf:"varint,2,opt,name=y,proto3" json:"y,omitempty"`
W int32 `protobuf:"varint,3,opt,name=w,proto3" json:"w,omitempty"`
H int32 `protobuf:"varint,4,opt,name=h,proto3" json:"h,omitempty"`
Queries []*Query `protobuf:"bytes,5,rep,name=queries" json:"queries,omitempty"`
Name string `protobuf:"bytes,6,opt,name=name,proto3" json:"name,omitempty"`
Type string `protobuf:"bytes,7,opt,name=type,proto3" json:"type,omitempty"`
ID string `protobuf:"bytes,8,opt,name=ID,proto3" json:"ID,omitempty"`
Axes string `protobuf:"bytes,9,rep,name=axes" json:"axes,omitempty" protobuf_key:"bytes,1,opt,name=key,proto3" protobuf_val:"bytes,2,opt,name=value"`
Colors string `protobuf:"bytes,10,rep,name=colors" json:"colors,omitempty"`
Legend string `protobuf:"bytes,11,opt,name=legend" json:"legend,omitempty"`
}
func (m *DashboardCell) Reset() { *m = DashboardCell{} }
func (m *DashboardCell) String() string { return proto.CompactTextString(m) }
func (*DashboardCell) ProtoMessage() {}
type Query struct {
Command string `protobuf:"bytes,1,opt,name=Command,proto3" json:"Command,omitempty"`
Command string `protobuf:"bytes,1,opt,name=Command,proto3" json:"Command,omitempty"`
DB string `protobuf:"bytes,2,opt,name=DB,proto3" json:"DB,omitempty"`
RP string `protobuf:"bytes,3,opt,name=RP,proto3" json:"RP,omitempty"`
GroupBys string `protobuf:"bytes,4,rep,name=GroupBys" json:"GroupBys,omitempty"`
Wheres string `protobuf:"bytes,5,rep,name=Wheres" json:"Wheres,omitempty"`
Label string `protobuf:"bytes,6,opt,name=Label,proto3" json:"Label,omitempty"`
Range string `protobuf:"bytes,7,opt,name=Range" json:"Range,omitempty"`
Source string `protobuf:"bytes,8,opt,name=Source,proto3" json:"Source,omitempty"`
Shifts string `protobuf:"bytes,9,rep,name=Shifts" json:"Shifts,omitempty"`
}
func (m *Query) Reset() { *m = Query{} }
func (m *Query) String() string { return proto.CompactTextString(m) }
func (*Query) ProtoMessage() {}

2
bolt/client.go
View File

@ -198,7 +198,7 @@ func (c *Client) migrate(ctx context.Context, build chronograf.BuildInfo) error
return err
}
MigrateAll(c.db)
MigrateAll(*c.db)
}
return nil
}