mirror of
https://github.com/tendermint/tendermint.git
synced 2026-01-08 14:21:14 +00:00
89 lines
2.1 KiB
Go
89 lines
2.1 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/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 (Provider) StartTendermint(_ context.Context, _ *e2e.Node) error { panic("unimplemented") }
|
|
func (Provider) KillTendermint(_ context.Context, _ *e2e.Node) error { panic("unimplemented") }
|
|
|
|
// 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
|
|
{{- 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: {{ .IP }}
|
|
|
|
{{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
|
|
}
|