server-go/socket/client/connector.go

package client

import (
	"sync/atomic"

	"git.loafle.net/commons/server-go/socket"
)

type OnDisconnectedFunc func(connector Connector)

type Connector interface {
	socket.ClientConnHandler
	socket.ReadWriteHandler

	Connect() (readChan <-chan socket.SocketMessage, writeChan chan<- socket.SocketMessage, err error)
	Disconnect() error

	GetName() string
	GetOnDisconnected() OnDisconnectedFunc
	SetOnDisconnected(fnc OnDisconnectedFunc)

	Clone() Connector
	Validate() error
}

type Connectors struct {
	socket.ClientConnHandlers
	socket.ReadWriteHandlers

	Name           string             `json:"name,omitempty"`
	OnDisconnected OnDisconnectedFunc `json:"-"`

	validated atomic.Value
}

func (c *Connectors) GetName() string {
	return c.Name
}

func (c *Connectors) GetOnDisconnected() OnDisconnectedFunc {
	return c.OnDisconnected
}

func (c *Connectors) SetOnDisconnected(fnc OnDisconnectedFunc) {
	c.OnDisconnected = fnc
}

func (c *Connectors) Clone() *Connectors {
	return &Connectors{
		Name:               c.Name,
		ClientConnHandlers: *c.ClientConnHandlers.Clone(),
		ReadWriteHandlers:  *c.ReadWriteHandlers.Clone(),
		validated:          c.validated,
	}
}

func (c *Connectors) Validate() error {
	if nil != c.validated.Load() {
		return nil
	}
	c.validated.Store(true)

	if err := c.ClientConnHandlers.Validate(); nil != err {
		return err
	}
	if err := c.ReadWriteHandlers.Validate(); nil != err {
		return err
	}

	if "" == c.Name {
		c.Name = "Connector"
	}

	return nil
}