From 9e011074bcc80e00e6deb5b914371718e5991ce2 Mon Sep 17 00:00:00 2001 From: Adam Mathes Date: Tue, 12 Jun 2018 14:58:02 -0700 Subject: sqlite3 support --- models/db.go | 25 ++++++++++++++----------- 1 file changed, 14 insertions(+), 11 deletions(-) (limited to 'models') diff --git a/models/db.go b/models/db.go index b8bc655..31d8bdd 100644 --- a/models/db.go +++ b/models/db.go @@ -4,21 +4,24 @@ Package neko/models implements behavior for the entities necessary for the subsc package models import ( - "database/sql" + "adammathes.com/neko/config" + "database/sql" _ "github.com/go-sql-driver/mysql" - "log" + _ "github.com/mattn/go-sqlite3" + "log" ) var DB *sql.DB -func InitDB(dataSourceName string) { - var err error - DB, err = sql.Open("mysql", dataSourceName) - if err != nil { - log.Panic(err) - } +func InitDB() { + var err error + // DB, err = sql.Open("mysql", dataSourceName) + DB, err = sql.Open(config.Config.DBDriver, config.Config.DBServer) + if err != nil { + log.Panic(err) + } - if err = DB.Ping(); err != nil { - log.Panic(err) - } + if err = DB.Ping(); err != nil { + log.Panic(err) + } } -- cgit v1.2.3