ing

kafka/kafka.go

@@ -10,8 +10,8 @@ import (
 )
 
 var (
-	kafkaWriters map[string]*kafka.Writer
-	kafkaReaders map[string]*kafka.Reader
+	writers map[string]*kafka.Writer
+	readers map[string]*kafka.Reader
 )
 
 func InitPackage(config *occe.Kafka) {
@@ -20,7 +20,7 @@ func InitPackage(config *occe.Kafka) {
 	}
 
 	if nil != config.Producers && 0 < len(config.Producers) {
-		kafkaWriters = make(map[string]*kafka.Writer)
+		writers = make(map[string]*kafka.Writer)
 		for n, c := range config.Producers {
 			wc := kafka.WriterConfig{
 				Brokers:           c.Brokers,
@@ -36,12 +36,12 @@ func InitPackage(config *occe.Kafka) {
 				RequiredAcks:      c.RequiredAcks,
 				Async:             c.Async,
 			}
-			kafkaWriters[n] = kafka.NewWriter(wc)
+			writers[n] = kafka.NewWriter(wc)
 		}
 	}
 
 	if nil != config.Consumers && 0 < len(config.Consumers) {
-		kafkaReaders = make(map[string]*kafka.Reader)
+		readers = make(map[string]*kafka.Reader)
 		for n, c := range config.Consumers {
 			rc := kafka.ReaderConfig{
 				Brokers:           c.Brokers,
@@ -58,7 +58,7 @@ func InitPackage(config *occe.Kafka) {
 				RebalanceTimeout:  c.RebalanceTimeout,
 				RetentionTime:     c.RetentionTime,
 			}
-			kafkaReaders[n] = kafka.NewReader(rc)
+			readers[n] = kafka.NewReader(rc)
 		}
 	}
 }
@@ -82,13 +82,13 @@ func DestroyPackage(config *occe.Kafka) {
 		return
 	}
 
-	for _, w := range kafkaWriters {
+	for _, w := range writers {
 		if err := w.Close(); nil != err {
 			logging.Logger().Errorf("%v", err)
 		}
 	}
 
-	for _, r := range kafkaReaders {
+	for _, r := range readers {
 		if err := r.Close(); nil != err {
 			logging.Logger().Errorf("%v", err)
 		}
@@ -96,25 +96,25 @@ func DestroyPackage(config *occe.Kafka) {
 }
 
 func Writer(name string) *kafka.Writer {
-	if nil == kafkaWriters {
+	if nil == writers {
 		return nil
 	}
-	return kafkaWriters[name]
+	return writers[name]
 }
 
 func Reader(name string) *kafka.Reader {
-	if nil == kafkaReaders {
+	if nil == readers {
 		return nil
 	}
-	return kafkaReaders[name]
+	return readers[name]
 }
 
 func Write(name string, key []byte, value []byte) error {
-	if nil == kafkaWriters {
+	if nil == writers {
 		return fmt.Errorf("Kafka client is not valid")
 	}
 
-	w, ok := kafkaWriters[name]
+	w, ok := writers[name]
 	if !ok {
 		return fmt.Errorf("Kafka client is not valid")
 	}