added db to docker-compose
This commit is contained in:
+16
-10
@@ -2,12 +2,15 @@ package main
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"log"
|
||||
_ "embed"
|
||||
"time"
|
||||
|
||||
_ "github.com/mattn/go-sqlite3"
|
||||
)
|
||||
|
||||
//go:embed seed.sql
|
||||
var seedSQL string
|
||||
|
||||
// logs level=('debug', 'info', 'warning', 'error')
|
||||
type logs struct {
|
||||
level string
|
||||
@@ -20,22 +23,25 @@ type dbStruct struct {
|
||||
db *sql.DB
|
||||
}
|
||||
|
||||
// Seed runs the schema in seed.sql. Idempotent via CREATE TABLE IF NOT EXISTS,
|
||||
// so it's safe to call on every startup.
|
||||
func (app *dbStruct) Seed() error {
|
||||
_, err := app.db.Exec(seedSQL)
|
||||
return err
|
||||
}
|
||||
|
||||
// InsertLog() : database method, only inserts level + traceback
|
||||
func (app *dbStruct) InsertLog(lg logs) {
|
||||
func (app *dbStruct) InsertLog(lg logs) error {
|
||||
query := `INSERT INTO logs (level, traceback) VALUES (?, ?)`
|
||||
|
||||
_, err := app.db.Exec(query, lg.level, lg.traceback)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
// InsertLog() : database method, logs level + Ip
|
||||
func (app *dbStruct) LogIp(lg logs) {
|
||||
// LogIp() : database method, logs level + Ip
|
||||
func (app *dbStruct) LogIp(lg logs) error {
|
||||
query := `INSERT INTO logs (level, ip) VALUES (?, ?)`
|
||||
|
||||
_, err := app.db.Exec(query, lg.level, lg.ip)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user