-
Notifications
You must be signed in to change notification settings - Fork 118
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
6 changed files
with
184 additions
and
17 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
172 changes: 172 additions & 0 deletions
172
internal/testrunner/runners/system/servicedeployer/custom_agent.go
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,172 @@ | ||
| // Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one | ||
| // or more contributor license agreements. Licensed under the Elastic License; | ||
| // you may not use this file except in compliance with the Elastic License. | ||
|
|
||
| package servicedeployer | ||
|
|
||
| import ( | ||
| "fmt" | ||
|
|
||
| "github.com/pkg/errors" | ||
|
|
||
| "github.com/elastic/elastic-package/internal/compose" | ||
| "github.com/elastic/elastic-package/internal/docker" | ||
| "github.com/elastic/elastic-package/internal/files" | ||
| "github.com/elastic/elastic-package/internal/logger" | ||
| "github.com/elastic/elastic-package/internal/stack" | ||
| ) | ||
|
|
||
| // CustomAgentDeployer knows how to deploy a custom elastic-agent defined via | ||
| // a Docker Compose file. | ||
| type CustomAgentDeployer struct { | ||
| ymlPaths []string | ||
| } | ||
|
|
||
| type deployedCustomAgent struct { | ||
| ctxt ServiceContext | ||
|
|
||
| ymlPaths []string | ||
| project string | ||
| } | ||
|
|
||
| // NewCustomAgentDeployer returns a new instance of a deployedCustomAgent. | ||
| func NewCustomAgentDeployer(ymlPaths []string) (*CustomAgentDeployer, error) { | ||
| return &CustomAgentDeployer{ | ||
| ymlPaths: ymlPaths, | ||
| }, nil | ||
| } | ||
|
|
||
| // SetUp sets up the service and returns any relevant information. | ||
| func (d *CustomAgentDeployer) SetUp(inCtxt ServiceContext) (DeployedService, error) { | ||
| logger.Debug("setting up service using Docker Compose service deployer") | ||
| service := deployedCustomAgent{ | ||
| ymlPaths: d.ymlPaths, | ||
| project: "elastic-package-service", | ||
| } | ||
| outCtxt := inCtxt | ||
|
|
||
| p, err := compose.NewProject(service.project, service.ymlPaths...) | ||
| if err != nil { | ||
| return nil, errors.Wrap(err, "could not create Docker Compose project for service") | ||
| } | ||
|
|
||
| // Verify the Elastic stack network | ||
| err = stack.EnsureStackNetworkUp() | ||
| if err != nil { | ||
| return nil, errors.Wrap(err, "Elastic stack network is not ready") | ||
| } | ||
|
|
||
| // Clean service logs | ||
| err = files.RemoveContent(outCtxt.Logs.Folder.Local) | ||
| if err != nil { | ||
| return nil, errors.Wrap(err, "removing service logs failed") | ||
| } | ||
|
|
||
| // Boot up service | ||
|
|
||
| serviceName := inCtxt.Name | ||
| opts := compose.CommandOptions{ | ||
| Env: []string{fmt.Sprintf("%s=%s", serviceLogsDirEnv, outCtxt.Logs.Folder.Local)}, | ||
| ExtraArgs: []string{"--build", "-d"}, | ||
| } | ||
| err = p.Up(opts) | ||
| if err != nil { | ||
| return nil, errors.Wrap(err, "could not boot up service using Docker Compose") | ||
| } | ||
|
|
||
| // Connect service network with stack network (for the purpose of metrics collection) | ||
| err = docker.ConnectToNetwork(p.ContainerName(serviceName), stack.Network()) | ||
| if err != nil { | ||
| return nil, errors.Wrapf(err, "can't attach service container to the stack network") | ||
| } | ||
|
|
||
| err = p.WaitForHealthy(opts) | ||
| if err != nil { | ||
| processServiceContainerLogs(p, compose.CommandOptions{ | ||
| Env: opts.Env, | ||
| }, outCtxt.Name) | ||
| return nil, errors.Wrap(err, "service is unhealthy") | ||
| } | ||
|
|
||
| // Build service container name | ||
| outCtxt.Hostname = p.ContainerName(serviceName) | ||
|
|
||
| logger.Debugf("adding service container %s internal ports to context", p.ContainerName(serviceName)) | ||
| serviceComposeConfig, err := p.Config(compose.CommandOptions{ | ||
| Env: []string{fmt.Sprintf("%s=%s", serviceLogsDirEnv, outCtxt.Logs.Folder.Local)}, | ||
| }) | ||
| if err != nil { | ||
| return nil, errors.Wrap(err, "could not get Docker Compose configuration for service") | ||
| } | ||
|
|
||
| s := serviceComposeConfig.Services[serviceName] | ||
| outCtxt.Ports = make([]int, len(s.Ports)) | ||
| for idx, port := range s.Ports { | ||
| outCtxt.Ports[idx] = port.InternalPort | ||
| } | ||
|
|
||
| // Shortcut to first port for convenience | ||
| if len(outCtxt.Ports) > 0 { | ||
| outCtxt.Port = outCtxt.Ports[0] | ||
| } | ||
|
|
||
| outCtxt.Agent.Host.NamePrefix = serviceName | ||
|
|
||
| service.ctxt = outCtxt | ||
| return &service, nil | ||
| } | ||
|
|
||
| // Signal sends a signal to the service. | ||
| func (s *deployedCustomAgent) Signal(signal string) error { | ||
| p, err := compose.NewProject(s.project, s.ymlPaths...) | ||
| if err != nil { | ||
| return errors.Wrap(err, "could not create Docker Compose project for service") | ||
| } | ||
|
|
||
| opts := compose.CommandOptions{ExtraArgs: []string{"-s", signal}} | ||
| if s.ctxt.Name != "" { | ||
| opts.Services = append(opts.Services, s.ctxt.Name) | ||
| } | ||
|
|
||
| return errors.Wrapf(p.Kill(opts), "could not send %q signal", signal) | ||
| } | ||
|
|
||
| // TearDown tears down the service. | ||
| func (s *deployedCustomAgent) TearDown() error { | ||
| logger.Debugf("tearing down service using Docker Compose runner") | ||
| defer func() { | ||
| err := files.RemoveContent(s.ctxt.Logs.Folder.Local) | ||
| if err != nil { | ||
| logger.Errorf("could not remove the service logs (path: %s)", s.ctxt.Logs.Folder.Local) | ||
| } | ||
| }() | ||
|
|
||
| p, err := compose.NewProject(s.project, s.ymlPaths...) | ||
| if err != nil { | ||
| return errors.Wrap(err, "could not create Docker Compose project for service") | ||
| } | ||
|
|
||
| opts := compose.CommandOptions{ | ||
| Env: []string{fmt.Sprintf("%s=%s", serviceLogsDirEnv, s.ctxt.Logs.Folder.Local)}, | ||
| } | ||
| processServiceContainerLogs(p, opts, s.ctxt.Name) | ||
|
|
||
| if err := p.Down(compose.CommandOptions{ | ||
| Env: []string{fmt.Sprintf("%s=%s", serviceLogsDirEnv, s.ctxt.Logs.Folder.Local)}, | ||
| ExtraArgs: []string{"--volumes"}, // Remove associated volumes. | ||
| }); err != nil { | ||
| return errors.Wrap(err, "could not shut down service using Docker Compose") | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| // Context returns the current context for the service. | ||
| func (s *deployedCustomAgent) Context() ServiceContext { | ||
| return s.ctxt | ||
| } | ||
|
|
||
| // SetContext sets the current context for the service. | ||
| func (s *deployedCustomAgent) SetContext(ctxt ServiceContext) error { | ||
| s.ctxt = ctxt | ||
| return nil | ||
| } |
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