logging initialization in init()

1 files changed, 27 insertions, 0 deletions

diff --git a/init.go b/init.go
new file mode 100644
index 0000000..612bb6d
--- /dev/null
+++ b/init.go
@@ -0,0 +1,27 @@
+package main
+
+import (
+	"log"
+	"os"
+)
+
+// Sets up logging before the main function executes
+func init() {
+	logfile, err := os.OpenFile("getwtxt.log", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0600)
+	if err != nil {
+		log.Printf("Could not open log file: %v\n", err)
+	}
+
+	// Listen for the signal to close the log file
+	go func() {
+		<-closelog
+		log.Printf("Closing log file ...\n")
+		err = logfile.Close()
+		if err != nil {
+			log.Printf("Couldn't close log file: %v\n", err)
+		}
+	}()
+
+	log.SetOutput(logfile)
+
+}