feat: uniq db auto (#26)

* feat: uniq db auto * feat: uniq db auto * feat: uniq db auto
rakutentech · Jan 16, 2025 · f372b5f · f372b5f
1 parent d6b43fc
commit f372b5f
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/pkg/database.go b/pkg/database.go
@@ -2,6 +2,7 @@ package pkg
 
 import (
 	"database/sql"
+	"fmt"
 	"log/slog"
 	"os"
 	"time"
@@ -29,6 +30,12 @@ func InitDB(dbName string) (*sql.DB, error) {
 	return db, nil
 }
 
+func GetUniqDBName(f Flags) string {
+	suffix := Hash(fmt.Sprintf("%s-%s-%s-%d", f.FilePath, f.Match, f.Ignore, f.Every)) + ".sqlite"
+	dbName := f.DBPath + "." + suffix
+	return dbName
+}
+
 func DeleteDB(dbName string) error {
 	if _, err := os.Stat(dbName); err == nil {
 		if err := os.Remove(dbName); err != nil {

diff --git a/pkg/strings.go b/pkg/strings.go
@@ -2,6 +2,7 @@ package pkg
 
 import (
 	"crypto/sha256"
+	"encoding/hex"
 	"log/slog"
 	"sync"
 
@@ -16,7 +17,7 @@ func Hash(s string) string {
 	h := sha256.New()
 	h.Write([]byte(s))
 	bs := h.Sum(nil)
-	return string(bs)
+	return hex.EncodeToString(bs[:3])
 }
 
 func Truncate(s string, n int) string {

diff --git a/pkg/watcher.go b/pkg/watcher.go
@@ -28,7 +28,7 @@ func NewWatcher(
 	filePath string,
 	f Flags,
 ) (*Watcher, error) {
-	dbName := f.DBPath + ".sqlite"
+	dbName := GetUniqDBName(f)
 	db, err := InitDB(dbName)
 	if err != nil {
 		return nil, err