yanic/database/socket/database.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 {
		c.Close()
	}
	conn.listener.Close()
}