// InitPluginGenerators XXX
func (agent *Agent) InitPluginGenerators(api *mackerel.API) {
payloads := agent.CollectGraphDefsOfPlugins()
if len(payloads) > 0 {
err := api.CreateGraphDefs(payloads)
if err != nil {
logger.Errorf("Failed to create graphdefs: %s", err)
}
}
}
// prepareCustomIdentiferHosts collects the host information based on the
// configuration of the custom_identifier fields.
func prepareCustomIdentiferHosts(conf *config.Config, api *mackerel.API) map[string]*mackerel.Host {
customIdentifierHosts := make(map[string]*mackerel.Host)
customIdentifiers := make(map[string]bool) // use a map to make them unique
for _, pluginConfigs := range conf.Plugin {
for _, pluginConfig := range pluginConfigs {
if pluginConfig.CustomIdentifier != nil {
customIdentifiers[*pluginConfig.CustomIdentifier] = true
}
}
}
for customIdentifier := range customIdentifiers {
host, err := api.FindHostByCustomIdentifier(customIdentifier)
if err != nil {
logger.Warningf("Failed to retrieve the host of custom_identifier: %s, %s", customIdentifier, err)
continue
}
customIdentifierHosts[customIdentifier] = host
}
return customIdentifierHosts
}
// prepareHost collects specs of the host and sends them to Mackerel server.
// A unique host-id is returned by the server if one is not specified.
func prepareHost(conf *config.Config, api *mackerel.API) (*mackerel.Host, error) {
// XXX this configuration should be moved to under spec/linux
os.Setenv("PATH", "/sbin:/usr/sbin:/bin:/usr/bin:"+os.Getenv("PATH"))
os.Setenv("LANG", "C") // prevent changing outputs of some command, e.g. ifconfig.
doRetry := func(f func() error) {
retry.Retry(retryNum, retryInterval, f)
}
filterErrorForRetry := func(err error) error {
if err != nil {
logger.Warningf("%s", err.Error())
}
if apiErr, ok := err.(*mackerel.Error); ok && apiErr.IsClientError() {
// don't retry when client error (APIKey error etc.) occurred
return nil
}
return err
}
hostname, meta, interfaces, lastErr := collectHostSpecs()
if lastErr != nil {
return nil, fmt.Errorf("error while collecting host specs: %s", lastErr.Error())
}
var result *mackerel.Host
if hostID, err := conf.LoadHostID(); err != nil { // create
logger.Debugf("Registering new host on mackerel...")
doRetry(func() error {
hostID, lastErr = api.CreateHost(hostname, meta, interfaces, conf.Roles, conf.DisplayName)
return filterErrorForRetry(lastErr)
})
if lastErr != nil {
return nil, fmt.Errorf("Failed to register this host: %s", lastErr.Error())
}
doRetry(func() error {
result, lastErr = api.FindHost(hostID)
return filterErrorForRetry(lastErr)
})
if lastErr != nil {
return nil, fmt.Errorf("Failed to find this host on mackerel: %s", lastErr.Error())
}
} else { // check the hostID is valid or not
doRetry(func() error {
result, lastErr = api.FindHost(hostID)
return filterErrorForRetry(lastErr)
})
if lastErr != nil {
if fsStorage, ok := conf.HostIDStorage.(*config.FileSystemHostIDStorage); ok {
return nil, fmt.Errorf("Failed to find this host on mackerel (You may want to delete file \"%s\" to register this host to an another organization): %s", fsStorage.HostIDFile(), lastErr.Error())
}
return nil, fmt.Errorf("Failed to find this host on mackerel: %s", lastErr.Error())
}
}
hostSt := conf.HostStatus.OnStart
if hostSt != "" && hostSt != result.Status {
doRetry(func() error {
lastErr = api.UpdateHostStatus(result.ID, hostSt)
return filterErrorForRetry(lastErr)
})
if lastErr != nil {
return nil, fmt.Errorf("Failed to set default host status: %s, %s", hostSt, lastErr.Error())
}
}
lastErr = conf.SaveHostID(result.ID)
if lastErr != nil {
return nil, fmt.Errorf("Failed to save host ID: %s", lastErr.Error())
}
return result, nil
}