mirror of https://github.com/VinGarcia/ksql.git
40 lines
901 B
Go
40 lines
901 B
Go
package ksqlite3
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
|
|
"github.com/vingarcia/ksql"
|
|
"github.com/vingarcia/ksql/sqldialect"
|
|
|
|
// This is imported here so the user don't
|
|
// have to worry about it when he uses it.
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
// NewFromSQLDB builds a ksql.DB from a *sql.DB instance
|
|
func NewFromSQLDB(db *sql.DB) (ksql.DB, error) {
|
|
return ksql.NewWithAdapter(NewSQLAdapter(db), sqldialect.Sqlite3Dialect{})
|
|
}
|
|
|
|
// New instantiates a new KSQL client using the "sqlite3" driver
|
|
func New(
|
|
_ context.Context,
|
|
connectionString string,
|
|
config ksql.Config,
|
|
) (ksql.DB, error) {
|
|
config.SetDefaultValues()
|
|
|
|
db, err := sql.Open("sqlite3", connectionString)
|
|
if err != nil {
|
|
return ksql.DB{}, err
|
|
}
|
|
if err = db.Ping(); err != nil {
|
|
return ksql.DB{}, err
|
|
}
|
|
|
|
db.SetMaxOpenConns(config.MaxOpenConns)
|
|
|
|
return ksql.NewWithAdapter(NewSQLAdapter(db), sqldialect.Sqlite3Dialect{})
|
|
}
|