Jason Kulatunga
b44ef5cb9c
/scrutiny/config/collector.yaml Adding ability to specify host identifier (label), that is updated on every collector run. Can be specified by `host-id` CLI or `COLLECTOR_HOST_ID` env var. Created a config class, interface and associated tests. Created a "TransformDetectedDrives" function, that will allow users to insert drives not detected by Smarctl --scan, ignore drives that they dont want, and override smartctl device type. Added Upsert functionality when registering devices. Replaced "github.com/jinzhu/gorm" with "gorm.io/gorm" (ORM location moved, was using incorrect lib url) Removed machineid library.
27 lines
744 B
Go
27 lines
744 B
Go
package config
|
|
|
|
import (
|
|
"github.com/analogj/scrutiny/collector/pkg/models"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
// Create mock using:
|
|
// mockgen -source=collector/pkg/config/interface.go -destination=collector/pkg/config/mock/mock_config.go
|
|
type Interface interface {
|
|
Init() error
|
|
ReadConfig(configFilePath string) error
|
|
Set(key string, value interface{})
|
|
SetDefault(key string, value interface{})
|
|
|
|
AllSettings() map[string]interface{}
|
|
IsSet(key string) bool
|
|
Get(key string) interface{}
|
|
GetBool(key string) bool
|
|
GetInt(key string) int
|
|
GetString(key string) string
|
|
GetStringSlice(key string) []string
|
|
UnmarshalKey(key string, rawVal interface{}, decoderOpts ...viper.DecoderConfigOption) error
|
|
|
|
GetScanOverrides() []models.ScanOverride
|
|
}
|