Initial event store draft

management/server/audit/event.go

@@ -0,0 +1,32 @@
+package audit
+
+import "time"
+
+const (
+	// DeviceEvent describes an event that happened of a device (e.g, connected/disconnected)
+	DeviceEvent EventType = "device"
+	// ManagementEvent describes an event that happened on a Management service (e.g., user added)
+	ManagementEvent EventType = "management"
+)
+
+type EventType string
+
+// EventSink provides an interface to store or stream events.
+type EventSink interface {
+	// Add an event to the sink.
+	Add(event *Event) error
+	// Close the sink flushing events if necessary
+	Close() error
+}
+
+// Event represents a network activity.
+type Event struct {
+	// Timestamp of the event
+	Timestamp time.Time
+	// Message of the event
+	Message string
+	// ID of the event (can be empty, meaning that it wasn't yet generated)
+	ID string
+	// Type of the event
+	Type EventType
+}

management/server/audit/sqlite.go

@@ -0,0 +1,66 @@
+package audit
+
+import (
+	"database/sql"
+	"fmt"
+	_ "github.com/mattn/go-sqlite3"
+)
+
+const (
+	SQLiteEventSinkDB = "events.db"
+	createTableQuery  = "CREATE TABLE IF NOT EXISTS events " +
+		"(id UNSIGNED BIG INT PRIMARY KEY AUTOINCREMENT, message TEXT, timestamp DATETIME);"
+)
+
+type SQLiteSink struct {
+	db *sql.DB
+}
+
+// NewSQLiteSink creates a new SQLiteSink with an event table if not exists.
+func NewSQLiteSink(dbPath string) (*SQLiteSink, error) {
+	db, err := sql.Open("sqlite3", dbPath)
+	if err != nil {
+		return nil, err
+	}
+
+	_, err = db.Exec(createTableQuery)
+	if err != nil {
+		return nil, err
+	}
+
+	return &SQLiteSink{db: db}, nil
+}
+
+// Add an event to the SQLite table
+func (sink *SQLiteSink) Add(event *Event) error {
+
+	_, err := sink.db.Exec(createTableQuery)
+	if err != nil {
+		return err
+	}
+
+	stmt, err := sink.db.Prepare("INSERT INTO events(message, timestamp) values(?, ?)")
+	if err != nil {
+		return err
+	}
+
+	result, err := stmt.Exec(event.Message, event.Timestamp)
+	if err != nil {
+		if err == sql.ErrNoRows {
+			// Handle the case of no rows returned.
+		}
+		return album, err
+	}
+
+	fmt.Println(result)
+
+	return nil
+}
+
+// Close the SQLiteSink
+func (sink *SQLiteSink) Close() error {
+	if sink.db != nil {
+		return sink.db.Close()
+	}
+	return nil
+}