mirror of
https://github.com/netbirdio/netbird.git
synced 2026-04-20 17:26:40 +00:00
Rename audit to event
This commit is contained in:
35
management/server/event/event.go
Normal file
35
management/server/event/event.go
Normal file
@@ -0,0 +1,35 @@
|
||||
package event
|
||||
|
||||
import "time"
|
||||
|
||||
const (
|
||||
// DeviceEvent describes an event that happened of a device (e.g, connected/disconnected)
|
||||
DeviceEvent Type = "device"
|
||||
// ManagementEvent describes an event that happened on a Management service (e.g., user added)
|
||||
ManagementEvent Type = "management"
|
||||
)
|
||||
|
||||
type Type string
|
||||
|
||||
// Sink provides an interface to store or stream events.
|
||||
type Sink 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 event.
|
||||
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 uint64
|
||||
// Type of the event
|
||||
Type Type
|
||||
|
||||
//
|
||||
|
||||
}
|
||||
66
management/server/event/sqlite.go
Normal file
66
management/server/event/sqlite.go
Normal file
@@ -0,0 +1,66 @@
|
||||
package event
|
||||
|
||||
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
|
||||
}
|
||||
Reference in New Issue
Block a user