41 lines
904 B
Go
41 lines
904 B
Go
|
package database
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"log"
|
||
|
|
"os"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
DbDriver = "sqlite3"
|
||
|
|
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 (
|
||
|
|
db *sql.DB
|
||
|
|
)
|
||
|
|
|
||
|
|
func getDb(file string, init bool) (*sql.DB, error) {
|
||
|
|
log.Println("init database ", init)
|
||
|
|
if init {
|
||
|
|
log.Println("remove database file", file)
|
||
|
|
os.Remove(file)
|
||
|
|
}
|
||
|
|
db, err := sql.Open(DbDriver, file)
|
||
|
|
defer db.Close()
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal("error in open database ", err)
|
||
|
|
}
|
||
|
|
_, err = db.Exec(TableCreate)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal("error in create telegram table", err)
|
||
|
|
}
|
||
|
|
return db, nil
|
||
|
|
}
|