Feature: add persistance of queue elements

Author: ThomasBoom89

internal/recover.go

@@ -14,6 +14,13 @@ func GetWebsocketPanicHandler() func(conn *websocket.Conn) {
 	}
 }
 
+func SaveOnPanic(storage *Storage) {
+	if err := recover(); err != nil {
+		storage.Save()
+		HandlePanic(err)
+	}
+}
+
 func RecoverGoroutine(function func()) {
 	if err := recover(); err != nil {
 		HandlePanic(err)

internal/storage.go

@@ -0,0 +1,58 @@
+package internal
+
+import (
+	"bufio"
+	"os"
+)
+
+type Storage struct {
+	queue Queue[string]
+}
+
+func NewStorage(queue Queue[string]) *Storage {
+	return &Storage{
+		queue: queue,
+	}
+}
+
+func (S *Storage) clear() {
+	err := os.WriteFile("./data", []byte(""), 0644)
+	if err != nil {
+		panic(err)
+	}
+}
+
+func (S *Storage) Save() {
+	S.clear()
+
+	file, err := os.OpenFile("./data", os.O_RDWR|os.O_CREATE, 0644)
+	if err != nil {
+		panic(err)
+	}
+	defer file.Close()
+	for !S.queue.IsEmpty() {
+		value, _ := S.queue.Dequeue()
+		//err := os.OpenFile("./data", []byte(value+"\n"), 0644)
+		_, err = file.WriteString(value + "\n")
+		if err != nil {
+			panic(err)
+		}
+	}
+}
+
+func (S *Storage) Load() {
+	file, err := os.OpenFile("./data", os.O_RDONLY, 0644)
+	if err != nil {
+		return
+	}
+	defer file.Close()
+
+	scanner := bufio.NewScanner(file)
+	for scanner.Scan() {
+		text := scanner.Text()
+		if text == "" {
+			continue
+		}
+		S.queue.Enqueue(text)
+	}
+}

internal/system.go

@@ -0,0 +1,20 @@
+package internal
+
+import (
+	"github.com/gofiber/fiber/v2/log"
+	"os"
+	"os/signal"
+)
+
+func HandleOsSignal(storage *Storage) {
+	sigchan := make(chan os.Signal, 1)
+	signals := []os.Signal{os.Kill, os.Interrupt}
+	signal.Notify(sigchan, signals...)
+	<-sigchan
+
+	storage.Save()
+
+	log.Debug("os signal received")
+
+	os.Exit(0)
+}

src/main.go

@@ -11,6 +11,12 @@ import (
 
 func main() {
 
+	queue := internal.NewLinkedListQueue[string]()
+	storage := internal.NewStorage(queue)
+	storage.Load()
+	defer internal.SaveOnPanic(storage)
+	go internal.HandleOsSignal(storage)
+
 	app := fiber.New(fiber.Config{
 		// Override default error handler
 		ErrorHandler: func(ctx *fiber.Ctx, err error) error {
@@ -22,7 +28,6 @@ func main() {
 	app.Use(logger.New())
 	app.Use(cors.New())
 
-	queue := internal.NewLinkedListQueue[string]()
 	http := internal.NewHTTP(app, queue)
 	http.SetupRoutes()