103 lines
2.5 KiB
Go
103 lines
2.5 KiB
Go
package reach
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"time"
|
|
|
|
"forge.cadoles.com/Pyxis/golang-socketio"
|
|
)
|
|
|
|
// Logger is a logger implementation for the ReachRS client
|
|
type Logger interface {
|
|
Printf(format string, args ...interface{})
|
|
}
|
|
|
|
// Options are ReachRS client configuration options
|
|
type Options struct {
|
|
Endpoint string
|
|
PingInterval time.Duration
|
|
PingTimeout time.Duration
|
|
ReceiveTimeout time.Duration
|
|
SendTimeout time.Duration
|
|
BufferSize int
|
|
Logger Logger
|
|
}
|
|
|
|
// OptionFunc is an option configuration for the ReachRS client
|
|
type OptionFunc func(opts *Options)
|
|
|
|
// WithEndpoint configures the client to target the given host and port
|
|
func WithEndpoint(host string, port int) OptionFunc {
|
|
return func(opts *Options) {
|
|
opts.Endpoint = gosocketio.GetUrl(host, port, false)
|
|
}
|
|
}
|
|
|
|
// WithPingInterval configures the client to use the given ping interval
|
|
func WithPingInterval(interval time.Duration) OptionFunc {
|
|
return func(opts *Options) {
|
|
opts.PingInterval = interval
|
|
}
|
|
}
|
|
|
|
// WithPingTimeout configures the client to use the given ping timeout
|
|
func WithPingTimeout(timeout time.Duration) OptionFunc {
|
|
return func(opts *Options) {
|
|
opts.PingTimeout = timeout
|
|
}
|
|
}
|
|
|
|
// WithReceiveTimeout configures the client to use the given receive timeout
|
|
func WithReceiveTimeout(timeout time.Duration) OptionFunc {
|
|
return func(opts *Options) {
|
|
opts.ReceiveTimeout = timeout
|
|
}
|
|
}
|
|
|
|
// WithSendTimeout configures the client to use the given send timeout
|
|
func WithSendTimeout(timeout time.Duration) OptionFunc {
|
|
return func(opts *Options) {
|
|
opts.SendTimeout = timeout
|
|
}
|
|
}
|
|
|
|
// WithBufferSize configures the client to use the given buffer size
|
|
func WithBufferSize(size int) OptionFunc {
|
|
return func(opts *Options) {
|
|
opts.BufferSize = size
|
|
}
|
|
}
|
|
|
|
// WithLogger configures the client to use the given logger
|
|
func WithLogger(logger Logger) OptionFunc {
|
|
return func(opts *Options) {
|
|
opts.Logger = logger
|
|
}
|
|
}
|
|
|
|
// WithStandardLogger configures the client to use the given standard logger
|
|
func WithStandardLogger() OptionFunc {
|
|
return func(opts *Options) {
|
|
logger := log.New(os.Stdout, "[reachrs] ", log.LstdFlags)
|
|
opts.Logger = logger
|
|
}
|
|
}
|
|
|
|
// DefaultOptions returns the default values for the client options
|
|
func DefaultOptions() *Options {
|
|
opts := &Options{}
|
|
defaults := []OptionFunc{
|
|
WithEndpoint("192.168.42.1", 80),
|
|
WithPingInterval(30 * time.Second),
|
|
WithPingTimeout(60 * time.Second),
|
|
WithReceiveTimeout(60 * time.Second),
|
|
WithSendTimeout(60 * time.Second),
|
|
WithBufferSize(1024 * 32),
|
|
}
|
|
for _, o := range defaults {
|
|
o(opts)
|
|
}
|
|
return opts
|
|
}
|