mirror of
https://github.com/gravitational/teleport.git
synced 2026-09-01 16:03:55 +08:00
rosstimothy
e81212dfcc
This converts hand rolled collections of resources via ListFoo RPCs in tctl and the cache to make use of clientutils.Resources. The main driver for the change is consistency and simplifcation. Having a single place that pagination logic exists should help reduce the chances of any pagination bugs in the future. The helper also reduces the amount of code needed at a call site to aggregate resources. The guts of clientutils.IterateResources have been moved to clientutils.Resources which returns an iter.Seq2[T, error] instead of the callback based approach. This makes the api simpler and allows making use of the itertools/stream helpers. Additionally, the page size used within clientutils.Resources was made to dynamically adjust based on receiving any errors indicating that the max message size was exceeding.
105 lines
3.3 KiB
Go
105 lines
3.3 KiB
Go
/*
|
|
* Teleport
|
|
* Copyright (C) 2024 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/client/gitserver"
|
|
"github.com/gravitational/teleport/api/types"
|
|
"github.com/gravitational/teleport/api/utils/clientutils"
|
|
"github.com/gravitational/teleport/lib/itertools/stream"
|
|
"github.com/gravitational/teleport/lib/services"
|
|
)
|
|
|
|
type gitServerIndex string
|
|
|
|
const gitServerNameIndex gitServerIndex = "name"
|
|
|
|
func newGitServerCollection(upstream services.GitServerGetter, w types.WatchKind) (*collection[types.Server, gitServerIndex], error) {
|
|
if upstream == nil {
|
|
return nil, trace.BadParameter("missing parameter GitServerGetter")
|
|
}
|
|
|
|
return &collection[types.Server, gitServerIndex]{
|
|
store: newStore(
|
|
types.KindGitServer,
|
|
types.Server.DeepCopy,
|
|
map[gitServerIndex]func(types.Server) string{
|
|
gitServerNameIndex: types.Server.GetName,
|
|
}),
|
|
fetcher: func(ctx context.Context, loadSecrets bool) ([]types.Server, error) {
|
|
out, err := stream.Collect(clientutils.Resources(ctx, upstream.ListGitServers))
|
|
return out, trace.Wrap(err)
|
|
},
|
|
headerTransform: func(hdr *types.ResourceHeader) types.Server {
|
|
return &types.ServerV2{
|
|
Kind: hdr.Kind,
|
|
Version: hdr.Version,
|
|
Metadata: types.Metadata{
|
|
Name: hdr.Metadata.Name,
|
|
},
|
|
}
|
|
},
|
|
watch: w,
|
|
}, nil
|
|
}
|
|
|
|
// GitServerReadOnlyClient returns the read-only client for Git servers.
|
|
//
|
|
// Note that Cache implements GitServerReadOnlyClient to satisfy
|
|
// auth.ProxyAccessPoint but also has the getter functions at top level to
|
|
// satisfy auth.Cache.
|
|
func (c *Cache) GitServerReadOnlyClient() gitserver.ReadOnlyClient {
|
|
return c
|
|
}
|
|
|
|
func (c *Cache) GetGitServer(ctx context.Context, name string) (types.Server, error) {
|
|
ctx, span := c.Tracer.Start(ctx, "cache/GetGitServer")
|
|
defer span.End()
|
|
|
|
getter := genericGetter[types.Server, gitServerIndex]{
|
|
cache: c,
|
|
collection: c.collections.gitServers,
|
|
index: gitServerNameIndex,
|
|
upstreamGet: c.Config.GitServers.GetGitServer,
|
|
}
|
|
out, err := getter.get(ctx, name)
|
|
return out, trace.Wrap(err)
|
|
}
|
|
|
|
func (c *Cache) ListGitServers(ctx context.Context, pageSize int, pageToken string) ([]types.Server, string, error) {
|
|
ctx, span := c.Tracer.Start(ctx, "cache/ListGitServers")
|
|
defer span.End()
|
|
|
|
lister := genericLister[types.Server, gitServerIndex]{
|
|
cache: c,
|
|
collection: c.collections.gitServers,
|
|
index: gitServerNameIndex,
|
|
upstreamList: c.Config.GitServers.ListGitServers,
|
|
nextToken: func(t types.Server) string {
|
|
return t.GetMetadata().Name
|
|
},
|
|
}
|
|
out, next, err := lister.list(ctx, pageSize, pageToken)
|
|
return out, next, trace.Wrap(err)
|
|
}
|