// SystemURL - XXX
func SystemURL() string {
settings := settings.Settings()
url := settings.AmonInstance + "/api/system/v2/?api_key=" + settings.APIKey
return url
}
// CollectHostData - XXX
func CollectHostData() HostDataStruct {
host := Host()
// Load settings
settings := settings.Settings()
var machineID string
var InstanceID string
var ip string
var distro DistroStruct
var wg sync.WaitGroup
wg.Add(4)
go func() {
defer wg.Done()
machineID = MachineID()
}()
go func() {
defer wg.Done()
InstanceID = CloudID()
}()
go func() {
defer wg.Done()
ip = IPAddress()
}()
go func() {
defer wg.Done()
distro = Distro()
}()
wg.Wait()
hoststruct := HostDataStruct{
Host: host,
MachineID: machineID,
Distro: distro,
IPAddress: ip,
ServerKey: settings.ServerKey,
InstanceID: InstanceID,
}
return hoststruct
}
func main() {
flag.Parse()
if *fListPlugins {
ListPlugins()
return
}
if len(*fPluginConfig) > 0 {
pluginConfig, _ := plugins.GetConfigPath(*fPluginConfig)
creator, ok := plugins.Plugins[pluginConfig.Name]
if ok {
plugin := creator()
conf := plugin.SampleConfig()
fmt.Println(conf)
} else {
fmt.Printf("Non existing plugin: %s", pluginConfig.Name)
ListPlugins()
}
return
}
if len(*fTestPlugin) > 0 {
pluginConfig, _ := plugins.GetConfigPath(*fTestPlugin)
creator, ok := plugins.Plugins[pluginConfig.Name]
if ok {
plugin := creator()
PluginResult, err := plugin.Collect(pluginConfig.Path)
if err != nil {
fmt.Printf("Can't get stats for plugin: %s", err)
}
fmt.Println(PluginResult)
} else {
fmt.Printf("Non existing plugin: %s", pluginConfig.Name)
ListPlugins()
}
return
}
if *fVersion {
v := fmt.Sprintf("Amon - Version %s", Version)
fmt.Println(v)
return
}
config := settings.Settings()
// Detect Machine ID or ask for a valid Server Key in Settings
machineID := collectors.MachineID()
serverKey := config.ServerKey
ag, err := amonagent.NewAgent(config)
if err != nil {
log.Fatal(err)
}
if *fTest {
err = ag.Test(config)
if err != nil {
log.Fatal(err)
}
return
}
if *fMachineID {
fmt.Print(machineID)
return
}
if len(machineID) == 0 && len(serverKey) == 0 {
log.Fatal("Can't detect Machine ID. Please define `server_key` in /etc/opt/amonagent/amonagent.conf ")
}
shutdown := make(chan struct{})
signals := make(chan os.Signal)
signal.Notify(signals, os.Interrupt)
go func() {
<-signals
close(shutdown)
}()
log.Printf("Starting Amon Agent (version %s)\n", Version)
if *fPidfile != "" {
// Ensure the required directory structure exists.
err := os.MkdirAll(filepath.Dir(*fPidfile), 0700)
if err != nil {
log.Fatal(3, "Failed to verify pid directory", err)
}
f, err := os.Create(*fPidfile)
if err != nil {
log.Fatalf("Unable to create pidfile: %s", err)
}
fmt.Fprintf(f, "%d\n", os.Getpid())
f.Close()
}
ag.Run(shutdown)
}