Jeff MitchellandDamian Debkowski
014a5a7aae
This addresses all found vet/race fixes in community edition. --------- Co-authored-by: Damian Debkowski <damian.debkowski@hashicorp.com>
43 lines
1.2 KiB
Go
43 lines
1.2 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package proxy
|
|
|
|
import (
|
|
"net/http"
|
|
"sync/atomic"
|
|
)
|
|
|
|
// ProxyState contains the current state of proxies in this process.
|
|
var ProxyState proxyState = proxyState{proxyCount: new(atomic.Int64)}
|
|
|
|
type proxyState struct {
|
|
proxyCount *atomic.Int64
|
|
}
|
|
|
|
// CurrentProxiedConnections returns the current number of ongoing proxied
|
|
// connections which are currently running the Handler's ProxyConnFn.
|
|
func (p *proxyState) CurrentProxiedConnections() int64 {
|
|
return p.proxyCount.Load()
|
|
}
|
|
|
|
// HttpHandlerCounter records how many requests are currently running in the
|
|
// wrapped Handler. This should be used for handlers that serve proxied traffic.
|
|
func ProxyHandlerCounter(h http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
|
|
ProxyState.proxyCount.Add(1)
|
|
defer ProxyState.proxyCount.Add(-1)
|
|
h.ServeHTTP(rw, r)
|
|
})
|
|
}
|
|
|
|
// proxyConnFnCounter wraps a ProxyState and keeps the proxyCount incremented
|
|
// while it runs.
|
|
func proxyConnFnCounter(fn ProxyConnFn) ProxyConnFn {
|
|
return func() {
|
|
ProxyState.proxyCount.Add(1)
|
|
defer ProxyState.proxyCount.Add(-1)
|
|
fn()
|
|
}
|
|
}
|