derElektrobesen/goconnpool

package goconnpool

import (
    "context"
    "net"
)

// Dialer interface used to dial specific server independently from the specific protocol.
type Dialer interface {
    // Dial dials the address passed (this address is one of the registered servers addresses).
    //
    // Function should return net.Conn-compatible connection.
    //
    // Function should be implemented in the thread-safe way.
    // Connect timeout is setuped before this function invocation: be sure your function not stuck in the case of
    // timeout.
    //
    // address variable is the same address used into RegisterServer call.
    //
    // You could get original connection returned by the Dial() call using Conn.OriginalConn().
    Dial(ctx context.Context, address string) (net.Conn, error)
}

// TCPDialer is the default implementation of Dialer interface.
// Use it for raw TCP connections.
type TCPDialer struct {
    d net.Dialer
}

// Dial dials some TCP server using net.Dialer structure.
func (d *TCPDialer) Dial(ctx context.Context, address string) (net.Conn, error) {
    return d.d.DialContext(ctx, "tcp", address)
}