test

internal/msg_transfer/logic/offline_history_msg_handler.go

@@ -98,23 +98,23 @@ func (OfflineHistoryConsumerHandler) Setup(_ sarama.ConsumerGroupSession) error
 func (OfflineHistoryConsumerHandler) Cleanup(_ sarama.ConsumerGroupSession) error { return nil }
 func (mc *OfflineHistoryConsumerHandler) ConsumeClaim(sess sarama.ConsumerGroupSession,
 	claim sarama.ConsumerGroupClaim) error { // a instance in the consumer group
-	log.NewDebug("", "offline new session msg come", claim.HighWaterMarkOffset(), claim.Topic(), claim.Partition())
-	for msg := range claim.Messages() {
-		log.NewDebug("", "kafka get info to delay mongo", "msgTopic", msg.Topic, "msgPartition", msg.Partition, "offline")
-		//mc.msgHandle[msg.Topic](msg.Value, string(msg.Key))
-	}
-	//select {
-	//case cmd := <-mc.cmdCh:
-	//	if cmd.Cmd == OnlineTopicVacancy {
-	//		for msg := range claim.Messages() {
-	//			if GetOnlineTopicStatus() == OnlineTopicVacancy {
-	//				log.NewDebug("", "kafka get info to mongo", "msgTopic", msg.Topic, "msgPartition", msg.Partition, "msg", string(msg.Value))
-	//				mc.msgHandle[msg.Topic](msg.Value, string(msg.Key))
-	//				sess.MarkMessage(msg, "")
-	//			}
-	//		}
-	//	}
-	//
+	//log.NewDebug("", "offline new session msg come", claim.HighWaterMarkOffset(), claim.Topic(), claim.Partition())
+	//for msg := range claim.Messages() {
+	//	log.NewDebug("", "kafka get info to delay mongo", "msgTopic", msg.Topic, "msgPartition", msg.Partition, "offline")
+	//	//mc.msgHandle[msg.Topic](msg.Value, string(msg.Key))
 	//}
+	select {
+	case cmd := <-mc.cmdCh:
+		if cmd.Cmd == OnlineTopicVacancy {
+			for msg := range claim.Messages() {
+				if GetOnlineTopicStatus() == OnlineTopicVacancy {
+					log.NewDebug("", "kafka get info to mongo", "msgTopic", msg.Topic, "msgPartition", msg.Partition, "msg", string(msg.Value))
+					mc.msgHandle[msg.Topic](msg.Value, string(msg.Key))
+					sess.MarkMessage(msg, "")
+				}
+			}
+		}
+
+	}
 	return nil
 }

internal/msg_transfer/logic/online_history_msg_handler.go

@@ -134,9 +134,9 @@ func (OnlineHistoryConsumerHandler) Cleanup(_ sarama.ConsumerGroupSession) error
 func (och *OnlineHistoryConsumerHandler) ConsumeClaim(sess sarama.ConsumerGroupSession,
 	claim sarama.ConsumerGroupClaim) error { // a instance in the consumer group
 	log.NewDebug("", "online new session msg come", claim.HighWaterMarkOffset(), claim.Topic(), claim.Partition())
-	och.TriggerCmd(OnlineTopicBusy)
-	SetOnlineTopicStatus(OnlineTopicBusy)
 	for msg := range claim.Messages() {
+		och.TriggerCmd(OnlineTopicBusy)
+		SetOnlineTopicStatus(OnlineTopicBusy)
 		log.NewDebug("", "kafka get info to mongo", "msgTopic", msg.Topic, "msgPartition", msg.Partition, "online")
 		och.msgHandle[msg.Topic](msg.Value, string(msg.Key))
 		sess.MarkMessage(msg, "")