client

type CallOption func(*CallOptions)

type CallOptions struct {
	// Used to select a codec
	ContentType string

	// PreferredTransports contains a list of transport names in preferred order.
	PreferredTransports []string

	// Allow any transport in any order.
	AnyTransport bool

	// Selector is the node selector.
	Selector SelectorFunc
	// Check if retriable func
	RetryFunc RetryFunc
	// Number of Call attempts
	Retries int
	// Transport Dial Timeout. Used for initial dial to establish a connection.
	DialTimeout time.Duration
	// ConnectionTimeout of one request to the server.
	// Set this lower than the RequestTimeout to enable retries on connection timeout.
	ConnectionTimeout time.Duration
	// Request/Response timeout of entire srv.Call, for single request timeout set ConnectionTimeout.
	RequestTimeout time.Duration
	// Stream timeout for the stream
	StreamTimeout time.Duration
	// ConnClose sets the Connection: close header.
	ConnClose bool
	// URL bypasses the registry when set. This is mainly for tests.
	// Only <scheme>://<host:port> will be used from it.
	URL string
	// TLS config.
	TLSConfig *tls.Config

	// Metadata to be sent with the request.
	Metadata map[string]string

	// ResponseMetadata will be written into `ResponseMetadata` when given.
	ResponseMetadata map[string]string

	// MaxCallRecvMsgSize is the maximum size of the call receive message size.
	MaxCallRecvMsgSize int

	// MaxCallSendMsgSize is the maximum size of the call send message size.
	MaxCallSendMsgSize int

	// Namespace registry filter.
	Namespace string

	// Region registry filter.
	Region string
}

type Client interface {
	types.Component

	// Config returns the internal config, this is for tests.
	Config() Config

	// With closes all transports and configures the client with the given options.
	With(opts ...Option) error

	// SelectService selects a service node.
	// Returns:
	// - address
	// - transport
	// - error
	SelectService(ctx context.Context, service string, opts ...CallOption) (string, string, error)

	// Request does the actual call.
	Request(ctx context.Context, service string, endpoint string, req any, result any, opts ...CallOption) error

	// Stream creates a streaming client to the specified service endpoint.
	Stream(ctx context.Context, service string, endpoint string, opts ...CallOption) (StreamIface[any, any], error)
}

type Config struct {
	// Plugin selects the client implementation.
	Plugin string `json:"plugin" yaml:"plugin"`

	Middleware []MiddlewareConfig

	// Used to select a codec
	ContentType string `json:"contentType" yaml:"contentType"`

	// PreferredTransports contains a list of transport names in preferred order.
	PreferredTransports []string `json:"preferredTransports" yaml:"preferredTransports"`

	// AnyTransport enables Transports which are not in PreferredTransports.
	AnyTransport bool `json:"anyTransport" yaml:"anyTransport"`

	// Connection Pool
	PoolHosts int             `json:"poolHosts" yaml:"poolHosts"`
	PoolSize  int             `json:"poolSize"  yaml:"poolSize"`
	PoolTTL   config.Duration `json:"poolTtl"   yaml:"poolTtl"`

	// SelectorFunc get's executed by client.SelectNode which get it's info's from client.ResolveService.
	Selector SelectorFunc `json:"-" yaml:"-"`

	// Transport Dial Timeout. Used for initial dial to establish a connection.
	DialTimeout config.Duration `json:"dialTimeout" yaml:"dialTimeout"`
	// ConnectionTimeout of one request to the server.
	// Set this lower than the RequestTimeout to enbale retries on connection timeout.
	ConnectionTimeout config.Duration `json:"connectionTimeout" yaml:"connectionTimeout"`
	// Request/Response timeout of entire srv.Call, for single request timeout set ConnectionTimeout.
	RequestTimeout config.Duration `json:"requestTimeout" yaml:"requestTimeout"`
	// Stream timeout for the stream
	StreamTimeout config.Duration `json:"streamTimeout" yaml:"streamTimeout"`
	// TLS config.
	TLSConfig *tls.Config

	// Namespace registry filter.
	Namespace string `json:"namespace" yaml:"namespace"`

	// Region registry filter.
	Region string `json:"region" yaml:"region"`
}

type ConfigType interface {
	// contains filtered or unexported methods
}

type MemoryServer interface {
	// Request does the actual call.
	Request(ctx context.Context, infos RequestInfos, req any, result any, opts *CallOptions) error

	// Stream creates a streaming client to the specified service endpoint.
	Stream(ctx context.Context, infos RequestInfos, opts *CallOptions) (StreamIface[any, any], error)
}

type Middleware interface {
	types.Component

	Request(
		next MiddlewareRequestHandler,
	) MiddlewareRequestHandler
}

type MiddlewareConfig struct {
	Name string `json:"name" yaml:"name"`
}

type MiddlewareFactory func(config map[string]any, client Type, logger log.Logger) (Middleware, error)

type MiddlewareRequestHandler func(ctx context.Context, service string, endpoint string, req any, result any, opts *CallOptions) error

type Option func(ConfigType)

type ProviderFunc func(
	configData map[string]any,
	components *types.Components,
	logger log.Logger,
	registry registry.Type,
	opts ...Option,
) (Type, error)

type RequestInfos struct {
	Service   string
	Endpoint  string
	Transport string
	Address   string
}

type RequestInfosKey struct{}

type RetryFunc func(ctx context.Context, err error, options *CallOptions) (bool, error)

type SelectorFunc func(
	ctx context.Context,
	service string,
	nodes []registry.ServiceNode,
) (registry.ServiceNode, error)

type StreamAdapter[TReq any, TResp any] struct {
	// contains filtered or unexported fields
}

type StreamIface[TReq any, TResp any] interface {
	// Send sends a message to the stream.
	Send(msg TReq) error

	// Recv receives a message from the stream.
	Recv(msg TResp) error

	// Close closes the stream.
	Close() error

	// CloseSend closes the send direction of the stream but allows receiving responses.
	CloseSend() error

	// Context returns the context for the stream.
	Context() context.Context
}

type Type struct {
	Client
}