server/server_handlers.go

package server

import (
	"errors"
	"io"
	"net"
	"time"
)

type ServerHandlers struct {
	// Address to listen to for incoming connections.
	//
	// The address format depends on the underlying transport provided
	// by Server.Listener. The following transports are provided
	// out of the box:
	//   * TCP - see NewTCPServer() and NewTCPClient().
	//   * TLS (aka SSL) - see NewTLSServer() and NewTLSClient().
	//   * Unix sockets - see NewUnixServer() and NewUnixClient().
	//
	// By default TCP transport is used.
	Addr string

	KeepAlivePeriod time.Duration
}

func (sh *ServerHandlers) OnStart() {
	// no op
}

func (sh *ServerHandlers) OnStop() {
	// no op
}

func (sh *ServerHandlers) Listen() (net.Listener, error) {
	return nil, errors.New("Server: Handler method[Listen] of Server is not implement")
}

func (sh *ServerHandlers) OnHandshake(remoteAddr string, rwc io.ReadWriteCloser) error {
	return nil
}

func (sh *ServerHandlers) Handle(remoteAddr string, rwc io.ReadWriteCloser, stopChan chan struct{}) {

}

func (sh *ServerHandlers) IsClientDisconnect(err error) bool {
	return err == io.ErrUnexpectedEOF || err == io.EOF
}

func (sh *ServerHandlers) GetAddr() string {
	return sh.Addr
}

func (sh *ServerHandlers) accept(l net.Listener) (io.ReadWriteCloser, string, error) {
	conn, err := l.Accept()
	if nil != err {
		return nil, "", err
	}

	if 0 < sh.KeepAlivePeriod {
		if err = setupKeepalive(conn, sh.KeepAlivePeriod); nil != err {
			conn.Close()
			return nil, "", err
		}
	}

	return conn, conn.RemoteAddr().String(), nil
}

func (sh *ServerHandlers) Validate() {
	if sh.KeepAlivePeriod <= 0 {
		sh.KeepAlivePeriod = DefaultKeepAlivePeriod
	}

}

func setupKeepalive(conn net.Conn, keepAlivePeriod time.Duration) error {
	tcpConn, ok := conn.(*net.TCPConn)
	if !ok {
		return errors.New("Server: Keepalive is valid when connection is net.TCPConn")
	}

	if err := tcpConn.SetKeepAlive(true); err != nil {
		return err
	}
	if err := tcpConn.SetKeepAlivePeriod(keepAlivePeriod * time.Second); err != nil {
		return err
	}
	return nil
}