优化：启动队列前检查数据目录是否存在

Author: yoonper

README.md

@@ -28,7 +28,10 @@ import (
 func main() {
 	// start
 	diskqueue.Config.Path = "/tmp"
-	queue := diskqueue.Start()
+	queue, err := diskqueue.Start()
+	if err != nil {
+		log.Fatalln(err)
+	}
 
 	// write data
 	err := queue.Write([]byte("data"))

diskqueue.go

@@ -3,6 +3,7 @@ package diskqueue
 import (
 	"errors"
 	"io"
+	"os"
 	"sync"
 	"time"
 )
@@ -30,7 +31,11 @@ var (
 )
 
 // Start diskqueue
-func Start() *Diskqueue {
+func Start() (*Diskqueue, error) {
+	if _, err := os.Stat(Config.Path); err != nil {
+		return nil, err
+	}
+
 	queue := &Diskqueue{close: false}
 	queue.ticker = time.NewTicker(Config.BatchTime)
 	Reader.restore()
@@ -44,7 +49,8 @@ func Start() *Diskqueue {
 			queue.Unlock()
 		}
 	}()
-	return queue
+
+	return queue, nil
 }
 
 // Write data