From 4cb2e42ea00b965dac2f5b7d90b300256b9669ee Mon Sep 17 00:00:00 2001
From: Ron Hao <919263118@qq.com>
Date: Thu, 14 Mar 2024 09:44:51 +0800
Subject: [PATCH] Init dag dir (#537)

---
 internal/persistence/client/store_factory.go | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/internal/persistence/client/store_factory.go b/internal/persistence/client/store_factory.go
index 92ff19d1b..9f7d6931a 100644
--- a/internal/persistence/client/store_factory.go
+++ b/internal/persistence/client/store_factory.go
@@ -1,6 +1,8 @@
 package client
 
 import (
+	"os"
+
 	"github.com/dagu-dev/dagu/internal/config"
 	"github.com/dagu-dev/dagu/internal/persistence"
 	"github.com/dagu-dev/dagu/internal/persistence/jsondb"
@@ -15,9 +17,22 @@ type dataStoreFactoryImpl struct {
 var _ persistence.DataStoreFactory = (*dataStoreFactoryImpl)(nil)
 
 func NewDataStoreFactory(cfg *config.Config) persistence.DataStoreFactory {
-	return &dataStoreFactoryImpl{
+	ds := &dataStoreFactoryImpl{
 		cfg: cfg,
 	}
+	_ = ds.InitDagDir()
+	return ds
+}
+
+func (f dataStoreFactoryImpl) InitDagDir() error {
+	_, err := os.Stat(f.cfg.DAGs)
+	if os.IsNotExist(err) {
+		if err := os.MkdirAll(f.cfg.DAGs, 0755); err != nil {
+			return err
+		}
+	}
+
+	return nil
 }
 
 func (f dataStoreFactoryImpl) NewHistoryStore() persistence.HistoryStore {