added check span!=nil && conn close

kgo.go

@@ -287,6 +287,7 @@ func (k *Broker) publish(ctx context.Context, msgs []*broker.Message, opts ...br
 			k.Unlock()
 			return err
 		}
+		k.c.Close()
 		k.c = c
 		k.connected.Store(1)
 	}

subscriber.go

@@ -274,11 +274,17 @@ func (pc *consumer) consume() {
 				} else if pc.opts.BodyOnly {
 					p.msg.Body = record.Value
 				} else {
+					if sp != nil {
 						sp.AddEvent("codec unmarshal start")
+					}
 					err := pc.kopts.Codec.Unmarshal(record.Value, p.msg)
+					if sp != nil {
 						sp.AddEvent("codec unmarshal stop")
+					}
 					if err != nil {
+						if sp != nil {
 							sp.SetStatus(tracer.SpanStatusError, err.Error())
+						}
 						pc.kopts.Meter.Counter(semconv.SubscribeMessageTotal, "endpoint", record.Topic, "topic", record.Topic, "status", "failure").Inc()
 						p.err = err
 						p.msg.Body = record.Value
@@ -308,17 +314,25 @@ func (pc *consumer) consume() {
 						eventPool.Put(p)
 						//						pc.connected.Store(0)
 						pc.kopts.Logger.Fatal(pc.kopts.Context, "[kgo] Unmarshal err not handled wtf?")
+						if sp != nil {
 							sp.Finish()
+						}
 						return
 					}
 				}
+				if sp != nil {
 					sp.AddEvent("handler start")
+				}
 				err := pc.handler(p)
+				if sp != nil {
 					sp.AddEvent("handler stop")
+				}
 				if err == nil {
 					pc.kopts.Meter.Counter(semconv.SubscribeMessageTotal, "endpoint", record.Topic, "topic", record.Topic, "status", "success").Inc()
 				} else {
+					if sp != nil {
 						sp.SetStatus(tracer.SpanStatusError, err.Error())
+					}
 					pc.kopts.Meter.Counter(semconv.SubscribeMessageTotal, "endpoint", record.Topic, "topic", record.Topic, "status", "failure").Inc()
 				}
 				pc.kopts.Meter.Counter(semconv.SubscribeMessageInflight, "endpoint", record.Topic, "topic", record.Topic).Dec()
@@ -327,9 +341,13 @@ func (pc *consumer) consume() {
 				} else if err != nil {
 					p.err = err
 					if eh != nil {
+						if sp != nil {
 							sp.AddEvent("error handler start")
+						}
 						_ = eh(p)
+						if sp != nil {
 							sp.AddEvent("error handler stop")
+						}
 					} else {
 						if pc.kopts.Logger.V(logger.ErrorLevel) {
 							pc.kopts.Logger.Error(pc.kopts.Context, "[kgo]: subscriber error", err)
@@ -346,12 +364,16 @@ func (pc *consumer) consume() {
 					eventPool.Put(p)
 					//					pc.connected.Store(0)
 					pc.kopts.Logger.Fatal(pc.kopts.Context, "[kgo] ErrLostMessage wtf?")
+					if sp != nil {
 						sp.SetStatus(tracer.SpanStatusError, "ErrLostMessage")
 						sp.Finish()
+					}
 					return
 				}
+				if sp != nil {
 					sp.Finish()
 				}
 			}
 		}
 	}
+}