Clean mqueue task now show before/after

openldap-log-parser/cmd/root.go

@@ -125,7 +125,7 @@ var (
 	File   os.File
 	Writer *bufio.Writer
 
-	Version = "0.6.9"
+	Version = "0.6.10"
 
 	BuildInfo = promauto.NewGaugeVec(prometheus.GaugeOpts{
 		Name: "openldaplogparser_build_info",
@@ -321,29 +321,37 @@ func writeOut(msg string, filename string) error {
 	return nil
 }
 
-// Every 24H, remove connections closed more than 24 hours ago
-func periodicallyCleanMQueue(mqueue map[string]*OpenLdapConnection, mqMtx *sync.Mutex) {
-	var ok bool
+func cleanMQueue(mqueue map[string]*OpenLdapConnection, mqMtx *sync.Mutex, age time.Duration) {
+    var ok bool
 
-	for range time.Tick(time.Hour * 24) {
-		// Do we need read lock?
-		for _, ldcon := range mqueue {
-			ok = false
-			// Check if a close operation exist
-			for _, op := range ldcon.Operations {
-				if op.OpType == "close" {
-					if op.Time.Add(time.Hour * 1 * 24).Before(time.Now()) {
-						ok = true
-					}
+	log.Printf("Start cleaning queue task: %d items in queue", len(mqueue))
+
+	// Do we need read lock?
+	for uid, ldcon := range mqueue {
+		ok = false
+		// Check if a close operation exist
+		for _, op := range ldcon.Operations {
+			if op.OpType == "close" {
+				if op.Time.Add(age).Before(time.Now()) {
+					ok = true
 				}
 			}
-			if ok == true {
-				mqMtx.Lock()
-				//delete(mqueue, ldcon.ConnId)
-				delete(mqueue, fmt.Sprintf("%s:%d", ldcon.Hostname, ldcon.ConnId))
-				mqMtx.Unlock()
-			}
 		}
+		if ok == true {
+			mqMtx.Lock()
+			delete(mqueue, uid)
+			mqMtx.Unlock()
+		}
+	}
+	log.Printf("Finished cleaning queue task: %d items in queue", len(mqueue))
+}
+
+
+
+// Every 24H, remove connections closed more than 24 hours ago
+func periodicallyCleanMQueue(mqueue map[string]*OpenLdapConnection, mqMtx *sync.Mutex) {
+	for range time.Tick(time.Hour * 24) {
+		cleanMQueue(mqueue, mqMtx, 24 * time.Hour)
 	}
 }
 
@@ -973,6 +981,17 @@ func processLogs(cmd *cobra.Command, args []string) {
 	// Cleaner thread
 	go periodicallyCleanMQueue(mQueue, &mqMtx)
 
+    // On demand Mqueue cleaning... For debug, dont try this at home, kids!
+/*	sig2 := make(chan os.Signal)
+	signal.Notify(sig2, syscall.SIGUSR2)
+	go func() {
+		for {
+			<-sig2
+			cleanMQueue(mQueue, &mqMtx, 5 * time.Minute)
+		}
+	}()
+*/
+
 	// Initialize Stdin input...
 	if true == strings.EqualFold(gSyslogListenAddress, "do-not-listen") {
 		useStdin = true