mirror of https://github.com/k3s-io/k3s
103 lines
2.4 KiB
Go
103 lines
2.4 KiB
Go
// +build cgo
|
|
|
|
package sqlite
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/mattn/go-sqlite3"
|
|
"github.com/pkg/errors"
|
|
"github.com/rancher/kine/pkg/drivers/generic"
|
|
"github.com/rancher/kine/pkg/logstructured"
|
|
"github.com/rancher/kine/pkg/logstructured/sqllog"
|
|
"github.com/rancher/kine/pkg/server"
|
|
"github.com/sirupsen/logrus"
|
|
|
|
// sqlite db driver
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
var (
|
|
schema = []string{
|
|
`CREATE TABLE IF NOT EXISTS kine
|
|
(
|
|
id INTEGER primary key autoincrement,
|
|
name INTEGER,
|
|
created INTEGER,
|
|
deleted INTEGER,
|
|
create_revision INTEGER,
|
|
prev_revision INTEGER,
|
|
lease INTEGER,
|
|
value BLOB,
|
|
old_value BLOB
|
|
)`,
|
|
`CREATE INDEX IF NOT EXISTS kine_name_index ON kine (name)`,
|
|
`CREATE UNIQUE INDEX IF NOT EXISTS kine_name_prev_revision_uindex ON kine (name, prev_revision)`,
|
|
}
|
|
)
|
|
|
|
func New(ctx context.Context, dataSourceName string) (server.Backend, error) {
|
|
backend, _, err := NewVariant(ctx, "sqlite3", dataSourceName)
|
|
return backend, err
|
|
}
|
|
|
|
func NewVariant(ctx context.Context, driverName, dataSourceName string) (server.Backend, *generic.Generic, error) {
|
|
if dataSourceName == "" {
|
|
if err := os.MkdirAll("./db", 0700); err != nil {
|
|
return nil, nil, err
|
|
}
|
|
dataSourceName = "./db/state.db?_journal=WAL&cache=shared"
|
|
}
|
|
|
|
dialect, err := generic.Open(ctx, driverName, dataSourceName, "?", false)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
dialect.LastInsertID = true
|
|
dialect.TranslateErr = func(err error) error {
|
|
if err, ok := err.(sqlite3.Error); ok && err.ExtendedCode == sqlite3.ErrConstraintUnique {
|
|
return server.ErrKeyExists
|
|
}
|
|
return err
|
|
}
|
|
|
|
// this is the first SQL that will be executed on a new DB conn so
|
|
// loop on failure here because in the case of dqlite it could still be initializing
|
|
for i := 0; i < 300; i++ {
|
|
err = setup(dialect.DB)
|
|
if err == nil {
|
|
break
|
|
}
|
|
logrus.Errorf("failed to setup db: %v", err)
|
|
select {
|
|
case <-ctx.Done():
|
|
return nil, nil, ctx.Err()
|
|
case <-time.After(time.Second):
|
|
}
|
|
time.Sleep(time.Second)
|
|
}
|
|
if err != nil {
|
|
return nil, nil, errors.Wrap(err, "setup db")
|
|
}
|
|
//if err := setup(dialect.DB); err != nil {
|
|
// return nil, nil, errors.Wrap(err, "setup db")
|
|
//}
|
|
|
|
dialect.Migrate(context.Background())
|
|
return logstructured.New(sqllog.New(dialect)), dialect, nil
|
|
}
|
|
|
|
func setup(db *sql.DB) error {
|
|
for _, stmt := range schema {
|
|
_, err := db.Exec(stmt)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|