move to utils package
This commit is contained in:
@@ -0,0 +1,45 @@
|
||||
package utils
|
||||
|
||||
import (
|
||||
"github.com/tarm/serial"
|
||||
"io"
|
||||
"log"
|
||||
"time"
|
||||
)
|
||||
|
||||
var(
|
||||
device string
|
||||
baudrate int
|
||||
bytesize byte
|
||||
port *serial.Port
|
||||
portOpened bool
|
||||
)
|
||||
|
||||
func OpenPort(device string, baudrate int, bytesize byte) error {
|
||||
log.Println("opening port device:", device, ", baudrate=", baudrate, ", bytesize=",bytesize)
|
||||
config := &serial.Config{Name: device, Baud: baudrate, Size: bytesize, ReadTimeout: time.Second * 1}
|
||||
var err error
|
||||
port, err = serial.OpenPort(config)
|
||||
if err!=nil {
|
||||
log.Fatalln("error in open serial port ", err)
|
||||
return err
|
||||
}
|
||||
log.Println("opened port ", port)
|
||||
portOpened = true
|
||||
return nil
|
||||
}
|
||||
|
||||
func ReadPort() ([]byte, int){
|
||||
buffer := make([]byte, 256)
|
||||
count,err := port.Read(buffer)
|
||||
if err != nil && err != io.EOF {
|
||||
log.Println("error in read port ", err)
|
||||
//TODO: 3 time fail try to
|
||||
}
|
||||
return buffer, count
|
||||
}
|
||||
|
||||
func IsPortOpen() bool {
|
||||
return portOpened
|
||||
}
|
||||
|
||||
@@ -0,0 +1,48 @@
|
||||
package utils
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
_ "github.com/mattn/go-sqlite3"
|
||||
"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)
|
||||
}
|
||||
log.Println("database ", db, " opened")
|
||||
return db, nil
|
||||
}
|
||||
|
||||
func InitDb(file string, init bool) error {
|
||||
var dbErr error
|
||||
db, dbErr = getDb(file, init)
|
||||
return dbErr
|
||||
}
|
||||
Reference in New Issue
Block a user