mirror of
https://github.com/gravitational/teleport.git
synced 2026-09-21 14:35:22 +08:00
rosstimothy
5540e35507
During the cache storage refactoring, all of the cache metrics that were recorded via the backend.Reporter that wrapped the cache backend were lost. This attempts to restore the metrics as close to the original meaning as possible, though due to the changes some labels are different and batch metrics don't apply. The biggest change is in the top requests metrics which has historically been based on the backend.Key, but since that the cache no longer deals with backends, the metric is now labeled by the resource kind. All existing dashboards and tctl top should still work as expected.
117 lines
3.5 KiB
Go
117 lines
3.5 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"
|
|
apidefaults "github.com/gravitational/teleport/api/defaults"
|
|
"github.com/gravitational/teleport/api/types"
|
|
"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) {
|
|
var all []types.Server
|
|
var nextToken string
|
|
for {
|
|
var page []types.Server
|
|
var err error
|
|
page, nextToken, err = upstream.ListGitServers(ctx, apidefaults.DefaultChunkSize, nextToken)
|
|
if err != nil {
|
|
return nil, trace.Wrap(err)
|
|
}
|
|
all = append(all, page...)
|
|
if nextToken == "" {
|
|
break
|
|
}
|
|
}
|
|
return all, nil
|
|
},
|
|
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)
|
|
}
|