Add a check for metrics directory existence.

Previously on startup the program would just quit without stating
explicitly why.

Change-Id: I833b85eb74d2dd27cdc3f0f2e65d7bb1c42caa39
This commit is contained in:
Conor Hennessy 2013-10-10 20:54:06 +02:00
parent 2d26db7e37
commit 9a48010cec
2 changed files with 39 additions and 2 deletions

View File

@ -112,8 +112,12 @@ const (
const watermarkCacheLimit = 1024 * 1024
func NewTieredStorage(appendToDiskQueueDepth, viewQueueDepth uint, flushMemoryInterval, memoryTTL time.Duration, root string) (*TieredStorage, error) {
diskStorage, err := NewLevelDBMetricPersistence(root)
func NewTieredStorage(appendToDiskQueueDepth, viewQueueDepth uint, flushMemoryInterval, memoryTTL time.Duration, rootDirectory string) (*TieredStorage, error) {
if isDir, _ := utility.IsDir(rootDirectory); !isDir {
return nil, fmt.Errorf("Could not find metrics directory %s", rootDirectory)
}
diskStorage, err := NewLevelDBMetricPersistence(rootDirectory)
if err != nil {
return nil, err
}

33
utility/file_ext.go Normal file
View File

@ -0,0 +1,33 @@
// Copyright 2013 Prometheus Team
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// Useful file/filesystem related functions.
package utility
import (
"fmt"
"os"
)
// Returns true iff dirPath is a valid directory path.
func IsDir(dirPath string) (bool, error) {
finfo, err := os.Stat(dirPath)
if err != nil {
return false, err
}
if !finfo.IsDir() {
return false, fmt.Errorf("%s not a directory", dirPath)
}
return true, nil
}