depth 기능 config에 추가

seungyongshim · May 17, 2020 · 44b62d0 · 44b62d0
1 parent 81d99cf
commit 44b62d0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/beater/lsbeat.go b/beater/lsbeat.go
@@ -21,7 +21,7 @@ type Lsbeat struct {
 	client        beat.Client
 	period        time.Duration
 	paths         []string
-	smbDrives     []string
+	depth         int
 	lastIndexTime time.Time // 가장 마지막 검색한 시간
 }
 
@@ -38,7 +38,7 @@ func New(b *beat.Beat, cfg *common.Config) (beat.Beater, error) {
 	}
 
 	bt.paths = c.Paths
-	bt.smbDrives = c.SmbDrives
+	bt.depth = c.Depth
 
 	return bt, nil
 }
@@ -61,7 +61,7 @@ func (bt *Lsbeat) Run(b *beat.Beat) error {
 				continue
 			}
 			path = s.Replace(path, "\\", "/", -1)
-			listDir(path, bt, b, 1)
+			listDir(path, bt, b, bt.depth)
 		}
 
 		select {

diff --git a/config/config.go b/config/config.go
@@ -8,9 +8,9 @@ import (
 )
 
 type Config struct {
-	Period    time.Duration `config:"period"`
-	Paths     []string      `config:"paths"`
-	SmbDrives []string      `config:"smb_drives"`
+	Period time.Duration `config:"period"`
+	Paths  []string      `config:"paths"`
+	Depth  int           `config:"depth"`
 }
 
 var (