2022-11-01 19:03:23 +00:00
|
|
|
package service
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"fmt"
|
|
|
|
"io"
|
|
|
|
"net/http"
|
|
|
|
|
|
|
|
"github.com/hashicorp/consul/api"
|
|
|
|
libnode "github.com/hashicorp/consul/test/integration/consul-container/libs/agent"
|
|
|
|
)
|
|
|
|
|
|
|
|
func CreateAndRegisterStaticServerAndSidecar(node libnode.Agent) (Service, Service, error) {
|
|
|
|
// Create a service and proxy instance
|
|
|
|
serverService, err := NewExampleService(context.Background(), "static-server", 8080, 8079, node)
|
|
|
|
if err != nil {
|
|
|
|
return nil, nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
serverConnectProxy, err := NewConnectService(context.Background(), "static-server-sidecar", "static-server", 8080, node) // bindPort not used
|
|
|
|
if err != nil {
|
|
|
|
return nil, nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
serverServiceIP, _ := serverService.GetAddr()
|
|
|
|
serverConnectProxyIP, _ := serverConnectProxy.GetAddr()
|
|
|
|
|
|
|
|
// Register the static-server service and sidecar
|
|
|
|
req := &api.AgentServiceRegistration{
|
|
|
|
Name: "static-server",
|
|
|
|
Port: 8080,
|
|
|
|
Address: serverServiceIP,
|
|
|
|
Connect: &api.AgentServiceConnect{
|
|
|
|
SidecarService: &api.AgentServiceRegistration{
|
|
|
|
Name: "static-server-sidecar-proxy",
|
|
|
|
Port: 20000,
|
|
|
|
Address: serverConnectProxyIP,
|
|
|
|
Kind: api.ServiceKindConnectProxy,
|
|
|
|
Checks: api.AgentServiceChecks{
|
|
|
|
&api.AgentServiceCheck{
|
|
|
|
Name: "Connect Sidecar Listening",
|
|
|
|
TCP: fmt.Sprintf("%s:%d", serverConnectProxyIP, 20000),
|
|
|
|
Interval: "10s",
|
2022-11-08 01:10:42 +00:00
|
|
|
Status: api.HealthPassing,
|
2022-11-01 19:03:23 +00:00
|
|
|
},
|
|
|
|
&api.AgentServiceCheck{
|
|
|
|
Name: "Connect Sidecar Aliasing Static Server",
|
|
|
|
AliasService: "static-server",
|
2022-11-08 01:10:42 +00:00
|
|
|
Status: api.HealthPassing,
|
2022-11-01 19:03:23 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
Proxy: &api.AgentServiceConnectProxyConfig{
|
|
|
|
DestinationServiceName: "static-server",
|
|
|
|
LocalServiceAddress: serverServiceIP,
|
|
|
|
LocalServicePort: 8080,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
Check: &api.AgentServiceCheck{
|
2022-11-08 01:10:42 +00:00
|
|
|
Name: "Static Server Listening",
|
2022-11-01 19:03:23 +00:00
|
|
|
TCP: fmt.Sprintf("%s:%d", serverServiceIP, 8080),
|
|
|
|
Interval: "10s",
|
2022-11-08 01:10:42 +00:00
|
|
|
Status: api.HealthPassing,
|
2022-11-01 19:03:23 +00:00
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
err = node.GetClient().Agent().ServiceRegister(req)
|
|
|
|
if err != nil {
|
|
|
|
return serverService, serverConnectProxy, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return serverService, serverConnectProxy, nil
|
|
|
|
}
|
|
|
|
|
2022-11-08 01:10:42 +00:00
|
|
|
func CreateAndRegisterStaticClientSidecar(node libnode.Agent, peerName string, localMeshGateway bool) (*ConnectContainer, error) {
|
2022-11-01 19:03:23 +00:00
|
|
|
// Create a service and proxy instance
|
|
|
|
clientConnectProxy, err := NewConnectService(context.Background(), "static-client-sidecar", "static-client", 5000, node)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
clientConnectProxyIP, _ := clientConnectProxy.GetAddr()
|
|
|
|
|
|
|
|
mgwMode := api.MeshGatewayModeRemote
|
|
|
|
if localMeshGateway {
|
|
|
|
mgwMode = api.MeshGatewayModeLocal
|
|
|
|
}
|
|
|
|
|
|
|
|
// Register the static-client service and sidecar
|
|
|
|
req := &api.AgentServiceRegistration{
|
|
|
|
Name: "static-client",
|
|
|
|
Port: 8080,
|
|
|
|
Connect: &api.AgentServiceConnect{
|
|
|
|
SidecarService: &api.AgentServiceRegistration{
|
|
|
|
Name: "static-client-sidecar-proxy",
|
|
|
|
Port: 20000,
|
|
|
|
Kind: api.ServiceKindConnectProxy,
|
|
|
|
Checks: api.AgentServiceChecks{
|
|
|
|
&api.AgentServiceCheck{
|
|
|
|
Name: "Connect Sidecar Listening",
|
|
|
|
TCP: fmt.Sprintf("%s:%d", clientConnectProxyIP, 20000),
|
|
|
|
Interval: "10s",
|
2022-11-08 01:10:42 +00:00
|
|
|
Status: api.HealthPassing,
|
2022-11-01 19:03:23 +00:00
|
|
|
},
|
|
|
|
},
|
|
|
|
Proxy: &api.AgentServiceConnectProxyConfig{
|
|
|
|
Upstreams: []api.Upstream{
|
|
|
|
{
|
|
|
|
DestinationName: "static-server",
|
|
|
|
DestinationPeer: peerName,
|
|
|
|
LocalBindAddress: "0.0.0.0",
|
|
|
|
LocalBindPort: 5000,
|
|
|
|
MeshGateway: api.MeshGatewayConfig{
|
|
|
|
Mode: mgwMode,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
Checks: api.AgentServiceChecks{},
|
|
|
|
}
|
|
|
|
|
|
|
|
err = node.GetClient().Agent().ServiceRegister(req)
|
|
|
|
if err != nil {
|
|
|
|
return clientConnectProxy, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return clientConnectProxy, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func GetEnvoyConfigDump(port int) (string, error) {
|
|
|
|
client := http.DefaultClient
|
|
|
|
url := fmt.Sprintf("http://localhost:%d/config_dump?include_eds", port)
|
|
|
|
|
|
|
|
res, err := client.Get(url)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
defer res.Body.Close()
|
|
|
|
|
|
|
|
body, err := io.ReadAll(res.Body)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
|
|
|
|
return string(body), nil
|
|
|
|
}
|