77 lines
1.7 KiB
Go
77 lines
1.7 KiB
Go
package all
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/bdlm/log"
|
|
|
|
"github.com/FreifunkBremen/yanic/database"
|
|
"github.com/FreifunkBremen/yanic/runtime"
|
|
)
|
|
|
|
type Connection struct {
|
|
database.Connection
|
|
list []database.Connection
|
|
}
|
|
|
|
func Connect(allConnection map[string]interface{}) (database.Connection, error) {
|
|
var list []database.Connection
|
|
for dbType, conn := range database.Adapters {
|
|
configForType := allConnection[dbType]
|
|
if configForType == nil {
|
|
log.WithField("database", dbType).Infof("no configuration found")
|
|
continue
|
|
}
|
|
dbConfigs, ok := configForType.([]map[string]interface{})
|
|
if !ok {
|
|
return nil, fmt.Errorf("the database type '%s' has the wrong format: read format %T should be an []map[string]interface{}", dbType, configForType)
|
|
}
|
|
|
|
for _, config := range dbConfigs {
|
|
if c, ok := config["enable"].(bool); ok && !c {
|
|
continue
|
|
}
|
|
connected, err := conn(config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if connected == nil {
|
|
continue
|
|
}
|
|
list = append(list, connected)
|
|
}
|
|
}
|
|
return &Connection{list: list}, nil
|
|
}
|
|
|
|
func (conn *Connection) InsertNode(node *runtime.Node) {
|
|
for _, item := range conn.list {
|
|
item.InsertNode(node)
|
|
}
|
|
}
|
|
|
|
func (conn *Connection) InsertLink(link *runtime.Link, time time.Time) {
|
|
for _, item := range conn.list {
|
|
item.InsertLink(link, time)
|
|
}
|
|
}
|
|
|
|
func (conn *Connection) InsertGlobals(stats *runtime.GlobalStats, time time.Time, site string, domain string) {
|
|
for _, item := range conn.list {
|
|
item.InsertGlobals(stats, time, site, domain)
|
|
}
|
|
}
|
|
|
|
func (conn *Connection) PruneNodes(deleteAfter time.Duration) {
|
|
for _, item := range conn.list {
|
|
item.PruneNodes(deleteAfter)
|
|
}
|
|
}
|
|
|
|
func (conn *Connection) Close() {
|
|
for _, item := range conn.list {
|
|
item.Close()
|
|
}
|
|
}
|