✨ Add once warpper with error

pkg/utils/utils.go

@@ -21,6 +21,7 @@ import (
 	"os"
 	"strconv"
 	"strings"
+	"sync"
 
 	"github.com/jinzhu/copier"
 	"github.com/labstack/echo/v4"
@@ -225,3 +226,19 @@ func GetUserFromCtxForDs(c echo.Context) (*models.User, bool, error) {
 	}
 	return user, false, nil
 }
+
+// OnceWithErr ...
+func OnceWithErr(once *sync.Once, fn func() error) error {
+	var errChan = make(chan error, 1)
+	defer close(errChan)
+	once.Do(func() {
+		defer func() {
+			if r := recover(); r != nil {
+				errChan <- fmt.Errorf("%v", r)
+			}
+		}()
+		err := fn()
+		errChan <- err
+	})
+	return <-errChan
+}

pkg/utils/utils_test.go

@@ -22,6 +22,7 @@ import (
 	"net/http/httptest"
 	"reflect"
 	"strings"
+	"sync"
 	"testing"
 
 	"github.com/labstack/echo/v4"
@@ -560,3 +561,44 @@ func TestGetUserFromCtxForDs(t *testing.T) {
 		})
 	}
 }
+
+func TestOnceWithErr(t *testing.T) {
+	type args struct {
+		once *sync.Once
+		fn   func() error
+	}
+	tests := []struct {
+		name    string
+		args    args
+		wantErr bool
+	}{
+		{
+			name: "normal",
+			args: args{
+				once: &sync.Once{},
+				fn: func() error {
+					fmt.Println("do something")
+					return nil
+				},
+			},
+			wantErr: false,
+		},
+		{
+			name: "panic",
+			args: args{
+				once: &sync.Once{},
+				fn: func() error {
+					panic("panic something")
+				},
+			},
+			wantErr: true,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if err := OnceWithErr(tt.args.once, tt.args.fn); (err != nil) != tt.wantErr {
+				t.Errorf("OnceWithErr() error = %v, wantErr %v", err, tt.wantErr)
+			}
+		})
+	}
+}