-
Notifications
You must be signed in to change notification settings - Fork 2
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Initial support for Icinga Notifications
Inspired by the existing code for the Icinga DB, support for Icinga Notifications was added. Thus, there might be some level of code duplication between those two. The custom Icinga 2 configuration was sourced from the Icinga Notifications repository, but edited to not being parsed as a faulty Go template.
- Loading branch information
Showing
12 changed files
with
674 additions
and
23 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,190 @@ | ||
| package notifications | ||
|
|
||
| import ( | ||
| "context" | ||
| "fmt" | ||
| "github.com/docker/docker/api/types" | ||
| "github.com/docker/docker/api/types/container" | ||
| "github.com/docker/docker/api/types/mount" | ||
| "github.com/docker/docker/api/types/network" | ||
| "github.com/docker/docker/client" | ||
| "github.com/icinga/icinga-testing/services" | ||
| "github.com/icinga/icinga-testing/utils" | ||
| "go.uber.org/zap" | ||
| "os" | ||
| "sync" | ||
| "sync/atomic" | ||
| ) | ||
|
|
||
| type dockerCreator struct { | ||
| logger *zap.Logger | ||
| dockerClient *client.Client | ||
| dockerNetworkId string | ||
| containerNamePrefix string | ||
| sharedDirPath string | ||
| containerCounter uint32 | ||
|
|
||
| runningMutex sync.Mutex | ||
| running map[*dockerBinaryInstance]struct{} | ||
| } | ||
|
|
||
| var _ Creator = (*dockerCreator)(nil) | ||
|
|
||
| func NewDockerCreator( | ||
| logger *zap.Logger, | ||
| dockerClient *client.Client, | ||
| containerNamePrefix string, | ||
| dockerNetworkId string, | ||
| sharedDirPath string, | ||
| ) Creator { | ||
| return &dockerCreator{ | ||
| logger: logger.With(zap.Bool("icinga_notifications", true)), | ||
| dockerClient: dockerClient, | ||
| dockerNetworkId: dockerNetworkId, | ||
| containerNamePrefix: containerNamePrefix, | ||
| sharedDirPath: sharedDirPath, | ||
| running: make(map[*dockerBinaryInstance]struct{}), | ||
| } | ||
| } | ||
|
|
||
| func (i *dockerCreator) CreateIcingaNotifications( | ||
| rdb services.RelationalDatabase, | ||
| options ...services.IcingaNotificationsOption, | ||
| ) services.IcingaNotificationsBase { | ||
| inst := &dockerBinaryInstance{ | ||
| info: info{ | ||
| rdb: rdb, | ||
| port: defaultPort, | ||
| }, | ||
| logger: i.logger, | ||
| icingaNotificationsDockerBinary: i, | ||
| } | ||
|
|
||
| configFile, err := os.CreateTemp("", "icinga_notifications.yml") | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
| err = configFile.Chmod(0666) // defaults to 0600, might result in being unreadable from container UID 1000 | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
| idb := &services.IcingaNotifications{IcingaNotificationsBase: inst} | ||
| for _, option := range options { | ||
| option(idb) | ||
| } | ||
| if err = idb.WriteConfig(configFile); err != nil { | ||
| panic(err) | ||
| } | ||
| inst.configFileName = configFile.Name() | ||
| err = configFile.Close() | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
|
|
||
| containerName := fmt.Sprintf("%s-%d", i.containerNamePrefix, atomic.AddUint32(&i.containerCounter, 1)) | ||
| inst.logger = inst.logger.With(zap.String("container-name", containerName)) | ||
| networkName, err := utils.DockerNetworkName(context.Background(), i.dockerClient, i.dockerNetworkId) | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
|
|
||
| dockerImage := utils.GetEnvDefault("ICINGA_TESTING_NOTIFICATIONS_IMAGE", "icinga-notifications:latest") | ||
| err = utils.DockerImagePull(context.Background(), inst.logger, i.dockerClient, dockerImage, false) | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
|
|
||
| cont, err := i.dockerClient.ContainerCreate(context.Background(), &container.Config{ | ||
| Image: dockerImage, | ||
| }, &container.HostConfig{ | ||
| Mounts: []mount.Mount{{ | ||
| Type: mount.TypeBind, | ||
| Source: inst.configFileName, | ||
| Target: "/etc/icinga-notifications/config.yml", | ||
| ReadOnly: true, | ||
| }, { | ||
| Type: mount.TypeBind, | ||
| Source: i.sharedDirPath, | ||
| Target: "/shared", | ||
| ReadOnly: true, | ||
| }}, | ||
| }, &network.NetworkingConfig{ | ||
| EndpointsConfig: map[string]*network.EndpointSettings{ | ||
| networkName: { | ||
| NetworkID: i.dockerNetworkId, | ||
| }, | ||
| }, | ||
| }, nil, containerName) | ||
| if err != nil { | ||
| inst.logger.Fatal("failed to create icinga-notifications container", zap.Error(err)) | ||
| } | ||
| inst.containerId = cont.ID | ||
| inst.logger = inst.logger.With(zap.String("container-id", cont.ID)) | ||
| inst.logger.Debug("created container") | ||
|
|
||
| err = utils.ForwardDockerContainerOutput(context.Background(), i.dockerClient, cont.ID, | ||
| false, utils.NewLineWriter(func(line []byte) { | ||
| inst.logger.Debug("container output", | ||
| zap.ByteString("line", line)) | ||
| })) | ||
| if err != nil { | ||
| inst.logger.Fatal("failed to attach to container output", zap.Error(err)) | ||
| } | ||
|
|
||
| err = i.dockerClient.ContainerStart(context.Background(), cont.ID, types.ContainerStartOptions{}) | ||
| if err != nil { | ||
| inst.logger.Fatal("failed to start container", zap.Error(err)) | ||
| } | ||
| inst.logger.Debug("started container") | ||
|
|
||
| inst.info.host = utils.MustString(utils.DockerContainerAddress(context.Background(), i.dockerClient, cont.ID)) | ||
|
|
||
| i.runningMutex.Lock() | ||
| i.running[inst] = struct{}{} | ||
| i.runningMutex.Unlock() | ||
|
|
||
| return inst | ||
| } | ||
|
|
||
| func (i *dockerCreator) Cleanup() { | ||
| i.runningMutex.Lock() | ||
| instances := make([]*dockerBinaryInstance, 0, len(i.running)) | ||
| for inst := range i.running { | ||
| instances = append(instances, inst) | ||
| } | ||
| i.runningMutex.Unlock() | ||
|
|
||
| for _, inst := range instances { | ||
| inst.Cleanup() | ||
| } | ||
| } | ||
|
|
||
| type dockerBinaryInstance struct { | ||
| info | ||
| icingaNotificationsDockerBinary *dockerCreator | ||
| logger *zap.Logger | ||
| containerId string | ||
| configFileName string | ||
| } | ||
|
|
||
| var _ services.IcingaNotificationsBase = (*dockerBinaryInstance)(nil) | ||
|
|
||
| func (i *dockerBinaryInstance) Cleanup() { | ||
| i.icingaNotificationsDockerBinary.runningMutex.Lock() | ||
| delete(i.icingaNotificationsDockerBinary.running, i) | ||
| i.icingaNotificationsDockerBinary.runningMutex.Unlock() | ||
|
|
||
| err := i.icingaNotificationsDockerBinary.dockerClient.ContainerRemove(context.Background(), i.containerId, types.ContainerRemoveOptions{ | ||
| Force: true, | ||
| RemoveVolumes: true, | ||
| }) | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
| i.logger.Debug("removed container") | ||
|
|
||
| err = os.Remove(i.configFileName) | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,33 @@ | ||
| package notifications | ||
|
|
||
| import ( | ||
| "github.com/icinga/icinga-testing/services" | ||
| ) | ||
|
|
||
| // defaultPort of the Icinga Notifications Web Listener. | ||
| const defaultPort string = "5680" | ||
|
|
||
| type Creator interface { | ||
| CreateIcingaNotifications(rdb services.RelationalDatabase, options ...services.IcingaNotificationsOption) services.IcingaNotificationsBase | ||
| Cleanup() | ||
| } | ||
|
|
||
| // info provides a partial implementation of the services.IcingaNotificationsBase interface. | ||
| type info struct { | ||
| host string | ||
| port string | ||
|
|
||
| rdb services.RelationalDatabase | ||
| } | ||
|
|
||
| func (i *info) Host() string { | ||
| return i.host | ||
| } | ||
|
|
||
| func (i *info) Port() string { | ||
| return i.port | ||
| } | ||
|
|
||
| func (i *info) RelationalDatabase() services.RelationalDatabase { | ||
| return i.rdb | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.