Fix minors

Signed-off-by: Jaehyun Nam <[email protected]>
boanlab · May 10, 2024 · b4675ea · b4675ea
1 parent 16afeaa
commit b4675ea
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/sentryflow/processor/apiClassifier.go b/sentryflow/processor/apiClassifier.go
@@ -61,17 +61,19 @@ func StartAPIClassifier(wg *sync.WaitGroup) bool {
 		return false
 	}
 
-	// Start serving gRPC server
-	log.Printf("[AI] Successfully connected to %s for APIMetrics", AIEngineService)
+	log.Printf("[AI] Connecting to %s for APIMetrics", AIEngineService)
 
 	client := protobuf.NewAPIClassificationClient(conn)
 
+	// Start serving gRPC server
 	aiStream, err := client.ClassifyAPIs(context.Background())
 	if err != nil {
 		log.Fatalf("[AI] Could not make stream: %v", err)
 		return false
 	}
 
+	log.Printf("[AI] Successfully connected to %s for APIMetrics", AIEngineService)
+
 	AH.AIStream = &streamInform{
 		AIStream: aiStream,
 	}