-
Notifications
You must be signed in to change notification settings - Fork 69
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #236 from cole-miller/direct-sqlite3
Add a `nosqlite3` build tag to disable linking libsqlite3
- Loading branch information
Showing
13 changed files
with
268 additions
and
509 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,157 @@ | ||
| // +build !nosqlite3 | ||
|
|
||
| package client | ||
|
|
||
| import ( | ||
| "context" | ||
| "database/sql" | ||
| "fmt" | ||
| "strings" | ||
|
|
||
| "github.com/pkg/errors" | ||
| _ "github.com/mattn/go-sqlite3" // Go SQLite bindings | ||
| ) | ||
|
|
||
| // Option that can be used to tweak node store parameters. | ||
| type NodeStoreOption func(*nodeStoreOptions) | ||
|
|
||
| type nodeStoreOptions struct { | ||
| Where string | ||
| } | ||
|
|
||
| // DatabaseNodeStore persists a list addresses of dqlite nodes in a SQL table. | ||
| type DatabaseNodeStore struct { | ||
| db *sql.DB // Database handle to use. | ||
| schema string // Name of the schema holding the servers table. | ||
| table string // Name of the servers table. | ||
| column string // Column name in the servers table holding the server address. | ||
| where string // Optional WHERE filter | ||
| } | ||
|
|
||
| // DefaultNodeStore creates a new NodeStore using the given filename. | ||
| // | ||
| // If the filename ends with ".yaml" then the YamlNodeStore implementation will | ||
| // be used. Otherwise the SQLite-based one will be picked, with default names | ||
| // for the schema, table and column parameters. | ||
| // | ||
| // It also creates the table if it doesn't exist yet. | ||
| func DefaultNodeStore(filename string) (NodeStore, error) { | ||
| if strings.HasSuffix(filename, ".yaml") { | ||
| return NewYamlNodeStore(filename) | ||
| } | ||
|
|
||
| // Open the database. | ||
| db, err := sql.Open("sqlite3", filename) | ||
| if err != nil { | ||
| return nil, errors.Wrap(err, "failed to open database") | ||
| } | ||
|
|
||
| // Since we're setting SQLite single-thread mode, we need to have one | ||
| // connection at most. | ||
| db.SetMaxOpenConns(1) | ||
|
|
||
| // Create the servers table if it does not exist yet. | ||
| _, err = db.Exec("CREATE TABLE IF NOT EXISTS servers (address TEXT, UNIQUE(address))") | ||
| if err != nil { | ||
| return nil, errors.Wrap(err, "failed to create servers table") | ||
| } | ||
|
|
||
| store := NewNodeStore(db, "main", "servers", "address") | ||
|
|
||
| return store, nil | ||
| } | ||
|
|
||
| // NewNodeStore creates a new NodeStore. | ||
| func NewNodeStore(db *sql.DB, schema, table, column string, options ...NodeStoreOption) *DatabaseNodeStore { | ||
| o := &nodeStoreOptions{} | ||
| for _, option := range options { | ||
| option(o) | ||
| } | ||
|
|
||
| return &DatabaseNodeStore{ | ||
| db: db, | ||
| schema: schema, | ||
| table: table, | ||
| column: column, | ||
| where: o.Where, | ||
| } | ||
| } | ||
|
|
||
| // WithNodeStoreWhereClause configures the node store to append the given | ||
| // hard-coded where clause to the SELECT query used to fetch nodes. Only the | ||
| // clause itself must be given, without the "WHERE" prefix. | ||
| func WithNodeStoreWhereClause(where string) NodeStoreOption { | ||
| return func(options *nodeStoreOptions) { | ||
| options.Where = where | ||
| } | ||
| } | ||
|
|
||
| // Get the current servers. | ||
| func (d *DatabaseNodeStore) Get(ctx context.Context) ([]NodeInfo, error) { | ||
| tx, err := d.db.Begin() | ||
| if err != nil { | ||
| return nil, errors.Wrap(err, "failed to begin transaction") | ||
| } | ||
| defer tx.Rollback() | ||
|
|
||
| query := fmt.Sprintf("SELECT %s FROM %s.%s", d.column, d.schema, d.table) | ||
| if d.where != "" { | ||
| query += " WHERE " + d.where | ||
| } | ||
| rows, err := tx.QueryContext(ctx, query) | ||
| if err != nil { | ||
| return nil, errors.Wrap(err, "failed to query servers table") | ||
| } | ||
| defer rows.Close() | ||
|
|
||
| servers := make([]NodeInfo, 0) | ||
| for rows.Next() { | ||
| var address string | ||
| err := rows.Scan(&address) | ||
| if err != nil { | ||
| return nil, errors.Wrap(err, "failed to fetch server address") | ||
| } | ||
| servers = append(servers, NodeInfo{ID: 1, Address: address}) | ||
| } | ||
| if err := rows.Err(); err != nil { | ||
| return nil, errors.Wrap(err, "result set failure") | ||
| } | ||
|
|
||
| return servers, nil | ||
| } | ||
|
|
||
| // Set the servers addresses. | ||
| func (d *DatabaseNodeStore) Set(ctx context.Context, servers []NodeInfo) error { | ||
| tx, err := d.db.Begin() | ||
| if err != nil { | ||
| return errors.Wrap(err, "failed to begin transaction") | ||
| } | ||
|
|
||
| query := fmt.Sprintf("DELETE FROM %s.%s", d.schema, d.table) | ||
| if _, err := tx.ExecContext(ctx, query); err != nil { | ||
| tx.Rollback() | ||
| return errors.Wrap(err, "failed to delete existing servers rows") | ||
| } | ||
|
|
||
| query = fmt.Sprintf("INSERT INTO %s.%s(%s) VALUES (?)", d.schema, d.table, d.column) | ||
| stmt, err := tx.PrepareContext(ctx, query) | ||
| if err != nil { | ||
| tx.Rollback() | ||
| return errors.Wrap(err, "failed to prepare insert statement") | ||
| } | ||
| defer stmt.Close() | ||
|
|
||
| for _, server := range servers { | ||
| if _, err := stmt.ExecContext(ctx, server.Address); err != nil { | ||
| tx.Rollback() | ||
| return errors.Wrapf(err, "failed to insert server %s", server.Address) | ||
| } | ||
| } | ||
|
|
||
| if err := tx.Commit(); err != nil { | ||
| return errors.Wrap(err, "failed to commit transaction") | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,20 @@ | ||
| // +build nosqlite3 | ||
|
|
||
| package client | ||
|
|
||
| import ( | ||
| "strings" | ||
|
|
||
| "github.com/pkg/errors" | ||
| ) | ||
|
|
||
| // DefaultNodeStore creates a new NodeStore using the given filename. | ||
| // | ||
| // The filename must end with ".yaml". | ||
| func DefaultNodeStore(filename string) (NodeStore, error) { | ||
| if strings.HasSuffix(filename, ".yaml") { | ||
| return NewYamlNodeStore(filename) | ||
| } | ||
|
|
||
| return nil, errors.New("built without support for DatabaseNodeStore") | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.