Merge pull request #1381 fix builder methods return self

rekby · web-flow · commit f9533b572f89 · 2024-07-30T19:44:35.000+03:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,3 +1,4 @@
+* Fixed test message builder, now all method return itself pointer
 * Fixed handle reconnection timeout error
 * Fixed experimental topic listener handle stop partition event
 
diff --git a/internal/topic/topicreadercommon/message.go b/internal/topic/topicreadercommon/message.go
@@ -209,18 +209,24 @@ func (pmb *PublicMessageBuilder) UncompressedSize(uncompressedSize int) *PublicM
 }
 
 // Context set message Context
-func (pmb *PublicMessageBuilder) Context(ctx context.Context) {
+func (pmb *PublicMessageBuilder) Context(ctx context.Context) *PublicMessageBuilder {
 	pmb.mess.commitRange.PartitionSession.SetContext(ctx)
+
+	return pmb
 }
 
 // Topic set message Topic
-func (pmb *PublicMessageBuilder) Topic(topic string) {
+func (pmb *PublicMessageBuilder) Topic(topic string) *PublicMessageBuilder {
 	pmb.mess.commitRange.PartitionSession.Topic = topic
+
+	return pmb
 }
 
 // PartitionID set message PartitionID
-func (pmb *PublicMessageBuilder) PartitionID(partitionID int64) {
+func (pmb *PublicMessageBuilder) PartitionID(partitionID int64) *PublicMessageBuilder {
 	pmb.mess.commitRange.PartitionSession.PartitionID = partitionID
+
+	return pmb
 }
 
 func (pmb *PublicMessageBuilder) RawDataLen(val int) *PublicMessageBuilder {

Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,4 @@`
	`1`	`+* Fixed test message builder, now all method return itself pointer`
`1`	`2`	`* Fixed handle reconnection timeout error`
`2`	`3`	`* Fixed experimental topic listener handle stop partition event`
`3`	`4`
Original file line number	Diff line number	Diff line change
`@@ -209,18 +209,24 @@ func (pmb PublicMessageBuilder) UncompressedSize(uncompressedSize int) PublicM`
`209`	`209`	`}`
`210`	`210`
`211`	`211`	`// Context set message Context`
`212`		`-func (pmb *PublicMessageBuilder) Context(ctx context.Context) {`
	`212`	`+func (pmb PublicMessageBuilder) Context(ctx context.Context) PublicMessageBuilder {`
`213`	`213`	`pmb.mess.commitRange.PartitionSession.SetContext(ctx)`
	`214`	`+`
	`215`	`+ return pmb`
`214`	`216`	`}`
`215`	`217`
`216`	`218`	`// Topic set message Topic`
`217`		`-func (pmb *PublicMessageBuilder) Topic(topic string) {`
	`219`	`+func (pmb PublicMessageBuilder) Topic(topic string) PublicMessageBuilder {`
`218`	`220`	`pmb.mess.commitRange.PartitionSession.Topic = topic`
	`221`	`+`
	`222`	`+ return pmb`
`219`	`223`	`}`
`220`	`224`
`221`	`225`	`// PartitionID set message PartitionID`
`222`		`-func (pmb *PublicMessageBuilder) PartitionID(partitionID int64) {`
	`226`	`+func (pmb PublicMessageBuilder) PartitionID(partitionID int64) PublicMessageBuilder {`
`223`	`227`	`pmb.mess.commitRange.PartitionSession.PartitionID = partitionID`
	`228`	`+`
	`229`	`+ return pmb`
`224`	`230`	`}`
`225`	`231`
`226`	`232`	`func (pmb PublicMessageBuilder) RawDataLen(val int) PublicMessageBuilder {`