overflow_backup_2018_0455/config_manager_go
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

.

Browse Source
This commit is contained in:
jackdaw@loafle.com 2017-05-10 15:14:31 +09:00
parent 19afde84eb
commit b9730776ab
2 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
config_manager.go
View File

@ -56,6 +56,12 @@ func (c *configManager) getContainerPath() string {
return c.getConfigPath() + "container/" return c.getConfigPath() + "container/"
} }
func (c *configManager) getConfigFilePath(config *config_manager.Config) string {
return c.getContainerPath() + appendSeperator(config.Crawler.Container) + appendSeperator(config.Crawler.Name) + config.Id
}
func appendSeperator(str string) string { func appendSeperator(str string) string {
if strings.LastIndex(str, "/") != len(str)-1 { if strings.LastIndex(str, "/") != len(str)-1 {
return str + "/" return str + "/"
@ -100,6 +106,8 @@ func (c *configManager) loadCrawlerConfig(root string, dir string) {
} }
for _, file := range files { for _, file := range files {
// 디렉터리라면 재귀 / 파일이라면 설정 로드
if file.IsDir() == true { if file.IsDir() == true {
c.loadCrawlerConfig(currentDir, file.Name()) c.loadCrawlerConfig(currentDir, file.Name())
} else { } else {
@ -135,7 +143,7 @@ func (c *configManager) addConfig() {
// agent 폴더 / config / crawler / .. / .. / .. / 에 해당하는 파일이 있는지 확인, 있다면 삭제 // agent 폴더 / config / crawler / .. / .. / .. / 에 해당하는 파일이 있는지 확인, 있다면 삭제
// Config 파일로 저장 // Config 파일로 저장
savePath := c.getContainerPath() + appendSeperator(m.Crawler.Container) + appendSeperator(m.Crawler.Name) + m.Id savePath := c.getConfigFilePath(&m)
ioutil.WriteFile(savePath, b, 0644) ioutil.WriteFile(savePath, b, 0644)
// tempfile remove // tempfile remove
@ -167,7 +175,7 @@ func (c *configManager) removeConfig() {
} }
// 해당 파일 삭제 // 해당 파일 삭제
path := c.getContainerPath() + appendSeperator(config.Crawler.Container) + appendSeperator(config.Crawler.Name) + config.Id path := c.getConfigFilePath(config)
err := os.Remove(path) err := os.Remove(path)
if err != nil { if err != nil {
// error check // error check

4
config_manager_test.go
View File

@ -1,5 +1,3 @@
// +build !test
package config_manager_go package config_manager_go
import ( import (
@ -34,7 +32,7 @@ func TestLoadConfig(t *testing.T) {
assert.NotEqual(t, len(c.configs),0) assert.NotEqual(t, len(c.configs),0)
observer.Notify(messages.CONFIGMANAGER_LOADED,c) observer.Notify(messages.CONFIGMANAGER_LOADED,c)
time.Sleep(2 * time.Second) time.Sleep(1 * time.Second)
} }