101 lines
2.0 KiB
Go
101 lines
2.0 KiB
Go
package sarama
|
|
|
|
type SyncGroupRequest struct {
|
|
GroupId string
|
|
GenerationId int32
|
|
MemberId string
|
|
GroupAssignments map[string][]byte
|
|
}
|
|
|
|
func (r *SyncGroupRequest) encode(pe packetEncoder) error {
|
|
if err := pe.putString(r.GroupId); err != nil {
|
|
return err
|
|
}
|
|
|
|
pe.putInt32(r.GenerationId)
|
|
|
|
if err := pe.putString(r.MemberId); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := pe.putArrayLength(len(r.GroupAssignments)); err != nil {
|
|
return err
|
|
}
|
|
for memberId, memberAssignment := range r.GroupAssignments {
|
|
if err := pe.putString(memberId); err != nil {
|
|
return err
|
|
}
|
|
if err := pe.putBytes(memberAssignment); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (r *SyncGroupRequest) decode(pd packetDecoder, version int16) (err error) {
|
|
if r.GroupId, err = pd.getString(); err != nil {
|
|
return
|
|
}
|
|
if r.GenerationId, err = pd.getInt32(); err != nil {
|
|
return
|
|
}
|
|
if r.MemberId, err = pd.getString(); err != nil {
|
|
return
|
|
}
|
|
|
|
n, err := pd.getArrayLength()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if n == 0 {
|
|
return nil
|
|
}
|
|
|
|
r.GroupAssignments = make(map[string][]byte)
|
|
for i := 0; i < n; i++ {
|
|
memberId, err := pd.getString()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
memberAssignment, err := pd.getBytes()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
r.GroupAssignments[memberId] = memberAssignment
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (r *SyncGroupRequest) key() int16 {
|
|
return 14
|
|
}
|
|
|
|
func (r *SyncGroupRequest) version() int16 {
|
|
return 0
|
|
}
|
|
|
|
func (r *SyncGroupRequest) requiredVersion() KafkaVersion {
|
|
return V0_9_0_0
|
|
}
|
|
|
|
func (r *SyncGroupRequest) AddGroupAssignment(memberId string, memberAssignment []byte) {
|
|
if r.GroupAssignments == nil {
|
|
r.GroupAssignments = make(map[string][]byte)
|
|
}
|
|
|
|
r.GroupAssignments[memberId] = memberAssignment
|
|
}
|
|
|
|
func (r *SyncGroupRequest) AddGroupAssignmentMember(memberId string, memberAssignment *ConsumerGroupMemberAssignment) error {
|
|
bin, err := encode(memberAssignment, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
r.AddGroupAssignment(memberId, bin)
|
|
return nil
|
|
}
|