83 lines
1.8 KiB
Go
83 lines
1.8 KiB
Go
|
package socket
|
||
|
|
|
||
|
|
/**
|
||
|
|
* This database type is just for,
|
||
|
|
* - debugging without a influxconn
|
||
|
|
* - example for other developers for new databases
|
||
|
|
*/
|
||
|
|
import (
|
||
|
|
"log"
|
||
|
|
"net"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/FreifunkBremen/yanic/database"
|
||
|
|
"github.com/FreifunkBremen/yanic/runtime"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Connection struct {
|
||
|
|
database.Connection
|
||
|
|
config Config
|
||
|
|
listener net.Listener
|
||
|
|
clients map[net.Addr]net.Conn
|
||
|
|
}
|
||
|
|
|
||
|
|
type Config map[string]interface{}
|
||
|
|
|
||
|
|
func (c Config) Enable() bool {
|
||
|
|
return c["enable"].(bool)
|
||
|
|
}
|
||
|
|
func (c Config) Type() string {
|
||
|
|
return c["type"].(string)
|
||
|
|
}
|
||
|
|
func (c Config) Address() string {
|
||
|
|
return c["address"].(string)
|
||
|
|
}
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
database.RegisterAdapter("socket", Connect)
|
||
|
|
}
|
||
|
|
|
||
|
|
func Connect(configuration interface{}) (database.Connection, error) {
|
||
|
|
var config Config
|
||
|
|
config = configuration.(map[string]interface{})
|
||
|
|
if !config.Enable() {
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
ln, err := net.Listen(config.Type(), config.Address())
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
conn := &Connection{config: config, listener: ln, clients: make(map[net.Addr]net.Conn)}
|
||
|
|
go conn.handleSocketConnection(ln)
|
||
|
|
|
||
|
|
log.Println("[socket-database] listen on: ", ln.Addr())
|
||
|
|
|
||
|
|
return conn, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (conn *Connection) InsertNode(node *runtime.Node) {
|
||
|
|
conn.sendJSON(EventMessage{Event: "insert_node", Body: node})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (conn *Connection) InsertGlobals(stats *runtime.GlobalStats, time time.Time) {
|
||
|
|
conn.sendJSON(EventMessage{Event: "insert_globals", Body: stats})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (conn *Connection) PruneNodes(deleteAfter time.Duration) {
|
||
|
|
conn.sendJSON(EventMessage{Event: "prune_nodes"})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (conn *Connection) Close() {
|
||
|
|
for _, c := range conn.clients {
|
||
|
|
err := c.Close()
|
||
|
|
if err != nil {
|
||
|
|
log.Println("[socket-database] client was not able to close:", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
err := conn.listener.Close()
|
||
|
|
if err != nil {
|
||
|
|
log.Println("[socket-database] server was not able to close:", err)
|
||
|
|
}
|
||
|
|
}
|