ugly fix for nil context

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
2020-10-14 14:47:04 +03:00 · 2020-10-14 14:47:04 +03:00 · 553fc8b998
commit 553fc8b998
parent 28190f9489
1 changed files with 4 additions and 2 deletions
--- a/segmentio.go
+++ b/segmentio.go
@ -177,8 +177,10 @@ func (k *kBroker) Publish(topic string, msg *broker.Message, opts ...broker.Publ
 		return err
 	}
 	kmsg := kafka.Message{Value: val}
-	if key, ok := options.Context.Value(publishKey{}).([]byte); ok && len(key) > 0 {
-		kmsg.Key = key
+	if options.Context != nil {
+		if key, ok := options.Context.Value(publishKey{}).([]byte); ok && len(key) > 0 {
+			kmsg.Key = key
+		}
 	}

 	k.Lock()