add insert new telegram

utils/sqlite.go

@@ -5,11 +5,12 @@ import (
 	_ "github.com/mattn/go-sqlite3"
 	"log"
 	"os"
+	"time"
 )
 
 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')),
@@ -17,32 +18,68 @@ const (
                     [tele_text] TEXT NOT NULL    
                 )
 `
-)
+	TableInsert = "insert into telegram (tele_text) values (?)"
-var (
-	db *sql.DB
 )
 
-func getDb(file string, init bool) (*sql.DB, error) {
+var (
-	log.Println("init database ", init)
+	dbFile      string
-	if init {
+	initTable   bool
-		log.Println("remove database file", file)
+	initialized bool
-		os.Remove(file)
+	isWriting   bool
+)
+
+func getDb() *sql.DB {
+	log.Println("init database ", initTable)
+	if initTable && !initialized {
+		log.Println("remove database file", dbFile)
+		os.Remove(dbFile)
 	}
-	db, err := sql.Open(DbDriver, file)
+	db, err := sql.Open(DbDriver, dbFile)
-	defer db.Close()
 	if err != nil {
 		log.Fatal("error in open database ", err)
 	}
-	_, err = db.Exec(TableCreate)
+	return db
-	if err != nil {
-		log.Fatal("error in create telegram table", err)
-	}
-	log.Println("database ", db, " opened")
-	return db, nil
 }
 
 func InitDb(file string, init bool) error {
+	dbFile = file
+	initTable = init
 	var dbErr error
-	db, dbErr = getDb(file, init)
+	db := getDb()
-	return dbErr
+	defer db.Close()
-}
+	_, dbErr = db.Exec(TableCreate)
+	if dbErr != nil {
+		log.Fatal("error in create telegram table", dbErr)
+	}
+	log.Println("database ", db, " initialized")
+	initialized = true
+	return nil
+}
+
+func InsertTelegram(teleString string) {
+	count := 0
+	for isWriting && count < 5 {
+		log.Println("waiting for write ")
+		time.Sleep(3 * time.Second)
+		count++
+	}
+	if count >= 5 {
+		log.Fatalln("database locked")
+	}
+	isWriting = true
+	db := getDb()
+	defer db.Close()
+	stmt, err := db.Prepare(TableInsert)
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer stmt.Close()
+	result, err := stmt.Exec(teleString)
+	id, _ := result.LastInsertId()
+	if err != nil {
+		isWriting = false
+		log.Fatalln("error in write telegram")
+	}
+	log.Println("telegram insert ", id)
+	isWriting = false
+}