mirror of
https://github.com/siderolabs/talos.git
synced 2025-12-10 12:01:33 +01:00
98d56d4d64
This is an alternative to #10283 which doesn't close the connections and instead allows us to easily track the lost grpc connections using pprof. Fix various grpc leaks while we are at it. Signed-off-by: Dmitriy Matrenichev <dmitry.matrenichev@siderolabs.com>
61 lines
1.5 KiB
Go
61 lines
1.5 KiB
Go
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
package cri
|
|
|
|
import (
|
|
"fmt"
|
|
"runtime/pprof"
|
|
"time"
|
|
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/credentials/insecure"
|
|
runtimeapi "k8s.io/cri-api/pkg/apis/runtime/v1"
|
|
|
|
"github.com/siderolabs/talos/pkg/grpc/dialer"
|
|
)
|
|
|
|
var newClientPprof = pprof.NewProfile("internal/pkg/cri.NewClient")
|
|
|
|
// Client is a lightweight implementation of CRI client.
|
|
type Client struct {
|
|
conn *grpc.ClientConn
|
|
runtimeClient runtimeapi.RuntimeServiceClient
|
|
imagesClient runtimeapi.ImageServiceClient
|
|
}
|
|
|
|
// maxMsgSize use 16MB as the default message size limit.
|
|
// grpc library default is 4MB.
|
|
const maxMsgSize = 1024 * 1024 * 16
|
|
|
|
// NewClient builds CRI client.
|
|
func NewClient(endpoint string, _ time.Duration) (*Client, error) {
|
|
conn, err := grpc.NewClient(endpoint,
|
|
grpc.WithTransportCredentials(insecure.NewCredentials()),
|
|
grpc.WithDefaultCallOptions(grpc.MaxCallRecvMsgSize(maxMsgSize)),
|
|
grpc.WithContextDialer(dialer.DialUnix()),
|
|
grpc.WithSharedWriteBuffer(true),
|
|
)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error connecting to CRI: %w", err)
|
|
}
|
|
|
|
res := &Client{
|
|
conn: conn,
|
|
runtimeClient: runtimeapi.NewRuntimeServiceClient(conn),
|
|
imagesClient: runtimeapi.NewImageServiceClient(conn),
|
|
}
|
|
|
|
newClientPprof.Add(res, 1)
|
|
|
|
return res, nil
|
|
}
|
|
|
|
// Close connection.
|
|
func (c *Client) Close() error {
|
|
newClientPprof.Remove(c)
|
|
|
|
return c.conn.Close()
|
|
}
|