mirror of
https://github.com/gravitational/teleport.git
synced 2026-11-02 14:59:53 +08:00
rosstimothy
bde53bc267
The store now takes a clone function and uses it to clone all resources within calls to put. This ensures that there are no races caused by any modifications of the resource by the event stream. This also allowed the clone function to be removed from the genericGetter and genericLister as they can now leverage the store clone function. All header transforms now populate the kind and version based on the information in the ResourceHeader instead of hardcoding the currently known values. All indexes derived from the resource name are specified using `Resource.GetName` instead of defining a function in line that called r.GetName(). The one caveat being this only works for types that are behind an interface and none of the RFD 153 style resources. The clone function of all non-gogo proto resources were switched from utils.CloneProtoMsg to proto.CloneOf. The clone utility is only required to prevent panics when cloning gogo proto resources. All remaining local resource caches have been removed. All unused legacy executors have been removed.
110 lines
3.3 KiB
Go
110 lines
3.3 KiB
Go
// Teleport
|
|
// Copyright (C) 2025 Gravitational, Inc.
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package cache
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/gravitational/trace"
|
|
|
|
"github.com/gravitational/teleport/api/types"
|
|
"github.com/gravitational/teleport/lib/services"
|
|
)
|
|
|
|
type integrationIndex string
|
|
|
|
const integrationNameIndex integrationIndex = "name"
|
|
|
|
func newIntegrationCollection(upstream services.Integrations, w types.WatchKind) (*collection[types.Integration, integrationIndex], error) {
|
|
if upstream == nil {
|
|
return nil, trace.BadParameter("missing parameter Integrations")
|
|
}
|
|
|
|
return &collection[types.Integration, integrationIndex]{
|
|
store: newStore(
|
|
types.Integration.Clone,
|
|
map[integrationIndex]func(types.Integration) string{
|
|
integrationNameIndex: types.Integration.GetName,
|
|
}),
|
|
fetcher: func(ctx context.Context, loadSecrets bool) ([]types.Integration, error) {
|
|
var startKey string
|
|
var resources []types.Integration
|
|
for {
|
|
var igs []types.Integration
|
|
var err error
|
|
igs, startKey, err = upstream.ListIntegrations(ctx, 0, startKey)
|
|
if err != nil {
|
|
return nil, trace.Wrap(err)
|
|
}
|
|
|
|
resources = append(resources, igs...)
|
|
|
|
if startKey == "" {
|
|
break
|
|
}
|
|
}
|
|
|
|
return resources, nil
|
|
},
|
|
headerTransform: func(hdr *types.ResourceHeader) types.Integration {
|
|
return &types.IntegrationV1{
|
|
ResourceHeader: types.ResourceHeader{
|
|
Kind: hdr.Kind,
|
|
Version: hdr.Version,
|
|
Metadata: types.Metadata{
|
|
Name: hdr.Metadata.Name,
|
|
},
|
|
},
|
|
}
|
|
},
|
|
watch: w,
|
|
}, nil
|
|
}
|
|
|
|
// ListIntegrations returns a paginated list of all Integrations resources.
|
|
func (c *Cache) ListIntegrations(ctx context.Context, pageSize int, pageToken string) ([]types.Integration, string, error) {
|
|
ctx, span := c.Tracer.Start(ctx, "cache/ListIntegrations")
|
|
defer span.End()
|
|
|
|
lister := genericLister[types.Integration, integrationIndex]{
|
|
cache: c,
|
|
collection: c.collections.integrations,
|
|
index: integrationNameIndex,
|
|
upstreamList: c.Config.Integrations.ListIntegrations,
|
|
nextToken: func(t types.Integration) string {
|
|
return t.GetMetadata().Name
|
|
},
|
|
}
|
|
out, next, err := lister.list(ctx, pageSize, pageToken)
|
|
return out, next, trace.Wrap(err)
|
|
}
|
|
|
|
// GetIntegration returns the specified Integration resources.
|
|
func (c *Cache) GetIntegration(ctx context.Context, name string) (types.Integration, error) {
|
|
ctx, span := c.Tracer.Start(ctx, "cache/GetIntegration")
|
|
defer span.End()
|
|
|
|
getter := genericGetter[types.Integration, integrationIndex]{
|
|
cache: c,
|
|
collection: c.collections.integrations,
|
|
index: integrationNameIndex,
|
|
upstreamGet: c.Config.Integrations.GetIntegration,
|
|
}
|
|
out, err := getter.get(ctx, name)
|
|
return out, trace.Wrap(err)
|
|
}
|