kuma

1
package envoyadmin
2

3
import (
4
	"context"
5

6
	"github.com/pkg/errors"
7

8
	mesh_proto "github.com/kumahq/kuma/api/mesh/v1alpha1"
9
	"github.com/kumahq/kuma/pkg/core"
10
	"github.com/kumahq/kuma/pkg/core/resources/manager"
11
	"github.com/kumahq/kuma/pkg/core/resources/model"
12
	"github.com/kumahq/kuma/pkg/core/resources/registry"
13
	core_store "github.com/kumahq/kuma/pkg/core/resources/store"
14
	"github.com/kumahq/kuma/pkg/envoy/admin"
15
)
16

17
var serverLog = core.Log.WithName("intercp").WithName("catalog").WithName("server")
18

19
type server struct {
20
	adminClient admin.EnvoyAdminClient
21
	resManager  manager.ReadOnlyResourceManager
22
	mesh_proto.UnimplementedInterCPEnvoyAdminForwardServiceServer
23
}
24

25
var _ mesh_proto.InterCPEnvoyAdminForwardServiceServer = &server{}
26

27
func NewServer(adminClient admin.EnvoyAdminClient, resManager manager.ReadOnlyResourceManager) mesh_proto.InterCPEnvoyAdminForwardServiceServer {
28
	return &server{
29
		adminClient: adminClient,
30
		resManager:  resManager,
31
	}
32
}
33

34
func (s *server) XDSConfig(ctx context.Context, req *mesh_proto.XDSConfigRequest) (*mesh_proto.XDSConfigResponse, error) {
35
	ctx = extractTenantMetadata(ctx)
36
	serverLog.V(1).Info("received forwarded request", "operation", "XDSConfig", "request", req)
37
	resWithAddr, err := s.resWithAddress(ctx, req.ResourceType, req.ResourceName, req.ResourceMesh)
38
	if err != nil {
39
		return nil, err
40
	}
41
	configDump, err := s.adminClient.ConfigDump(ctx, resWithAddr)
42
	if err != nil {
43
		if errors.Is(err, &admin.KDSTransportError{}) {
44
			return &mesh_proto.XDSConfigResponse{
45
				Result: &mesh_proto.XDSConfigResponse_Error{
46
					Error: err.Error(),
47
				},
48
			}, nil
49
		}
50
		return nil, err
51
	}
52
	return &mesh_proto.XDSConfigResponse{
53
		Result: &mesh_proto.XDSConfigResponse_Config{
54
			Config: configDump,
55
		},
56
	}, nil
57
}
58

59
func (s *server) Stats(ctx context.Context, req *mesh_proto.StatsRequest) (*mesh_proto.StatsResponse, error) {
60
	ctx = extractTenantMetadata(ctx)
61
	serverLog.V(1).Info("received forwarded request", "operation", "Stats", "request", req)
62
	resWithAddr, err := s.resWithAddress(ctx, req.ResourceType, req.ResourceName, req.ResourceMesh)
63
	if err != nil {
64
		return nil, err
65
	}
66
	stats, err := s.adminClient.Stats(ctx, resWithAddr)
67
	if err != nil {
68
		if errors.Is(err, &admin.KDSTransportError{}) {
69
			return &mesh_proto.StatsResponse{
70
				Result: &mesh_proto.StatsResponse_Error{
71
					Error: err.Error(),
72
				},
73
			}, nil
74
		}
75
		return nil, err
76
	}
77
	return &mesh_proto.StatsResponse{
78
		Result: &mesh_proto.StatsResponse_Stats{
79
			Stats: stats,
80
		},
81
	}, nil
82
}
83

84
func (s *server) Clusters(ctx context.Context, req *mesh_proto.ClustersRequest) (*mesh_proto.ClustersResponse, error) {
85
	ctx = extractTenantMetadata(ctx)
86
	serverLog.V(1).Info("received forwarded request", "operation", "Clusters", "request", req)
87
	resWithAddr, err := s.resWithAddress(ctx, req.ResourceType, req.ResourceName, req.ResourceMesh)
88
	if err != nil {
89
		return nil, err
90
	}
91
	clusters, err := s.adminClient.Clusters(ctx, resWithAddr)
92
	if err != nil {
93
		if errors.Is(err, &admin.KDSTransportError{}) {
94
			return &mesh_proto.ClustersResponse{
95
				Result: &mesh_proto.ClustersResponse_Error{
96
					Error: err.Error(),
97
				},
98
			}, nil
99
		}
100
		return nil, err
101
	}
102
	return &mesh_proto.ClustersResponse{
103
		Result: &mesh_proto.ClustersResponse_Clusters{
104
			Clusters: clusters,
105
		},
106
	}, nil
107
}
108

109
func (s *server) resWithAddress(ctx context.Context, typ, name, mesh string) (model.ResourceWithAddress, error) {
110
	obj, err := registry.Global().NewObject(model.ResourceType(typ))
111
	if err != nil {
112
		return nil, err
113
	}
114
	if err := s.resManager.Get(ctx, obj, core_store.GetByKey(name, mesh)); err != nil {
115
		return nil, err
116
	}
117
	resourceWithAddr, ok := obj.(model.ResourceWithAddress)
118
	if !ok {
119
		return nil, errors.New("invalid resource type")
120
	}
121
	return resourceWithAddr, nil
122
}
123