protocol

type Body interface {
	Encoder
	Key() int16
	Version() int16
}

type Broker struct {
	NodeID int32
	Host   string
	Port   int32
}

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

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

type CRCField struct {
	StartOffset int
}

type Coordinator struct {
	NodeID int32
	Host   string
	Port   int32
}

type CreateTopicRequest struct {
	Topic             string
	NumPartitions     int32
	ReplicationFactor int16
	ReplicaAssignment map[int32][]int32
	Configs           map[string]string
}

type CreateTopicRequests struct {
	Requests []*CreateTopicRequest
	Timeout  int32
}

type CreateTopicsResponse struct {
	TopicErrorCodes []*TopicErrorCode
}

type Data struct {
	Partition int32
	RecordSet []byte
}

type Decoder interface {
	Decode(d PacketDecoder) error
}

type DeleteTopicsRequest struct {
	Topics  []string
	Timeout int32
}

type DeleteTopicsResponse struct {
	TopicErrorCodes []*TopicErrorCode
}

type DescribeGroupsRequest struct {
	GroupIDs []string
}

type DescribeGroupsResponse struct {
	Groups []*Group
}

type Encoder interface {
	Encode(e PacketEncoder) error
}

type FetchPartition struct {
	Partition   int32
	FetchOffset int64
	MaxBytes    int32
}

type FetchPartitionResponse struct {
	Partition     int32
	ErrorCode     int16
	HighWatermark int64
	RecordSet     []byte
}

type FetchRequest struct {
	ReplicaID   int32
	MaxWaitTime int32
	MinBytes    int32
	// MaxBytes    int32
	Topics []*FetchTopic
}

type FetchResponse struct {
	Topic              string
	PartitionResponses []*FetchPartitionResponse
}

type FetchResponses struct {
	ThrottleTimeMs int32
	Responses      []*FetchResponse
}

type FetchTopic struct {
	Topic      string
	Partitions []*FetchPartition
}

type Group struct {
	ErrorCode    int16
	GroupID      string
	State        string
	ProtocolType string
	Protocol     string
	GroupMembers map[string]*GroupMember
}

type GroupCoordinatorRequest struct {
	GroupID string
}

type GroupCoordinatorResponse struct {
	ErrorCode   int16
	Coordinator *Coordinator
}

type GroupMember struct {
	ClientID              string
	ClientHost            string
	GroupMemberMetadata   []byte
	GroupMemberAssignment []byte
}

type GroupProtocol struct {
	ProtocolName     string
	ProtocolMetadata []byte
}

type HeartbeatRequest struct {
	GroupID           string
	GroupGenerationID int32
	MemberID          string
}

type HeartbeatResponse struct {
	ErrorCode int16
}

type JoinGroupRequest struct {
	GroupID        string
	SessionTimeout int32
	MemberID       string
	ProtocolType   string
	GroupProtocols []*GroupProtocol
}

type JoinGroupResponse struct {
	ErrorCode     int16
	GenerationID  int32
	GroupProtocol string
	LeaderID      string
	MemberID      string
	Members       map[string][]byte
}

type LeaderAndISRPartition struct {
	Topic     string
	Partition int32
	ErrorCode int16
}

type LeaderAndISRRequest struct {
	ControllerID    int32
	ControllerEpoch int32
	PartitionStates []*PartitionState
	LiveLeaders     []*LiveLeader
}

type LeaderAndISRResponse struct {
	ErrorCode  int16
	Partitions []*LeaderAndISRPartition
}

type LeaveGroupRequest struct {
	GroupID  string
	MemberID string
}

type LeaveGroupResponse struct {
	ErrorCode int16
}

type LenEncoder struct {
	Length int
	// contains filtered or unexported fields
}

type ListGroupsRequest struct {
}

type ListGroupsResponse struct {
	ErrorCode int16
	Groups    map[string]string
}

type LiveLeader struct {
	ID   int32
	Host string
	Port int32
}

type Member struct {
	MemberID       string
	MemberMetadata []byte
}

type Message struct {
	Timestamp time.Time
	Key       []byte
	Value     []byte
	MagicByte int8
}

type MessageSet struct {
	Offset                  int64
	Size                    int32
	Messages                []*Message
	PartialTrailingMessages bool
}

type MetadataRequest struct {
	Topics []string
}

type MetadataResponse struct {
	Brokers []*Broker
	// unsupported: ClusterID *string
	// unsupported: ControllerID string
	TopicMetadata []*TopicMetadata
}

type OffsetResponse struct {
	Topic              string
	PartitionResponses []*PartitionResponse
}

type OffsetsPartition struct {
	Partition int32
	Timestamp int64 // -1 to receive latest offset, -2 to receive earliest offset
}

type OffsetsRequest struct {
	ReplicaID     int32
	Topics        []*OffsetsTopic
	MaxNumOffsets int32
}

type OffsetsResponse struct {
	Responses []*OffsetResponse
}

type OffsetsTopic struct {
	Topic      string
	Partitions []*OffsetsPartition
}

type PacketDecoder interface {
	Bool() (bool, error)
	Int8() (int8, error)
	Int16() (int16, error)
	Int32() (int32, error)
	Int64() (int64, error)
	ArrayLength() (int, error)
	Bytes() ([]byte, error)
	String() (string, error)
	Int32Array() ([]int32, error)
	Int64Array() ([]int64, error)
	StringArray() ([]string, error)
	Push(pd PushDecoder) error
	Pop() error
	// contains filtered or unexported methods
}

type PacketEncoder interface {
	PutBool(in bool)
	PutInt8(in int8)
	PutInt16(in int16)
	PutInt32(in int32)
	PutInt64(in int64)
	PutArrayLength(in int) error
	PutRawBytes(in []byte) error
	PutBytes(in []byte) error
	PutString(in string) error
	PutStringArray(in []string) error
	PutInt32Array(in []int32) error
	PutInt64Array(in []int64) error
	Push(pe PushEncoder)
	Pop()
}

type PartitionMetadata struct {
	PartitionErrorCode int16
	ParititionID       int32
	Leader             int32
	Replicas           []int32
	ISR                []int32
}

type PartitionResponse struct {
	Partition int32
	ErrorCode int16
	// Timestamp int64
	Offsets []int64
}

type PartitionState struct {
	Topic           string
	Partition       int32
	ControllerEpoch int32
	Leader          int32
	LeaderEpoch     int32
	ISR             []int32
	ZKVersion       int32
	Replicas        []int32
}

type ProducePartitionResponse struct {
	Partition  int32
	ErrorCode  int16
	BaseOffset int64
	Timestamp  int64
}

type ProduceRequest struct {
	Acks      int16
	Timeout   int32
	TopicData []*TopicData
}

type ProduceResponse struct {
	Topic              string
	PartitionResponses []*ProducePartitionResponse
}

type ProduceResponses struct {
	Responses      []*ProduceResponse
	ThrottleTimeMs int32
}

type PushDecoder interface {
	SaveOffset(in int)
	ReserveSize() int
	Fill(curOffset int, buf []byte) error
}

type PushEncoder interface {
	SaveOffset(in int)
	ReserveSize() int
	Fill(curOffset int, buf []byte) error
}

type Request struct {
	CorrelationID int32
	ClientID      string
	Body          Body
}

type RequestHeader struct {
	// Size of the request
	Size int32
	// ID of the API (e.g. produce, fetch, metadata)
	APIKey int16
	// Version of the API to use
	APIVersion int16
	// User defined ID to correlate requests between server and client
	CorrelationID int32
	// Size of the Client ID
	ClientID string
}

type Response struct {
	Size          int32
	CorrelationID int32
	Body          ResponseBody
}

type ResponseBody interface {
	Encoder
	Decoder
}

type SizeField struct {
	StartOffset int
}

type StopReplicaPartition struct {
	Topic     string
	Partition int32
}

type StopReplicaRequest struct {
	ControllerID     int32
	ControllerEpoch  int32
	DeletePartitions bool
	Partitions       []*StopReplicaPartition
}

type StopReplicaResponse struct {
	ErrorCode  int16
	Partitions []*StopReplicaResponsePartition
}

type StopReplicaResponsePartition struct {
	Topic     string
	Partition int32
	ErrorCode int16
}

type SyncGroupRequest struct {
	GroupID          string
	GenerationID     int32
	MemberID         string
	GroupAssignments map[string][]byte
}

type SyncGroupResponse struct {
	ErrorCode        int16
	MemberAssignment []byte
}

type TopicData struct {
	Topic string
	Data  []*Data
}

type TopicErrorCode struct {
	Topic     string
	ErrorCode int16
}

type TopicMetadata struct {
	TopicErrorCode    int16
	Topic             string
	PartitionMetadata []*PartitionMetadata
}