diff options
Diffstat (limited to 'internal/storage/sqlite/writer.go')
| -rw-r--r-- | internal/storage/sqlite/writer.go | 11 |
1 files changed, 3 insertions, 8 deletions
diff --git a/internal/storage/sqlite/writer.go b/internal/storage/sqlite/writer.go index c35494d..ec0d6d0 100644 --- a/internal/storage/sqlite/writer.go +++ b/internal/storage/sqlite/writer.go @@ -32,7 +32,7 @@ type Options struct { Compress bool } -func openDB(dbPath string) (*sql.DB, error) { +func OpenDB(dbPath string) (*sql.DB, error) { return sql.Open( "sqlite", fmt.Sprintf( @@ -44,14 +44,9 @@ func openDB(dbPath string) (*sql.DB, error) { ) } -func NewWriter(dbPath string, logger *log.Logger, opts *Options) (*Writer, error) { - db, err := openDB(dbPath) - if err != nil { - return nil, errors.WithMessage(err, "opening sqlite database") - } - +func NewWriter(db *sql.DB, logger *log.Logger, opts *Options) (*Writer, error) { // WIP: only memory database for now - _, err = db.Exec(` + _, err := db.Exec(` CREATE TABLE IF NOT EXISTS url ( id INTEGER PRIMARY KEY, path TEXT NOT NULL |