mirror of https://github.com/hashicorp/consul
145 lines
3.8 KiB
Go
145 lines
3.8 KiB
Go
|
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",
|
||
|
},
|
||
|
&api.AgentServiceCheck{
|
||
|
Name: "Connect Sidecar Aliasing Static Server",
|
||
|
AliasService: "static-server",
|
||
|
},
|
||
|
},
|
||
|
Proxy: &api.AgentServiceConnectProxyConfig{
|
||
|
DestinationServiceName: "static-server",
|
||
|
LocalServiceAddress: serverServiceIP,
|
||
|
LocalServicePort: 8080,
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
Check: &api.AgentServiceCheck{
|
||
|
Name: "Connect Sidecar Listening",
|
||
|
TCP: fmt.Sprintf("%s:%d", serverServiceIP, 8080),
|
||
|
Interval: "10s",
|
||
|
},
|
||
|
}
|
||
|
|
||
|
err = node.GetClient().Agent().ServiceRegister(req)
|
||
|
if err != nil {
|
||
|
return serverService, serverConnectProxy, err
|
||
|
}
|
||
|
|
||
|
return serverService, serverConnectProxy, nil
|
||
|
}
|
||
|
|
||
|
func CreateAndRegisterStaticClientSidecar(node libnode.Agent, peerName string, localMeshGateway bool) (Service, error) {
|
||
|
// 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",
|
||
|
},
|
||
|
},
|
||
|
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
|
||
|
}
|