add test db

This commit is contained in:
w1ndyb0y
2020-12-21 11:36:08 +08:00
parent 787a949428
commit 8e417699c0
7 changed files with 31 additions and 3 deletions
+10 -2
View File
@@ -2,20 +2,21 @@ package database
import (
"database/sql"
_ "github.com/mattn/go-sqlite3"
"log"
"os"
)
const (
DbDriver = "sqlite3"
TableCreate =`'''
TableCreate =`
create table IF NOT EXISTS telegram (
[tele_id] INTEGER PRIMARY KEY AUTOINCREMENT,
[tele_recv_time] TIMESTAMP NOT NULL DEFAULT (datetime('now', 'localtime')),
[tele_processed] int(1) NOT NULL DEFAULT 0,
[tele_text] TEXT NOT NULL
)
'''
`
)
var (
@@ -37,5 +38,12 @@ func getDb(file string, init bool) (*sql.DB, error) {
if err != nil {
log.Fatal("error in create telegram table", err)
}
log.Println("database ", db, " opened")
return db, nil
}
func Init(file string, init bool) error {
var dbErr error
db, dbErr = getDb(file, init)
return dbErr
}