init

database/sqlite.go

@@ -0,0 +1,41 @@
+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
+}