Merge pull request #9 from calyptia/fix-segfault

fix segfault when thread is finalised without golang's runtime gc as …

cshared.go

@@ -12,6 +12,7 @@ import (
 	"log"
 	"os"
 	"reflect"
+	"runtime"
 	"time"
 	"unsafe"
 
@@ -124,6 +125,10 @@ func FLBPluginInputCallback(data *unsafe.Pointer, csize *C.size_t) int {
 			fmt.Fprintf(os.Stderr, "run: %s\n", err)
 			return input.FLB_ERROR
 		}
+		// enforce a runtime gc, to prevent the thread finalizer on
+		// fluent-bit to kick in before any remaining data has not been GC'ed
+		// causing a sigsegv.
+		defer runtime.GC()
 	}
 
 	return input.FLB_OK
@@ -252,17 +257,17 @@ func FLBPluginFlush(data unsafe.Pointer, clength C.int, ctag *C.char) int {
 func FLBPluginExit() int {
 	log.Printf("calling FLBPluginExit(): name=%q\n", theName)
 
+	if unregister != nil {
+		unregister()
+	}
+
 	if runCancel != nil {
 		runCancel()
 	}
 
-	// if unregister != nil {
-	// 	unregister()
-	// }
-
-	// if theChannel != nil {
-	// 	defer close(theChannel)
-	// }
+	if theChannel != nil {
+		defer close(theChannel)
+	}
 
 	return input.FLB_OK
 }

plugin.go

@@ -45,7 +45,7 @@ type ConfigLoader interface {
 
 type Message struct {
 	Time   time.Time
-	Record map[string]string
+	Record any
 	tag    *string
 }