101 lines
1.9 KiB
Go
101 lines
1.9 KiB
Go
package gobot
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"log"
|
|
"reflect"
|
|
)
|
|
|
|
type Connection interface {
|
|
Connect() bool
|
|
Finalize() bool
|
|
port() string
|
|
name() string
|
|
setName(string)
|
|
params() map[string]interface{}
|
|
}
|
|
|
|
type JSONConnection struct {
|
|
Name string `json:"name"`
|
|
Port string `json:"port"`
|
|
Adaptor string `json:"adaptor"`
|
|
}
|
|
|
|
type connection struct {
|
|
Name string
|
|
Type string
|
|
Adaptor AdaptorInterface
|
|
Robot *Robot
|
|
}
|
|
|
|
type connections []*connection
|
|
|
|
// Start() starts all the connections.
|
|
func (c connections) Start() error {
|
|
var err error
|
|
log.Println("Starting connections...")
|
|
for _, connection := range c {
|
|
log.Println("Starting connection " + connection.Name + "...")
|
|
if connection.Connect() == false {
|
|
err = errors.New("Could not start connection")
|
|
break
|
|
}
|
|
}
|
|
return err
|
|
}
|
|
|
|
// Filanize() finalizes all the connections.
|
|
func (c connections) Finalize() {
|
|
for _, connection := range c {
|
|
connection.Finalize()
|
|
}
|
|
}
|
|
|
|
func NewConnection(adaptor AdaptorInterface, r *Robot) *connection {
|
|
if adaptor.name() == "" {
|
|
adaptor.setName(fmt.Sprintf("%X", Rand(int(^uint(0)>>1))))
|
|
}
|
|
t := reflect.ValueOf(adaptor).Type().String()
|
|
return &connection{
|
|
Type: t[1:len(t)],
|
|
Name: adaptor.name(),
|
|
Robot: r,
|
|
Adaptor: adaptor,
|
|
}
|
|
}
|
|
|
|
func (c *connection) Connect() bool {
|
|
log.Println("Connecting to " + c.Name + " on port " + c.port() + "...")
|
|
return c.Adaptor.Connect()
|
|
}
|
|
|
|
func (c *connection) Finalize() bool {
|
|
log.Println("Finalizing " + c.Name + "...")
|
|
return c.Adaptor.Finalize()
|
|
}
|
|
|
|
func (c *connection) ToJSON() *JSONConnection {
|
|
return &JSONConnection{
|
|
Name: c.Name,
|
|
Port: c.port(),
|
|
Adaptor: c.Type,
|
|
}
|
|
}
|
|
|
|
func (c *connection) port() string {
|
|
return c.Adaptor.port()
|
|
}
|
|
|
|
func (c *connection) name() string {
|
|
return c.Name
|
|
}
|
|
|
|
func (c *connection) setName(s string) {
|
|
c.Name = s
|
|
}
|
|
|
|
func (c *connection) params() map[string]interface{} {
|
|
return c.Adaptor.params()
|
|
}
|