.github
alerting
client
config
core
docs
example
jsonpath
k8s
k8stest
metric
pattern
security
static
vendor
cloud.google.com
github.com
golang.org
google.golang.org
appengine
internal
app_identity
base
datastore
log
modules
remote_api
urlfetch
api.go
api_classic.go
api_common.go
app_id.go
identity.go
identity_classic.go
identity_flex.go
identity_vm.go
internal.go
main.go
main_common.go
main_vm.go
metadata.go
net.go
regen.sh
transaction.go
urlfetch
.travis.yml
CONTRIBUTING.md
LICENSE
README.md
appengine.go
appengine_vm.go
errors.go
go.mod
go.sum
identity.go
namespace.go
timeout.go
travis_install.sh
travis_test.sh
protobuf
gopkg.in
k8s.io
sigs.k8s.io
modules.txt
watchdog
.dockerignore
.gitattributes
.gitignore
Dockerfile
LICENSE.md
README.md
config.yaml
go.mod
go.sum
gzip.go
main.go
57 lines
1.2 KiB
Go
57 lines
1.2 KiB
Go
// Copyright 2014 Google Inc. All rights reserved.
|
|
// Use of this source code is governed by the Apache 2.0
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package internal
|
|
|
|
// This file implements a network dialer that limits the number of concurrent connections.
|
|
// It is only used for API calls.
|
|
|
|
import (
|
|
"log"
|
|
"net"
|
|
"runtime"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
var limitSem = make(chan int, 100) // TODO(dsymonds): Use environment variable.
|
|
|
|
func limitRelease() {
|
|
// non-blocking
|
|
select {
|
|
case <-limitSem:
|
|
default:
|
|
// This should not normally happen.
|
|
log.Print("appengine: unbalanced limitSem release!")
|
|
}
|
|
}
|
|
|
|
func limitDial(network, addr string) (net.Conn, error) {
|
|
limitSem <- 1
|
|
|
|
// Dial with a timeout in case the API host is MIA.
|
|
// The connection should normally be very fast.
|
|
conn, err := net.DialTimeout(network, addr, 10*time.Second)
|
|
if err != nil {
|
|
limitRelease()
|
|
return nil, err
|
|
}
|
|
lc := &limitConn{Conn: conn}
|
|
runtime.SetFinalizer(lc, (*limitConn).Close) // shouldn't usually be required
|
|
return lc, nil
|
|
}
|
|
|
|
type limitConn struct {
|
|
close sync.Once
|
|
net.Conn
|
|
}
|
|
|
|
func (lc *limitConn) Close() error {
|
|
defer lc.close.Do(func() {
|
|
limitRelease()
|
|
runtime.SetFinalizer(lc, nil)
|
|
})
|
|
return lc.Conn.Close()
|
|
}
|