mirror of
https://github.com/rocky-linux/peridot.git
synced 2024-10-19 07:55:07 +00:00
ad0f7a5305
Upgrade to Go 1.20.5, Hydra v2 SDK, rules-go v0.44.2 (with proper resolves), protobuf v25.3 and mass upgrade of Go dependencies.
53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
// Copyright 2019 Google LLC.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
//go:build go1.11 && linux
|
|
// +build go1.11,linux
|
|
|
|
package grpc
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"syscall"
|
|
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
const (
|
|
// defaultTCPUserTimeout is the default TCP_USER_TIMEOUT socket option. By
|
|
// default is 20 seconds.
|
|
tcpUserTimeoutMilliseconds = 20000
|
|
|
|
// Copied from golang.org/x/sys/unix.TCP_USER_TIMEOUT.
|
|
tcpUserTimeoutOp = 0x12
|
|
)
|
|
|
|
func init() {
|
|
// timeoutDialerOption is a grpc.DialOption that contains dialer with
|
|
// socket option TCP_USER_TIMEOUT. This dialer requires go versions 1.11+.
|
|
timeoutDialerOption = grpc.WithContextDialer(dialTCPUserTimeout)
|
|
}
|
|
|
|
func dialTCPUserTimeout(ctx context.Context, addr string) (net.Conn, error) {
|
|
control := func(network, address string, c syscall.RawConn) error {
|
|
var syscallErr error
|
|
controlErr := c.Control(func(fd uintptr) {
|
|
syscallErr = syscall.SetsockoptInt(
|
|
int(fd), syscall.IPPROTO_TCP, tcpUserTimeoutOp, tcpUserTimeoutMilliseconds)
|
|
})
|
|
if syscallErr != nil {
|
|
return syscallErr
|
|
}
|
|
if controlErr != nil {
|
|
return controlErr
|
|
}
|
|
return nil
|
|
}
|
|
d := &net.Dialer{
|
|
Control: control,
|
|
}
|
|
return d.DialContext(ctx, "tcp", addr)
|
|
}
|