You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
86 lines
1.6 KiB
86 lines
1.6 KiB
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"git.binarythought.com/cdramey/alrm/alarm"
|
|
"runtime"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
type Config struct {
|
|
Groups map[string]*Group
|
|
Alarms map[string]alarm.Alarm
|
|
Interval time.Duration
|
|
Threads int
|
|
}
|
|
|
|
func NewConfig() *Config {
|
|
return &Config{
|
|
// Default check interval, 30 seconds
|
|
Interval: time.Second * 30,
|
|
// Default number of threads, use local CPU count
|
|
Threads: runtime.NumCPU(),
|
|
}
|
|
}
|
|
|
|
func (c *Config) NewAlarm(name string, typename string) (alarm.Alarm, error) {
|
|
if c.Alarms == nil {
|
|
c.Alarms = make(map[string]alarm.Alarm)
|
|
}
|
|
|
|
if _, exists := c.Alarms[name]; exists {
|
|
return nil, fmt.Errorf("alarm %s already exists", name)
|
|
}
|
|
|
|
a, err := alarm.NewAlarm(name, typename)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
c.Alarms[name] = a
|
|
|
|
return a, nil
|
|
}
|
|
|
|
func (c *Config) NewGroup(name string) (*Group, error) {
|
|
if c.Groups == nil {
|
|
c.Groups = make(map[string]*Group)
|
|
}
|
|
|
|
if _, exists := c.Groups[name]; exists {
|
|
return nil, fmt.Errorf("group %s already exists", name)
|
|
}
|
|
|
|
group := &Group{Name: name}
|
|
c.Groups[name] = group
|
|
return group, nil
|
|
}
|
|
|
|
func (c *Config) SetInterval(val string) error {
|
|
interval, err := time.ParseDuration(val)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
c.Interval = interval
|
|
return nil
|
|
}
|
|
|
|
func (c *Config) SetThreads(val string) error {
|
|
threads, err := strconv.Atoi(val)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
c.Threads = threads
|
|
return nil
|
|
}
|
|
|
|
func ReadConfig(fn string, debuglvl int) (*Config, error) {
|
|
parser := &Parser{DebugLevel: debuglvl}
|
|
config, err := parser.Parse(fn)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return config, nil
|
|
}
|