deprecated_overflow_gateway_websocket/socket_handlers.go

package overflow_gateway_websocket

import (
	"log"
	"time"
)

const (
	// DefaultWriteTimeout is default value of Write Timeout
	DefaultWriteTimeout = 0
	// DefaultReadTimeout is default value of Read Timeout
	DefaultReadTimeout = 0
	// DefaultPongTimeout is default value of Pong Timeout
	DefaultPongTimeout = 60 * time.Second
	// DefaultPingTimeout is default value of Ping Timeout
	DefaultPingTimeout = 10 * time.Second
	// DefaultPingPeriod is default value of Ping Period
	DefaultPingPeriod = (DefaultPongTimeout * 9) / 10
	// DefaultMaxMessageSize is default value of Max Message Size
	DefaultMaxMessageSize = 1024
)

// SocketOptions is configuration of the websocket server
type SocketHandlers struct {
	onDisconnected func(soc Socket)

	Handler ProtocolHandler

	MaxMessageSize int64
	WriteTimeout   time.Duration
	ReadTimeout    time.Duration
	PongTimeout    time.Duration
	PingTimeout    time.Duration
	PingPeriod     time.Duration
	BinaryMessage  bool
}

func (sh *SocketHandlers) GetMaxMessageSize() int64 {
	return sh.MaxMessageSize
}
func (sh *SocketHandlers) GetWriteTimeout() time.Duration {
	return sh.WriteTimeout
}
func (sh *SocketHandlers) GetReadTimeout() time.Duration {
	return sh.ReadTimeout
}
func (sh *SocketHandlers) GetPongTimeout() time.Duration {
	return sh.PongTimeout
}
func (sh *SocketHandlers) GetPingTimeout() time.Duration {
	return sh.PingTimeout
}
func (sh *SocketHandlers) GetPingPeriod() time.Duration {
	return sh.PingPeriod
}
func (sh *SocketHandlers) IsBinaryMessage() bool {
	return sh.BinaryMessage
}

func (sh *SocketHandlers) GetProtocolHandler() ProtocolHandler {
	return sh.Handler
}

// Validate validates the configuration
func (o *SocketHandlers) Validate() {
	if nil == o.Handler {
		log.Fatalf("Message Handler must specified.\n")
	}

	if o.WriteTimeout < 0 {
		o.WriteTimeout = DefaultWriteTimeout
	}

	if o.ReadTimeout < 0 {
		o.ReadTimeout = DefaultReadTimeout
	}

	if o.PongTimeout < 0 {
		o.PongTimeout = DefaultPongTimeout
	}

	if o.PingPeriod <= 0 {
		o.PingPeriod = DefaultPingPeriod
	}

	if o.MaxMessageSize <= 0 {
		o.MaxMessageSize = DefaultMaxMessageSize
	}

}