97 lines
2.1 KiB
Go
97 lines
2.1 KiB
Go
package respondd
|
|
|
|
/**
|
|
* This database type is for injecting into another yanic instance.
|
|
*/
|
|
import (
|
|
"bufio"
|
|
"compress/flate"
|
|
"encoding/json"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/bdlm/log"
|
|
|
|
"github.com/FreifunkBremen/yanic/data"
|
|
"github.com/FreifunkBremen/yanic/database"
|
|
"github.com/FreifunkBremen/yanic/runtime"
|
|
)
|
|
|
|
type Connection struct {
|
|
database.Connection
|
|
config Config
|
|
conn net.Conn
|
|
}
|
|
|
|
type Config map[string]interface{}
|
|
|
|
func (c Config) Type() string {
|
|
return c["type"].(string)
|
|
}
|
|
|
|
func (c Config) Address() string {
|
|
return c["address"].(string)
|
|
}
|
|
|
|
func init() {
|
|
database.RegisterAdapter("respondd", Connect)
|
|
}
|
|
|
|
func Connect(configuration map[string]interface{}) (database.Connection, error) {
|
|
config := Config(configuration)
|
|
|
|
conn, err := net.Dial(config.Type(), config.Address())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &Connection{conn: conn, config: config}, nil
|
|
}
|
|
|
|
func (conn *Connection) InsertNode(node *runtime.Node) {
|
|
res := &data.ResponseData{
|
|
Nodeinfo: node.Nodeinfo,
|
|
Statistics: node.Statistics,
|
|
Neighbours: node.Neighbours,
|
|
}
|
|
|
|
writer := bufio.NewWriterSize(conn.conn, 8192)
|
|
|
|
flater, err := flate.NewWriter(writer, flate.BestCompression)
|
|
if err != nil {
|
|
log.Errorf("[database-yanic] could not create flater: %s", err)
|
|
return
|
|
}
|
|
defer flater.Close()
|
|
err = json.NewEncoder(flater).Encode(res)
|
|
if err != nil {
|
|
nodeid := "unknown"
|
|
if node.Nodeinfo != nil && node.Nodeinfo.NodeID != "" {
|
|
nodeid = node.Nodeinfo.NodeID
|
|
}
|
|
log.WithField("node_id", nodeid).Errorf("[database-yanic] could not encode node: %s", err)
|
|
return
|
|
}
|
|
err = flater.Flush()
|
|
if err != nil {
|
|
log.Errorf("[database-yanic] could not compress: %s", err)
|
|
}
|
|
err = writer.Flush()
|
|
if err != nil {
|
|
log.Errorf("[database-yanic] could not send: %s", err)
|
|
}
|
|
}
|
|
|
|
func (conn *Connection) InsertLink(link *runtime.Link, time time.Time) {
|
|
}
|
|
|
|
func (conn *Connection) InsertGlobals(stats *runtime.GlobalStats, time time.Time, site string, domain string) {
|
|
}
|
|
|
|
func (conn *Connection) PruneNodes(deleteAfter time.Duration) {
|
|
}
|
|
|
|
func (conn *Connection) Close() {
|
|
conn.conn.Close()
|
|
}
|