github.com/volts-dev/volts@v0.0.0-20240120094013-5e9c65924106/transport/transport.go

package transport

import (
	"net"

	"github.com/volts-dev/volts/codec"
	"github.com/volts-dev/volts/internal/body"
)

type (
	ITransport interface {
		Init(...Option) error
		Config() *Config
		Dial(addr string, opts ...DialOption) (IClient, error)       // for client 详细查看pool.NewPool
		Listen(addr string, opts ...ListenOption) (IListener, error) // for server
		String() string
		Protocol() string
	}

	ISocket interface {
		Recv(*Message) error
		Send(*Message) error
		Close() error
		Local() string  // Local IP
		Remote() string // Remote IP
		Conn() net.Conn // 接口提供更灵活扩展
	}

	IClient interface {
		ISocket
		Transport() ITransport
	}

	IListener interface {
		Addr() net.Addr
		Close() error
		Accept() (net.Conn, error)
		//Serve(func(ISocket)) error // 阻塞监听
		Serve(Handler) error // 阻塞监听
		//Sock() ISocket
	}

	// the handler interface
	Handler interface {
		String() string
		Handler() interface{}
	}

	IRequest interface {
		// The service to call
		Service() string
		// The action to take
		Method() string
		// The content type
		ContentType() string
		// write a response directly to the client
		Body() IBody // *body.TBody

		Codec() codec.ICodec
	}

	// 提供给服务器客户端最基本接口
	IResponse interface {
		// write a response directly to the client
		Write([]byte) (int, error)
		WriteStream(interface{}) error
		Body() *body.TBody
	}

	IBody interface {
		Read(interface{}) error
		Write(interface{}) error
	}
)

var defaultTransport ITransport

func Default(set ...ITransport) ITransport {
	if len(set) > 0 {
		defaultTransport = set[0]
	} else if defaultTransport == nil {
		defaultTransport = NewHTTPTransport()
	}

	return defaultTransport
}