mirror of
https://github.com/tendermint/tendermint.git
synced 2026-01-06 13:26:23 +00:00
120 lines
3.3 KiB
Go
120 lines
3.3 KiB
Go
package docker
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
"text/template"
|
|
|
|
e2e "github.com/tendermint/tendermint/test/e2e/pkg"
|
|
"github.com/tendermint/tendermint/test/e2e/pkg/exec"
|
|
"github.com/tendermint/tendermint/test/e2e/pkg/infra"
|
|
)
|
|
|
|
var _ infra.Provider = &Provider{}
|
|
|
|
// Provider implements a docker-compose backed infrastructure provider.
|
|
type Provider struct {
|
|
Testnet *e2e.Testnet
|
|
}
|
|
|
|
// Setup generates the docker-compose file and write it to disk, erroring if
|
|
// any of these operations fail.
|
|
func (p *Provider) Setup() error {
|
|
compose, err := dockerComposeBytes(p.Testnet)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
//nolint: gosec
|
|
// G306: Expect WriteFile permissions to be 0600 or less
|
|
err = os.WriteFile(filepath.Join(p.Testnet.Dir, "docker-compose.yml"), compose, 0644)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p Provider) CreateNode(ctx context.Context, n *e2e.Node) error {
|
|
return ExecCompose(ctx, p.Testnet.Dir, "up", "-d", "--no-start", n.Name)
|
|
}
|
|
func (p Provider) StartTendermint(ctx context.Context, n *e2e.Node) error {
|
|
return ExecCompose(ctx, p.Testnet.Dir, "start", n.Name)
|
|
}
|
|
func (p Provider) TerminateTendermint(ctx context.Context, n *e2e.Node) error {
|
|
return ExecCompose(ctx, p.Testnet.Dir, "kill", "-s", "SIGTERM", n.Name)
|
|
}
|
|
func (p Provider) KillTendermint(ctx context.Context, n *e2e.Node) error {
|
|
return ExecCompose(ctx, p.Testnet.Dir, "kill", "-s", "SIGKILL", n.Name)
|
|
}
|
|
|
|
// dockerComposeBytes generates a Docker Compose config file for a testnet and returns the
|
|
// file as bytes to be written out to disk.
|
|
func dockerComposeBytes(testnet *e2e.Testnet) ([]byte, error) {
|
|
// Must use version 2 Docker Compose format, to support IPv6.
|
|
tmpl, err := template.New("docker-compose").Parse(`version: '2.4'
|
|
networks:
|
|
{{ .Name }}:
|
|
labels:
|
|
e2e: true
|
|
driver: bridge
|
|
{{- if .IPv6 }}
|
|
enable_ipv6: true
|
|
{{- end }}
|
|
ipam:
|
|
driver: default
|
|
config:
|
|
- subnet: {{ .IP }}
|
|
|
|
services:
|
|
{{- range .Nodes }}
|
|
{{ .Name }}:
|
|
labels:
|
|
e2e: true
|
|
container_name: {{ .Name }}
|
|
image: tendermint/e2e-node:{{ .Version }}
|
|
{{- if eq .ABCIProtocol "builtin" }}
|
|
entrypoint: /usr/bin/entrypoint-builtin
|
|
{{- end }}
|
|
init: true
|
|
ports:
|
|
- 26656
|
|
- {{ if .ProxyPort }}{{ .ProxyPort }}:{{ end }}26657
|
|
- 6060
|
|
volumes:
|
|
- ./{{ .Name }}:/tendermint
|
|
networks:
|
|
{{ $.Name }}:
|
|
ipv{{ if $.IPv6 }}6{{ else }}4{{ end}}_address: {{ .InternalIP }}
|
|
|
|
{{end}}`)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var buf bytes.Buffer
|
|
err = tmpl.Execute(&buf, testnet)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return buf.Bytes(), nil
|
|
}
|
|
|
|
// ExecCompose runs a Docker Compose command for a testnet.
|
|
func ExecCompose(ctx context.Context, dir string, args ...string) error {
|
|
return exec.Command(ctx, append(
|
|
[]string{"docker-compose", "-f", filepath.Join(dir, "docker-compose.yml")},
|
|
args...)...)
|
|
}
|
|
|
|
// ExecComposeVerbose runs a Docker Compose command for a testnet and displays its output.
|
|
func ExecComposeVerbose(ctx context.Context, dir string, args ...string) error {
|
|
return exec.CommandVerbose(ctx, append(
|
|
[]string{"docker-compose", "-f", filepath.Join(dir, "docker-compose.yml")},
|
|
args...)...)
|
|
}
|
|
|
|
// Exec runs a Docker command.
|
|
func Exec(ctx context.Context, args ...string) error {
|
|
return exec.Command(ctx, append([]string{"docker"}, args...)...)
|
|
}
|