libatbus_types

type BusIdType uint64

type ChannelAddress interface {
	GetAddress() string // Full address string (e.g., "tcp://127.0.0.1:8080")
	GetScheme() string  // Protocol scheme (e.g., "tcp", "unix", "shm")
	GetHost() string    // Host part of the address
	GetPort() int       // Port number (0 if not specified)
}

type Connection interface {
	Reset()

	Proc() ErrorType

	Listen() ErrorType

	Connect() ErrorType

	Disconnect() ErrorType

	Push(buffer []byte) ErrorType

	AddStatFault() uint64

	ClearStatFault()

	GetAddress() ChannelAddress

	IsConnected() bool

	IsRunning() bool

	GetBinding() Endpoint

	GetStatus() ConnectionState

	CheckFlag(flag ConnectionFlag) bool

	SetTemporary()

	GetStatistic() ConnectionStatistic

	GetConnectionContext() ConnectionContext

	RemoveOwnerChecker()
}

type ConnectionContext interface {
	// IsClosing reports whether the connection is in a closing state.
	IsClosing() bool

	// SetClosing sets the closing state of the connection.
	SetClosing(closing bool)

	// IsHandshakeDone reports whether the handshake has been completed.
	IsHandshakeDone() bool

	// GetHandshakeStartTime returns the time when the handshake was started.
	GetHandshakeStartTime() time.Time

	// GetCryptoKeyExchangeAlgorithm returns the key exchange algorithm used for crypto handshake.
	GetCryptoKeyExchangeAlgorithm() protocol.ATBUS_CRYPTO_KEY_EXCHANGE_TYPE

	// GetCryptoSelectKdfType returns the KDF type selected during handshake.
	GetCryptoSelectKdfType() protocol.ATBUS_CRYPTO_KDF_TYPE

	// GetCryptoSelectAlgorithm returns the crypto algorithm selected during handshake.
	GetCryptoSelectAlgorithm() protocol.ATBUS_CRYPTO_ALGORITHM_TYPE

	// GetCompressSelectAlgorithm returns the compression algorithm selected during handshake.
	GetCompressSelectAlgorithm() protocol.ATBUS_COMPRESSION_ALGORITHM_TYPE

	// GetNextSequence returns the next sequence number for message ordering.
	// Each call increments the internal counter and returns the new value.
	GetNextSequence() uint64

	// PackMessage packs a Message into a StaticBufferBlock for transmission.
	//
	// Parameters:
	//   - m: the message to pack (head will be modified with version, body_size, etc.)
	//   - protocolVersion: the protocol version to set in the message head
	//   - maxBodySize: maximum allowed body size (0 means no limit)
	//
	// Returns:
	//   - StaticBufferBlock containing the packed message data
	//   - error code if packing fails, EN_ATBUS_ERR_SUCCESS on success
	PackMessage(m *Message, protocolVersion int32, maxBodySize int) (*buffer.StaticBufferBlock, ErrorType)

	// UnpackMessage unpacks binary data into a Message.
	//
	// Parameters:
	//   - m: the message to populate (will be modified)
	//   - input: the binary data to unpack
	//   - maxBodySize: maximum allowed body size (0 means no limit)
	//
	// Returns:
	//   - error code if unpacking fails, EN_ATBUS_ERR_SUCCESS on success
	UnpackMessage(m *Message, input []byte, maxBodySize int) ErrorType

	// HandshakeGenerateSelfKey generates the local ECDH key pair for handshake.
	//
	// In client mode, peerSequenceId should be 0 to generate a new sequence.
	// In server mode, peerSequenceId should be the peer's handshake sequence.
	//
	// Returns error code, EN_ATBUS_ERR_SUCCESS on success.
	HandshakeGenerateSelfKey(peerSequenceId uint64) ErrorType

	// HandshakeReadPeerKey reads the peer's public key and computes the shared secret.
	//
	// Parameters:
	//   - peerPubKey: the peer's handshake data containing public key
	//   - supportedCryptoAlgorithms: list of locally supported crypto algorithms
	//
	// Returns error code, EN_ATBUS_ERR_SUCCESS on success.
	HandshakeReadPeerKey(peerPubKey *protocol.CryptoHandshakeData,
		supportedCryptoAlgorithms []protocol.ATBUS_CRYPTO_ALGORITHM_TYPE) ErrorType

	// HandshakeWriteSelfPublicKey writes the local public key to the handshake data structure.
	//
	// Parameters:
	//   - selfPubKey: output handshake data to populate with local public key
	//   - supportedCryptoAlgorithms: list of locally supported crypto algorithms
	//
	// Returns error code, EN_ATBUS_ERR_SUCCESS on success.
	HandshakeWriteSelfPublicKey(
		selfPubKey *protocol.CryptoHandshakeData,
		supportedCryptoAlgorithms []protocol.ATBUS_CRYPTO_ALGORITHM_TYPE) ErrorType

	// UpdateCompressionAlgorithm updates the list of supported compression algorithms.
	//
	// Returns error code, EN_ATBUS_ERR_SUCCESS on success.
	UpdateCompressionAlgorithm(algorithm []protocol.ATBUS_COMPRESSION_ALGORITHM_TYPE) ErrorType

	// SetupCryptoWithKey directly sets the encryption key and IV, skipping key exchange.
	// This is primarily used for testing purposes.
	//
	// Parameters:
	//   - algorithm: the crypto algorithm type
	//   - key: the encryption key data
	//   - iv: the initialization vector data
	//
	// Returns error code, EN_ATBUS_ERR_SUCCESS on success.
	SetupCryptoWithKey(algorithm protocol.ATBUS_CRYPTO_ALGORITHM_TYPE, key []byte, iv []byte) ErrorType
}

type ConnectionFlag uint32

type ConnectionState int32

type ConnectionStatistic struct {
	PushStartTimes   uint64
	PushStartSize    uint64
	PushSuccessTimes uint64
	PushSuccessSize  uint64
	PushFailedTimes  uint64
	PushFailedSize   uint64

	PullStartTimes uint64
	PullStartSize  uint64

	FaultCount uint64
}

type Endpoint interface {
	GetOwner() Node

	Reset()

	GetId() BusIdType

	GetPid() int32

	GetHostname() string

	GetHashCode() string

	UpdateHashCode(code string)

	AddConnection(conn Connection, forceData bool) bool

	RemoveConnection(conn Connection) bool

	IsAvailable() bool

	GetFlags() uint32

	GetFlag(f EndpointFlag) bool

	SetFlag(f EndpointFlag, v bool)

	GetListenAddress() []ChannelAddress

	ClearListenAddress()

	AddListenAddress(addr string)

	UpdateSupportSchemes(schemes []string)

	IsSchemeSupported(scheme string) bool

	AddPingTimer()

	ClearPingTimer()

	// ============== connection functions ==============
	GetCtrlConnection(peer Endpoint) Connection

	GetDataConnection(peer Endpoint, enableFallbackCtrl bool) Connection

	GetDataConnectionCount(enableFallbackCtrl bool) int

	// ============== statistic functions ==============
	AddStatisticFault() uint64

	ClearStatisticFault()

	SetStatisticUnfinishedPing(p uint64)

	GetStatisticUnfinishedPing() uint64

	SetStatisticPingDelay(pd time.Duration, pongTimepoint time.Time)

	GetStatisticPingDelay() time.Duration

	GetStatisticLastPong() time.Time

	GetStatisticCreatedTime() time.Time

	GetStatisticPushStartTimes() uint64
	GetStatisticPushStartSize() uint64
	GetStatisticPushSuccessTimes() uint64
	GetStatisticPushSuccessSize() uint64
	GetStatisticPushFailedTimes() uint64
	GetStatisticPushFailedSize() uint64
	GetStatisticPullTimes() uint64
	GetStatisticPullSize() uint64
}

type EndpointCollectionType = map[BusIdType]Endpoint

type EndpointFlag uint32

type ErrorType = error_code.ErrorType

type IoStreamCallbackEventHandleSet struct {
	// contains filtered or unexported fields
}

type IoStreamCallbackEventType int32

type IoStreamCallbackFunc func(channel IoStreamChannel, conn IoStreamConnection, status int32, privData interface{})

type IoStreamChannel interface {
	GetContext() context.Context

	SetFlag(f IoStreamConnectionFlag, v bool)

	GetFlag(f IoStreamConnectionFlag) bool

	// 事件响应函数集合
	GetEventHandleSet() *IoStreamCallbackEventHandleSet

	GetStatisticActiveRequestCount() uint64
	GetStatisticReadNetEgainCount() uint64
	GetStatisticCheckBlockSizeFailedCount() uint64
	GetStatisticCheckHashFailedCount() uint64

	// 自定义数据区域
	SetPrivateData(data interface{})

	GetPrivateData() interface{}

	// 核心操作方法
	Listen(addr string) ErrorType
	Connect(addr string) (IoStreamConnection, ErrorType)
	Send(conn IoStreamConnection, data []byte) ErrorType
	Disconnect(conn IoStreamConnection) ErrorType
	Close() ErrorType
}

type IoStreamChannelFlag uint16

type IoStreamConfigure struct {
	Keepalive time.Duration

	NoBlock bool
	NoDelay bool

	SendBufferStatic       uint64
	ReceiveBufferStatic    uint64
	SendBufferMaxSize      uint64
	SendBufferLimitSize    uint64
	ReceiveBufferMaxSize   uint64
	ReceiveBufferLimitSize uint64

	Backlog        int32
	ConfirmTimeout time.Duration

	MaxReadNetEgainCount             uint64
	MaxReadCheckBlockSizeFailedCount uint64
	MaxReadCheckHashFailedCount      uint64
}

type IoStreamConnection interface {
	GetAddress() ChannelAddress
	GetStatus() IoStreamConnectionStatus
	SetFlag(f IoStreamConnectionFlag, v bool)
	GetFlag(f IoStreamConnectionFlag) bool

	GetChannel() IoStreamChannel

	// 事件响应函数集合
	GetEventHandleSet() *IoStreamCallbackEventHandleSet

	// 主动关闭连接的回调（为了减少额外分配而采用的缓存策略）
	GetProactivelyDisconnectCallback() IoStreamCallbackFunc

	GetReadBufferManager() *buffer.BufferManager

	// 自定义数据区域
	SetPrivateData(data interface{})

	GetPrivateData() interface{}
}

type IoStreamConnectionFlag uint16

type IoStreamConnectionStatus uint16

type Message struct {
	// contains filtered or unexported fields
}

type MessageBodyType = protocol.MessageBody_EnMessageTypeID

type Node interface {
	Init(id BusIdType, conf *NodeConfigure) ErrorType

	ReloadCrypto(cryptoKeyExchangeType protocol.ATBUS_CRYPTO_KEY_EXCHANGE_TYPE,
		cryptoKeyRefreshInterval time.Duration,
		cryptoAllowAlgorithms []protocol.ATBUS_CRYPTO_ALGORITHM_TYPE) ErrorType

	ReloadCompression(compressionAllowAlgorithms []protocol.ATBUS_COMPRESSION_ALGORITHM_TYPE,
		compressionLevel protocol.ATBUS_COMPRESSION_LEVEL) ErrorType

	Start() ErrorType

	StartWithConfigure(conf *StartConfigure) ErrorType

	Reset() ErrorType

	Proc(now time.Time) ErrorType

	Poll() ErrorType

	Listen(address string) ErrorType

	Connect(address string) ErrorType

	ConnectWithEndpoint(address string, ep Endpoint) ErrorType

	Disconnect(id BusIdType) ErrorType

	GetCryptoKeyExchangeType() protocol.ATBUS_CRYPTO_KEY_EXCHANGE_TYPE

	// ================= send/receive functions =================
	// SendData 发送数据。
	//
	// Parameters:
	//   - tid: 发送目标 ID
	//   - t: 自定义类型，将作为 message.head.type 字段传递，可用于业务区分服务类型
	//   - data: 要发送的数据块
	//
	// Return: 0 或错误码。
	//
	// Note: 接收端收到的数据很可能不是地址对齐的，所以不建议发送内存数据对象（struct/class）。
	// 如果必须发送内存数据对象，接收端一定要 memcpy，不能直接类型转换，除非手动设置了地址对齐规则。
	SendData(tid BusIdType, t int32, data []byte) ErrorType

	// SendDataWithOptions 发送数据（带选项）。
	//
	// Parameters:
	//   - tid: 发送目标 ID
	//   - t: 自定义类型，将作为 message.head.type 字段传递，可用于业务区分服务类型
	//   - data: 要发送的数据块
	//   - options: 发送选项；如果未设置 sequence，会自动分配并写回
	//
	// Return: 0 或错误码。
	//
	// Note: 接收端收到的数据很可能不是地址对齐的，所以不建议发送内存数据对象（struct/class）。
	// 如果必须发送内存数据对象，接收端一定要 memcpy，不能直接类型转换，除非手动设置了地址对齐规则。
	SendDataWithOptions(tid BusIdType, t int32, data []byte, options *NodeSendDataOptions) ErrorType

	// SendCustomCommand 发送自定义命令。
	//
	// Parameters:
	//   - tid: 发送目标 ID
	//   - args: 自定义消息内容数组
	//
	// Return: 0 或错误码。
	SendCustomCommand(tid BusIdType, args [][]byte) ErrorType

	// SendCustomCommandWithOptions 发送自定义命令（带选项）。
	//
	// Parameters:
	//   - tid: 发送目标 ID
	//   - args: 自定义消息内容数组
	//   - options: 发送选项；如果未设置 sequence，会自动分配并写回
	//
	// Return: 0 或错误码。
	SendCustomCommandWithOptions(tid BusIdType, args [][]byte, options *NodeSendDataOptions) ErrorType

	// ================= endpoint management functions =================
	// GetPeerChannel 获取远程发送目标信息。
	//
	// Parameters:
	//   - tid: 发送目标 ID，不能是自己的 BUS ID
	//   - fn: 获取有效连接的接口，用以区分数据通道和控制通道
	//   - options: 获取选项
	//
	// Return: (0 或错误码, 发送目标 endpoint, 发送连接 connection)。
	GetPeerChannel(tid BusIdType, fn func(from Endpoint, to Endpoint) Connection, options *NodeGetPeerOptions) (ErrorType, Endpoint, Connection, TopologyPeer)

	// SetTopologyUpstream 设置节点的上游节点信息。
	//
	// Parameters:
	//   - tid: 上游节点 ID
	SetTopologyUpstream(tid BusIdType)

	// CreateEndpoint 创建 endpoint 对象，并设置生命周期检查。
	// 并不会自动添加到本地，请在初始化完成后调用 AddEndpoint 添加。
	//
	// Parameters:
	//   - tid: 目标 endpoint 的 bus id
	//   - hostName: 目标 endpoint 的主机名
	//   - pid: 目标 endpoint 的进程 ID
	// Return: endpoint 对象，可能为 nil。
	CreateEndpoint(tid BusIdType, hostName string, pid int) Endpoint

	// GetEndpoint 获取本地 endpoint 对象。
	// Return: endpoint 对象，可能为 nil。
	GetEndpoint(tid BusIdType) Endpoint

	// AddEndpoint 添加目标端点。
	// Return: 0 或错误码。
	AddEndpoint(ep Endpoint) ErrorType

	// RemoveEndpoint 移除目标端点。
	// Return: 0 或错误码。
	RemoveEndpoint(ep Endpoint) ErrorType

	// IsEndpointAvailable 是否有到对端的数据通道（可以向对端发送数据）。
	// Note: 如果只有控制通道没有数据通道，返回 false。
	IsEndpointAvailable(tid BusIdType) bool

	// CheckAccessHash 检查 access token 集合的有效性。
	// Return: 没有检查通过的 access token 则返回 false。
	CheckAccessHash(accessKey *protocol.AccessData, plainText string, conn Connection) bool

	// GetAccessCode 获取节点的 hash code。
	GetAccessCode() string

	GetIoStreamChannel() IoStreamChannel

	GetSelfEndpoint() Endpoint

	GetUpstreamEndpoint() Endpoint

	GetTopologyRelation(tid BusIdType) (TopologyRelationType, TopologyPeer)

	GetImmediateEndpointSet() EndpointCollectionType

	// ================= internal message functions =================
	// SendDataMessage 发送数据消息。
	// Return: (0 或错误码, 发送目标 endpoint, 发送连接 connection)。
	SendDataMessage(tid BusIdType, message *Message, options *NodeSendDataOptions) (ErrorType, Endpoint, Connection)

	// SendCtrlMessage 发送控制消息。
	// Return: (0 或错误码, 发送目标 endpoint, 发送连接 connection)。
	SendCtrlMessage(tid BusIdType, message *Message, options *NodeSendDataOptions) (ErrorType, Endpoint, Connection)

	OnReceiveData(ep Endpoint, conn Connection, message *Message, data []byte)

	OnReceiveForwardResponse(ep Endpoint, conn Connection, message *Message)

	OnDisconnect(ep Endpoint, conn Connection) ErrorType

	OnNewConnection(conn Connection) ErrorType

	OnRegister(ep Endpoint, conn Connection, code ErrorType)

	OnActived()

	OnShutdown(code ErrorType) ErrorType

	OnUpstreamRegisterDone()

	OnCustomCommandRequest(ep Endpoint, conn Connection, from BusIdType, argv [][]byte) (ErrorType, [][]byte)

	OnCustomCommandResponse(ep Endpoint, conn Connection, from BusIdType, argv [][]byte, sequence uint64) ErrorType

	OnPing(ep Endpoint, message *Message, body *protocol.PingData) ErrorType

	OnPong(ep Endpoint, message *Message, body *protocol.PingData) ErrorType

	DispatchAllSelfMessages() int32

	GetContext() context.Context

	// Shutdown 关闭 node。
	//
	// Note: 如果需要在关闭前执行资源回收，可以在 on_node_down_fn_t 回调中返回非 0 值来阻止 node 的 reset 操作，
	// 并在资源释放完成后再调用 Shutdown，在第二次 on_node_down_fn_t 回调中返回 0 值。
	//
	// Note: 或者也可以通过 ref_object 和 unref_object 来标记和解除数据引用，reset 函数会执行事件 loop 直到所有引用的资源被移除。
	Shutdown(reason ErrorType) ErrorType

	FatalShutdown(ep Endpoint, conn Connection, code ErrorType, err error) ErrorType

	SetLogger(logger *utils_log.Logger)

	GetLogger() *utils_log.Logger

	IsDebugMessageVerboseEnabled() bool

	EnableDebugMessageVerbose()

	DisableDebugMessageVerbose()

	AddEndpointGcList(ep Endpoint)

	AddConnectionGcList(conn Connection)

	// ================= event handle functions =================
	SetEventHandleOnForwardRequest(handle NodeOnForwardRequestFunc)
	GetEventHandleOnForwardRequest() NodeOnForwardRequestFunc

	SetEventHandleOnForwardResponse(handle NodeOnForwardResponseFunc)
	GetEventHandleOnForwardResponse() NodeOnForwardResponseFunc

	SetEventHandleOnRegister(handle NodeOnRegisterFunc)
	GetEventHandleOnRegister() NodeOnRegisterFunc

	SetEventHandleOnShutdown(handle NodeOnNodeDownFunc)
	GetEventHandleOnShutdown() NodeOnNodeDownFunc

	SetEventHandleOnAvailable(handle NodeOnNodeUpFunc)
	GetEventHandleOnAvailable() NodeOnNodeUpFunc

	SetEventHandleOnInvalidConnection(handle NodeOnInvalidConnectionFunc)
	GetEventHandleOnInvalidConnection() NodeOnInvalidConnectionFunc

	SetEventHandleOnNewConnection(handle NodeOnNewConnectionFunc)
	GetEventHandleOnNewConnection() NodeOnNewConnectionFunc

	SetEventHandleOnCloseConnection(handle NodeOnCloseConnectionFunc)
	GetEventHandleOnCloseConnection() NodeOnCloseConnectionFunc

	SetEventHandleOnCustomCommandRequest(handle NodeOnCustomCommandRequestFunc)
	GetEventHandleOnCustomCommandRequest() NodeOnCustomCommandRequestFunc

	SetEventHandleOnCustomCommandResponse(handle NodeOnCustomCommandResponseFunc)
	GetEventHandleOnCustomCommandResponse() NodeOnCustomCommandResponseFunc

	SetEventHandleOnAddEndpoint(handle NodeOnEndpointEventFunc)
	GetEventHandleOnAddEndpoint() NodeOnEndpointEventFunc

	SetEventHandleOnRemoveEndpoint(handle NodeOnEndpointEventFunc)
	GetEventHandleOnRemoveEndpoint() NodeOnEndpointEventFunc

	SetEventHandleOnPingEndpoint(handle NodeOnPingPongEndpointFunc)
	GetEventHandleOnPingEndpoint() NodeOnPingPongEndpointFunc

	SetEventHandleOnPongEndpoint(handle NodeOnPingPongEndpointFunc)
	GetEventHandleOnPongEndpoint() NodeOnPingPongEndpointFunc

	SetEventHandleOnTopologyUpdateUpstream(handle NodeOnTopologyUpdateUpstreamFunc)
	GetEventHandleOnTopologyUpdateUpstream() NodeOnTopologyUpdateUpstreamFunc

	// ================= date field setter and getter =================
	GetIoStreamConfigure() *IoStreamConfigure

	GetId() BusIdType

	GetConfigure() *NodeConfigure

	CheckFlag(f NodeFlag) bool

	GetState() NodeState

	GetTimerTick() time.Time

	GetTopologyRegistry() TopologyRegistry

	GetPid() int

	GetHostname() string

	SetHostname(hostname string, force bool) bool

	GetProtocolVersion() int32

	GetProtocolMinimalVersion() int32

	GetListenList() []ChannelAddress

	// ================= statistic functions =================
	AddStatisticDispatchTimes()

	// ================= utility functions =================
	AllocateMessageSequence() uint64

	LogDebug(ep Endpoint, conn Connection, m *Message, msg string, args ...any)
	LogInfo(ep Endpoint, conn Connection, msg string, args ...any)
	LogError(ep Endpoint, conn Connection, status int, errcode ErrorType, msg string, args ...any)
}

type NodeConfigure struct {
	EventLoopContext context.Context

	UpstreamAddress        string            // 上游节点地址
	TopologyLabels         map[string]string // 拓扑标签
	LoopTimes              int32             // 消息循环次数限制，防止某些通道繁忙把其他通道堵死
	TTL                    int32             // 消息转发跳转限制
	ProtocolVersion        int32
	ProtocolMinimalVersion int32

	// ===== 连接配置 =====
	BackLog              int32
	FirstIdleTimeout     time.Duration // 第一个包允许的空闲时间
	PingInterval         time.Duration // ping包间隔
	RetryInterval        time.Duration // 重试包间隔
	FaultTolerant        uint32        // 容错次数（次）
	AccessTokenMaxNumber uint32        // 最大 access token 数量，请不要设置的太大，验证次数最大可能是 N^2
	AccessTokens         [][]byte      // access token 列表
	OverwriteListenPath  bool          // 是否覆盖已存在的 listen path(unix/pipe socket)

	// ===== 加密算法配置 =====
	CryptoKeyExchangeType    protocol.ATBUS_CRYPTO_KEY_EXCHANGE_TYPE
	CryptoKeyRefreshInterval time.Duration
	CryptoAllowAlgorithms    []protocol.ATBUS_CRYPTO_ALGORITHM_TYPE

	// ===== 压缩算法配置 =====
	CompressionAllowAlgorithms []protocol.ATBUS_COMPRESSION_ALGORITHM_TYPE
	CompressionLevel           protocol.ATBUS_COMPRESSION_LEVEL

	// ===== 缓冲区配置 =====
	MessageSize      uint64 // max message size
	RecvBufferSize   uint64 // 接收缓冲区，和数据包大小有关
	SendBufferSize   uint64 // 发送缓冲区限制
	SendBufferNumber uint64 // 发送缓冲区静态 Buffer 数量限制，0 则为动态缓冲区
}

type NodeEventHandleSet struct {
	OnForwardRequest         NodeOnForwardRequestFunc
	OnForwardResponse        NodeOnForwardResponseFunc
	OnRegister               NodeOnRegisterFunc
	OnNodeDown               NodeOnNodeDownFunc
	OnNodeUp                 NodeOnNodeUpFunc
	OnInvalidConnection      NodeOnInvalidConnectionFunc
	OnNewConnection          NodeOnNewConnectionFunc
	OnCloseConnection        NodeOnCloseConnectionFunc
	OnCustomCommandRequest   NodeOnCustomCommandRequestFunc
	OnCustomCommandResponse  NodeOnCustomCommandResponseFunc
	OnEndpointAdded          NodeOnEndpointEventFunc
	OnEndpointRemoved        NodeOnEndpointEventFunc
	OnEndpointPing           NodeOnPingPongEndpointFunc
	OnEndpointPong           NodeOnPingPongEndpointFunc
	OnTopologyUpdateUpstream NodeOnTopologyUpdateUpstreamFunc
}

type NodeFlag uint32

type NodeGetPeerOptionFlag uint16

type NodeGetPeerOptions struct {
	// contains filtered or unexported fields
}

type NodeOnCloseConnectionFunc func(n Node, ep Endpoint, conn Connection) ErrorType

type NodeOnCustomCommandRequestFunc func(n Node, ep Endpoint, conn Connection, from BusIdType, argv [][]byte) (result ErrorType, response [][]byte)

type NodeOnCustomCommandResponseFunc func(n Node, ep Endpoint, conn Connection, from BusIdType, rspData [][]byte, sequence uint64) ErrorType

type NodeOnEndpointEventFunc func(n Node, ep Endpoint, status ErrorType) ErrorType

type NodeOnForwardRequestFunc func(n Node, ep Endpoint, conn Connection, msg *Message, data []byte) ErrorType

type NodeOnForwardResponseFunc func(n Node, ep Endpoint, conn Connection, msg *Message) ErrorType

type NodeOnInvalidConnectionFunc func(n Node, conn Connection, errCode ErrorType) ErrorType

type NodeOnNewConnectionFunc func(n Node, conn Connection) ErrorType

type NodeOnNodeDownFunc func(n Node, errCode ErrorType) ErrorType

type NodeOnNodeUpFunc func(n Node, errCode ErrorType) ErrorType

type NodeOnPingPongEndpointFunc func(n Node, ep Endpoint, msg *Message, ping *protocol.PingData) ErrorType

type NodeOnRegisterFunc func(n Node, ep Endpoint, conn Connection, errCode ErrorType) ErrorType

type NodeOnTopologyUpdateUpstreamFunc func(n Node, self TopologyPeer, upstream TopologyPeer, data *TopologyData) ErrorType

type NodeSendDataOptionFlag uint32

type NodeSendDataOptions struct {
	// contains filtered or unexported fields
}

type NodeState int32

type StartConfigure struct {
	TimerTimepoint time.Time
}

type TimerDescPair[V any] struct {
	Timeout time.Time
	Value   V
}

type TimerDescType[K comparable, V any] = utils_memory.LRUMap[K, TimerDescPair[V]]

type TopologyData struct {
	// Pid is the process id of the peer.
	Pid int32
	// Hostname is the hostname of the peer.
	Hostname string
	// Labels contains arbitrary labels for policy matching (e.g. region/zone/group/version).
	Labels map[string]string
}

type TopologyPeer interface {
	// GetBusId returns the bus id of this peer.
	GetBusId() BusIdType

	// GetUpstream returns the upstream (parent) peer, or nil if this peer is a root.
	GetUpstream() TopologyPeer

	// GetTopologyData returns current topology data of this peer.
	GetTopologyData() *TopologyData

	// ContainsDownstream checks whether a downstream(peer) with given bus id exists.
	ContainsDownstream(busId BusIdType) bool

	// ForeachDownstream iterates all downstream peers.
	ForeachDownstream(fn func(peer TopologyPeer) bool) bool
}

type TopologyPolicyRule struct {
	// RequireSameProcess requires same pid and same hostname if true.
	RequireSameProcess bool
	// RequireSameHostName requires same hostname if true.
	RequireSameHostName bool
	// RequireLabelValues contains label constraints.
	// Key: label name.
	// Value: allowed label values (set).
	RequireLabelValues map[string]map[string]struct{}
}

type TopologyRegistry interface {
	// GetPeer returns peer by bus id, or nil if not found.
	GetPeer(busId BusIdType) TopologyPeer

	// RemovePeer removes a peer and fixes relationships.
	// If the peer has an upstream, it will be removed from the upstream's downstream set.
	// For each downstream peer whose upstream is this peer, its upstream will be cleared.
	RemovePeer(targetBusId BusIdType)

	// UpdatePeer creates or updates a peer.
	//   - If targetBusId is 0, this call is ignored.
	//   - If upstreamBusId is 0, the peer becomes a root (no upstream).
	//   - If peer exists and upstream changed, downstream links will be updated accordingly.
	// true on success, false on failure (e.g. invalid target_bus_id or there will be a circle).
	UpdatePeer(targetBusId BusIdType, upstreamBusId BusIdType, data *TopologyData) bool

	// GetRelation returns the topology relation between two peers.
	// The first return value is the relation type.
	// The second return value is the next-hop peer (may be nil):
	//   - If the relation is TransitiveUpstream or TransitiveDownstream, it's to.
	//   - If the relation is SameUpstreamPeer, it's the upstream of both from and to.
	//   - If the relation is OtherUpstreamPeer, it prefers the upstream of from;
	//     if from has no upstream then returns to.
	GetRelation(from BusIdType, to BusIdType) (TopologyRelationType, TopologyPeer)

	// ForeachPeer iterates all peers in registry.
	// The callback function returns false to stop iteration early.
	// Returns true if iterated all peers, false if stopped by callback.
	ForeachPeer(fn func(peer TopologyPeer) bool) bool

	// CheckPolicy checks whether toData satisfies the fromPolicy.
	// The checks include:
	//   - same hostname (optional)
	//   - same process (optional, implies same hostname)
	//   - required labels (optional)
	CheckPolicy(rule *TopologyPolicyRule, fromData *TopologyData, toData *TopologyData) bool
}

type TopologyRelationType uint8