Merge pull request #132 from uw-labs/discardable_freezer

mjgarton · web-flow · commit f5a9d4c10680 · 2020-08-06T13:38:22.000+01:00
Make freezer messages discardable
diff --git a/freezer/freezer.go b/freezer/freezer.go
@@ -197,5 +197,12 @@ type consumerMessage struct {
 }
 
 func (cm *consumerMessage) Data() []byte {
+	if cm.data == nil {
+		panic("attempt to use payload after discarding.")
+	}
 	return cm.data
 }
+
+func (cm *consumerMessage) DiscardPayload() {
+	cm.data = nil
+}

Original file line number	Diff line number	Diff line change
`@@ -197,5 +197,12 @@ type consumerMessage struct {`
`197`	`197`	`}`
`198`	`198`
`199`	`199`	`func (cm *consumerMessage) Data() []byte {`
	`200`	`+ if cm.data == nil {`
	`201`	`+ panic("attempt to use payload after discarding.")`
	`202`	`+ }`
`200`	`203`	`return cm.data`
`201`	`204`	`}`
	`205`	`+`
	`206`	`+func (cm *consumerMessage) DiscardPayload() {`
	`207`	`+ cm.data = nil`
	`208`	`+}`