mirror of
https://source.quilibrium.com/quilibrium/ceremonyclient.git
synced 2024-11-14 04:05:17 +00:00
815 lines
19 KiB
Go
815 lines
19 KiB
Go
|
package rcmgr
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"math"
|
||
|
"math/big"
|
||
|
"strings"
|
||
|
"sync"
|
||
|
|
||
|
"github.com/libp2p/go-libp2p/core/network"
|
||
|
)
|
||
|
|
||
|
// resources tracks the current state of resource consumption
|
||
|
type resources struct {
|
||
|
limit Limit
|
||
|
|
||
|
nconnsIn, nconnsOut int
|
||
|
nstreamsIn, nstreamsOut int
|
||
|
nfd int
|
||
|
|
||
|
memory int64
|
||
|
}
|
||
|
|
||
|
// A resourceScope can be a DAG, where a downstream node is not allowed to outlive an upstream node
|
||
|
// (ie cannot call Done in the upstream node before the downstream node) and account for resources
|
||
|
// using a linearized parent set.
|
||
|
// A resourceScope can be a span scope, where it has a specific owner; span scopes create a tree rooted
|
||
|
// at the owner (which can be a DAG scope) and can outlive their parents -- this is important because
|
||
|
// span scopes are the main *user* interface for memory management, and the user may call
|
||
|
// Done in a span scope after the system has closed the root of the span tree in some background
|
||
|
// goroutine.
|
||
|
// If we didn't make this distinction we would have a double release problem in that case.
|
||
|
type resourceScope struct {
|
||
|
sync.Mutex
|
||
|
done bool
|
||
|
refCnt int
|
||
|
|
||
|
spanID int
|
||
|
|
||
|
rc resources
|
||
|
owner *resourceScope // set in span scopes, which define trees
|
||
|
edges []*resourceScope // set in DAG scopes, it's the linearized parent set
|
||
|
|
||
|
name string // for debugging purposes
|
||
|
trace *trace // debug tracing
|
||
|
metrics *metrics // metrics collection
|
||
|
}
|
||
|
|
||
|
var _ network.ResourceScope = (*resourceScope)(nil)
|
||
|
var _ network.ResourceScopeSpan = (*resourceScope)(nil)
|
||
|
|
||
|
func newResourceScope(limit Limit, edges []*resourceScope, name string, trace *trace, metrics *metrics) *resourceScope {
|
||
|
for _, e := range edges {
|
||
|
e.IncRef()
|
||
|
}
|
||
|
r := &resourceScope{
|
||
|
rc: resources{limit: limit},
|
||
|
edges: edges,
|
||
|
name: name,
|
||
|
trace: trace,
|
||
|
metrics: metrics,
|
||
|
}
|
||
|
r.trace.CreateScope(name, limit)
|
||
|
return r
|
||
|
}
|
||
|
|
||
|
func newResourceScopeSpan(owner *resourceScope, id int) *resourceScope {
|
||
|
r := &resourceScope{
|
||
|
rc: resources{limit: owner.rc.limit},
|
||
|
owner: owner,
|
||
|
name: fmt.Sprintf("%s.span-%d", owner.name, id),
|
||
|
trace: owner.trace,
|
||
|
metrics: owner.metrics,
|
||
|
}
|
||
|
r.trace.CreateScope(r.name, r.rc.limit)
|
||
|
return r
|
||
|
}
|
||
|
|
||
|
// IsSpan will return true if this name was created by newResourceScopeSpan
|
||
|
func IsSpan(name string) bool {
|
||
|
return strings.Contains(name, ".span-")
|
||
|
}
|
||
|
|
||
|
func addInt64WithOverflow(a int64, b int64) (c int64, ok bool) {
|
||
|
c = a + b
|
||
|
return c, (c > a) == (b > 0)
|
||
|
}
|
||
|
|
||
|
// mulInt64WithOverflow checks for overflow in multiplying two int64s. See
|
||
|
// https://groups.google.com/g/golang-nuts/c/h5oSN5t3Au4/m/KaNQREhZh0QJ
|
||
|
func mulInt64WithOverflow(a, b int64) (c int64, ok bool) {
|
||
|
const mostPositive = 1<<63 - 1
|
||
|
const mostNegative = -(mostPositive + 1)
|
||
|
c = a * b
|
||
|
if a == 0 || b == 0 || a == 1 || b == 1 {
|
||
|
return c, true
|
||
|
}
|
||
|
if a == mostNegative || b == mostNegative {
|
||
|
return c, false
|
||
|
}
|
||
|
return c, c/b == a
|
||
|
}
|
||
|
|
||
|
// Resources implementation
|
||
|
func (rc *resources) checkMemory(rsvp int64, prio uint8) error {
|
||
|
if rsvp < 0 {
|
||
|
return fmt.Errorf("can't reserve negative memory. rsvp=%v", rsvp)
|
||
|
}
|
||
|
|
||
|
limit := rc.limit.GetMemoryLimit()
|
||
|
if limit == math.MaxInt64 {
|
||
|
// Special case where we've set max limits.
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
newmem, addOk := addInt64WithOverflow(rc.memory, rsvp)
|
||
|
|
||
|
threshold, mulOk := mulInt64WithOverflow(1+int64(prio), limit)
|
||
|
if !mulOk {
|
||
|
thresholdBig := big.NewInt(limit)
|
||
|
thresholdBig = thresholdBig.Mul(thresholdBig, big.NewInt(1+int64(prio)))
|
||
|
thresholdBig.Rsh(thresholdBig, 8) // Divide 256
|
||
|
if !thresholdBig.IsInt64() {
|
||
|
// Shouldn't happen since the threshold can only be <= limit
|
||
|
threshold = limit
|
||
|
}
|
||
|
threshold = thresholdBig.Int64()
|
||
|
} else {
|
||
|
threshold = threshold / 256
|
||
|
}
|
||
|
|
||
|
if !addOk || newmem > threshold {
|
||
|
return &ErrMemoryLimitExceeded{
|
||
|
current: rc.memory,
|
||
|
attempted: rsvp,
|
||
|
limit: limit,
|
||
|
priority: prio,
|
||
|
err: network.ErrResourceLimitExceeded,
|
||
|
}
|
||
|
}
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (rc *resources) reserveMemory(size int64, prio uint8) error {
|
||
|
if err := rc.checkMemory(size, prio); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
rc.memory += size
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (rc *resources) releaseMemory(size int64) {
|
||
|
rc.memory -= size
|
||
|
|
||
|
// sanity check for bugs upstream
|
||
|
if rc.memory < 0 {
|
||
|
log.Warn("BUG: too much memory released")
|
||
|
rc.memory = 0
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (rc *resources) addStream(dir network.Direction) error {
|
||
|
if dir == network.DirInbound {
|
||
|
return rc.addStreams(1, 0)
|
||
|
}
|
||
|
return rc.addStreams(0, 1)
|
||
|
}
|
||
|
|
||
|
func (rc *resources) addStreams(incount, outcount int) error {
|
||
|
if incount > 0 {
|
||
|
limit := rc.limit.GetStreamLimit(network.DirInbound)
|
||
|
if rc.nstreamsIn+incount > limit {
|
||
|
return &ErrStreamOrConnLimitExceeded{
|
||
|
current: rc.nstreamsIn,
|
||
|
attempted: incount,
|
||
|
limit: limit,
|
||
|
err: fmt.Errorf("cannot reserve inbound stream: %w", network.ErrResourceLimitExceeded),
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
if outcount > 0 {
|
||
|
limit := rc.limit.GetStreamLimit(network.DirOutbound)
|
||
|
if rc.nstreamsOut+outcount > limit {
|
||
|
return &ErrStreamOrConnLimitExceeded{
|
||
|
current: rc.nstreamsOut,
|
||
|
attempted: outcount,
|
||
|
limit: limit,
|
||
|
err: fmt.Errorf("cannot reserve outbound stream: %w", network.ErrResourceLimitExceeded),
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if limit := rc.limit.GetStreamTotalLimit(); rc.nstreamsIn+incount+rc.nstreamsOut+outcount > limit {
|
||
|
return &ErrStreamOrConnLimitExceeded{
|
||
|
current: rc.nstreamsIn + rc.nstreamsOut,
|
||
|
attempted: incount + outcount,
|
||
|
limit: limit,
|
||
|
err: fmt.Errorf("cannot reserve stream: %w", network.ErrResourceLimitExceeded),
|
||
|
}
|
||
|
}
|
||
|
|
||
|
rc.nstreamsIn += incount
|
||
|
rc.nstreamsOut += outcount
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (rc *resources) removeStream(dir network.Direction) {
|
||
|
if dir == network.DirInbound {
|
||
|
rc.removeStreams(1, 0)
|
||
|
} else {
|
||
|
rc.removeStreams(0, 1)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (rc *resources) removeStreams(incount, outcount int) {
|
||
|
rc.nstreamsIn -= incount
|
||
|
rc.nstreamsOut -= outcount
|
||
|
|
||
|
if rc.nstreamsIn < 0 {
|
||
|
log.Warn("BUG: too many inbound streams released")
|
||
|
rc.nstreamsIn = 0
|
||
|
}
|
||
|
if rc.nstreamsOut < 0 {
|
||
|
log.Warn("BUG: too many outbound streams released")
|
||
|
rc.nstreamsOut = 0
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (rc *resources) addConn(dir network.Direction, usefd bool) error {
|
||
|
var fd int
|
||
|
if usefd {
|
||
|
fd = 1
|
||
|
}
|
||
|
|
||
|
if dir == network.DirInbound {
|
||
|
return rc.addConns(1, 0, fd)
|
||
|
}
|
||
|
|
||
|
return rc.addConns(0, 1, fd)
|
||
|
}
|
||
|
|
||
|
func (rc *resources) addConns(incount, outcount, fdcount int) error {
|
||
|
if incount > 0 {
|
||
|
limit := rc.limit.GetConnLimit(network.DirInbound)
|
||
|
if rc.nconnsIn+incount > limit {
|
||
|
return &ErrStreamOrConnLimitExceeded{
|
||
|
current: rc.nconnsIn,
|
||
|
attempted: incount,
|
||
|
limit: limit,
|
||
|
err: fmt.Errorf("cannot reserve inbound connection: %w", network.ErrResourceLimitExceeded),
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
if outcount > 0 {
|
||
|
limit := rc.limit.GetConnLimit(network.DirOutbound)
|
||
|
if rc.nconnsOut+outcount > limit {
|
||
|
return &ErrStreamOrConnLimitExceeded{
|
||
|
current: rc.nconnsOut,
|
||
|
attempted: outcount,
|
||
|
limit: limit,
|
||
|
err: fmt.Errorf("cannot reserve outbound connection: %w", network.ErrResourceLimitExceeded),
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if connLimit := rc.limit.GetConnTotalLimit(); rc.nconnsIn+incount+rc.nconnsOut+outcount > connLimit {
|
||
|
return &ErrStreamOrConnLimitExceeded{
|
||
|
current: rc.nconnsIn + rc.nconnsOut,
|
||
|
attempted: incount + outcount,
|
||
|
limit: connLimit,
|
||
|
err: fmt.Errorf("cannot reserve connection: %w", network.ErrResourceLimitExceeded),
|
||
|
}
|
||
|
}
|
||
|
if fdcount > 0 {
|
||
|
limit := rc.limit.GetFDLimit()
|
||
|
if rc.nfd+fdcount > limit {
|
||
|
return &ErrStreamOrConnLimitExceeded{
|
||
|
current: rc.nfd,
|
||
|
attempted: fdcount,
|
||
|
limit: limit,
|
||
|
err: fmt.Errorf("cannot reserve file descriptor: %w", network.ErrResourceLimitExceeded),
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
rc.nconnsIn += incount
|
||
|
rc.nconnsOut += outcount
|
||
|
rc.nfd += fdcount
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (rc *resources) removeConn(dir network.Direction, usefd bool) {
|
||
|
var fd int
|
||
|
if usefd {
|
||
|
fd = 1
|
||
|
}
|
||
|
|
||
|
if dir == network.DirInbound {
|
||
|
rc.removeConns(1, 0, fd)
|
||
|
} else {
|
||
|
rc.removeConns(0, 1, fd)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (rc *resources) removeConns(incount, outcount, fdcount int) {
|
||
|
rc.nconnsIn -= incount
|
||
|
rc.nconnsOut -= outcount
|
||
|
rc.nfd -= fdcount
|
||
|
|
||
|
if rc.nconnsIn < 0 {
|
||
|
log.Warn("BUG: too many inbound connections released")
|
||
|
rc.nconnsIn = 0
|
||
|
}
|
||
|
if rc.nconnsOut < 0 {
|
||
|
log.Warn("BUG: too many outbound connections released")
|
||
|
rc.nconnsOut = 0
|
||
|
}
|
||
|
if rc.nfd < 0 {
|
||
|
log.Warn("BUG: too many file descriptors released")
|
||
|
rc.nfd = 0
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (rc *resources) stat() network.ScopeStat {
|
||
|
return network.ScopeStat{
|
||
|
Memory: rc.memory,
|
||
|
NumStreamsInbound: rc.nstreamsIn,
|
||
|
NumStreamsOutbound: rc.nstreamsOut,
|
||
|
NumConnsInbound: rc.nconnsIn,
|
||
|
NumConnsOutbound: rc.nconnsOut,
|
||
|
NumFD: rc.nfd,
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// resourceScope implementation
|
||
|
func (s *resourceScope) wrapError(err error) error {
|
||
|
return fmt.Errorf("%s: %w", s.name, err)
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) ReserveMemory(size int, prio uint8) error {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
if s.done {
|
||
|
return s.wrapError(network.ErrResourceScopeClosed)
|
||
|
}
|
||
|
|
||
|
if err := s.rc.reserveMemory(int64(size), prio); err != nil {
|
||
|
log.Debugw("blocked memory reservation", logValuesMemoryLimit(s.name, "", s.rc.stat(), err)...)
|
||
|
s.trace.BlockReserveMemory(s.name, prio, int64(size), s.rc.memory)
|
||
|
s.metrics.BlockMemory(size)
|
||
|
return s.wrapError(err)
|
||
|
}
|
||
|
|
||
|
if err := s.reserveMemoryForEdges(size, prio); err != nil {
|
||
|
s.rc.releaseMemory(int64(size))
|
||
|
s.metrics.BlockMemory(size)
|
||
|
return s.wrapError(err)
|
||
|
}
|
||
|
|
||
|
s.trace.ReserveMemory(s.name, prio, int64(size), s.rc.memory)
|
||
|
s.metrics.AllowMemory(size)
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) reserveMemoryForEdges(size int, prio uint8) error {
|
||
|
if s.owner != nil {
|
||
|
return s.owner.ReserveMemory(size, prio)
|
||
|
}
|
||
|
|
||
|
var reserved int
|
||
|
var err error
|
||
|
for _, e := range s.edges {
|
||
|
var stat network.ScopeStat
|
||
|
stat, err = e.ReserveMemoryForChild(int64(size), prio)
|
||
|
if err != nil {
|
||
|
log.Debugw("blocked memory reservation from constraining edge", logValuesMemoryLimit(s.name, e.name, stat, err)...)
|
||
|
break
|
||
|
}
|
||
|
|
||
|
reserved++
|
||
|
}
|
||
|
|
||
|
if err != nil {
|
||
|
// we failed because of a constraint; undo memory reservations
|
||
|
for _, e := range s.edges[:reserved] {
|
||
|
e.ReleaseMemoryForChild(int64(size))
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) releaseMemoryForEdges(size int) {
|
||
|
if s.owner != nil {
|
||
|
s.owner.ReleaseMemory(size)
|
||
|
return
|
||
|
}
|
||
|
|
||
|
for _, e := range s.edges {
|
||
|
e.ReleaseMemoryForChild(int64(size))
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) ReserveMemoryForChild(size int64, prio uint8) (network.ScopeStat, error) {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
if s.done {
|
||
|
return s.rc.stat(), s.wrapError(network.ErrResourceScopeClosed)
|
||
|
}
|
||
|
|
||
|
if err := s.rc.reserveMemory(size, prio); err != nil {
|
||
|
s.trace.BlockReserveMemory(s.name, prio, size, s.rc.memory)
|
||
|
return s.rc.stat(), s.wrapError(err)
|
||
|
}
|
||
|
|
||
|
s.trace.ReserveMemory(s.name, prio, size, s.rc.memory)
|
||
|
return network.ScopeStat{}, nil
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) ReleaseMemory(size int) {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
if s.done {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
s.rc.releaseMemory(int64(size))
|
||
|
s.releaseMemoryForEdges(size)
|
||
|
s.trace.ReleaseMemory(s.name, int64(size), s.rc.memory)
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) ReleaseMemoryForChild(size int64) {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
if s.done {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
s.rc.releaseMemory(size)
|
||
|
s.trace.ReleaseMemory(s.name, size, s.rc.memory)
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) AddStream(dir network.Direction) error {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
if s.done {
|
||
|
return s.wrapError(network.ErrResourceScopeClosed)
|
||
|
}
|
||
|
|
||
|
if err := s.rc.addStream(dir); err != nil {
|
||
|
log.Debugw("blocked stream", logValuesStreamLimit(s.name, "", dir, s.rc.stat(), err)...)
|
||
|
s.trace.BlockAddStream(s.name, dir, s.rc.nstreamsIn, s.rc.nstreamsOut)
|
||
|
return s.wrapError(err)
|
||
|
}
|
||
|
|
||
|
if err := s.addStreamForEdges(dir); err != nil {
|
||
|
s.rc.removeStream(dir)
|
||
|
return s.wrapError(err)
|
||
|
}
|
||
|
|
||
|
s.trace.AddStream(s.name, dir, s.rc.nstreamsIn, s.rc.nstreamsOut)
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) addStreamForEdges(dir network.Direction) error {
|
||
|
if s.owner != nil {
|
||
|
return s.owner.AddStream(dir)
|
||
|
}
|
||
|
|
||
|
var err error
|
||
|
var reserved int
|
||
|
for _, e := range s.edges {
|
||
|
var stat network.ScopeStat
|
||
|
stat, err = e.AddStreamForChild(dir)
|
||
|
if err != nil {
|
||
|
log.Debugw("blocked stream from constraining edge", logValuesStreamLimit(s.name, e.name, dir, stat, err)...)
|
||
|
break
|
||
|
}
|
||
|
reserved++
|
||
|
}
|
||
|
|
||
|
if err != nil {
|
||
|
for _, e := range s.edges[:reserved] {
|
||
|
e.RemoveStreamForChild(dir)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) AddStreamForChild(dir network.Direction) (network.ScopeStat, error) {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
if s.done {
|
||
|
return s.rc.stat(), s.wrapError(network.ErrResourceScopeClosed)
|
||
|
}
|
||
|
|
||
|
if err := s.rc.addStream(dir); err != nil {
|
||
|
s.trace.BlockAddStream(s.name, dir, s.rc.nstreamsIn, s.rc.nstreamsOut)
|
||
|
return s.rc.stat(), s.wrapError(err)
|
||
|
}
|
||
|
|
||
|
s.trace.AddStream(s.name, dir, s.rc.nstreamsIn, s.rc.nstreamsOut)
|
||
|
return network.ScopeStat{}, nil
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) RemoveStream(dir network.Direction) {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
if s.done {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
s.rc.removeStream(dir)
|
||
|
s.removeStreamForEdges(dir)
|
||
|
s.trace.RemoveStream(s.name, dir, s.rc.nstreamsIn, s.rc.nstreamsOut)
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) removeStreamForEdges(dir network.Direction) {
|
||
|
if s.owner != nil {
|
||
|
s.owner.RemoveStream(dir)
|
||
|
return
|
||
|
}
|
||
|
|
||
|
for _, e := range s.edges {
|
||
|
e.RemoveStreamForChild(dir)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) RemoveStreamForChild(dir network.Direction) {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
if s.done {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
s.rc.removeStream(dir)
|
||
|
s.trace.RemoveStream(s.name, dir, s.rc.nstreamsIn, s.rc.nstreamsOut)
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) AddConn(dir network.Direction, usefd bool) error {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
if s.done {
|
||
|
return s.wrapError(network.ErrResourceScopeClosed)
|
||
|
}
|
||
|
|
||
|
if err := s.rc.addConn(dir, usefd); err != nil {
|
||
|
log.Debugw("blocked connection", logValuesConnLimit(s.name, "", dir, usefd, s.rc.stat(), err)...)
|
||
|
s.trace.BlockAddConn(s.name, dir, usefd, s.rc.nconnsIn, s.rc.nconnsOut, s.rc.nfd)
|
||
|
return s.wrapError(err)
|
||
|
}
|
||
|
|
||
|
if err := s.addConnForEdges(dir, usefd); err != nil {
|
||
|
s.rc.removeConn(dir, usefd)
|
||
|
return s.wrapError(err)
|
||
|
}
|
||
|
|
||
|
s.trace.AddConn(s.name, dir, usefd, s.rc.nconnsIn, s.rc.nconnsOut, s.rc.nfd)
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) addConnForEdges(dir network.Direction, usefd bool) error {
|
||
|
if s.owner != nil {
|
||
|
return s.owner.AddConn(dir, usefd)
|
||
|
}
|
||
|
|
||
|
var err error
|
||
|
var reserved int
|
||
|
for _, e := range s.edges {
|
||
|
var stat network.ScopeStat
|
||
|
stat, err = e.AddConnForChild(dir, usefd)
|
||
|
if err != nil {
|
||
|
log.Debugw("blocked connection from constraining edge", logValuesConnLimit(s.name, e.name, dir, usefd, stat, err)...)
|
||
|
break
|
||
|
}
|
||
|
reserved++
|
||
|
}
|
||
|
|
||
|
if err != nil {
|
||
|
for _, e := range s.edges[:reserved] {
|
||
|
e.RemoveConnForChild(dir, usefd)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) AddConnForChild(dir network.Direction, usefd bool) (network.ScopeStat, error) {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
if s.done {
|
||
|
return s.rc.stat(), s.wrapError(network.ErrResourceScopeClosed)
|
||
|
}
|
||
|
|
||
|
if err := s.rc.addConn(dir, usefd); err != nil {
|
||
|
s.trace.BlockAddConn(s.name, dir, usefd, s.rc.nconnsIn, s.rc.nconnsOut, s.rc.nfd)
|
||
|
return s.rc.stat(), s.wrapError(err)
|
||
|
}
|
||
|
|
||
|
s.trace.AddConn(s.name, dir, usefd, s.rc.nconnsIn, s.rc.nconnsOut, s.rc.nfd)
|
||
|
return network.ScopeStat{}, nil
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) RemoveConn(dir network.Direction, usefd bool) {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
if s.done {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
s.rc.removeConn(dir, usefd)
|
||
|
s.removeConnForEdges(dir, usefd)
|
||
|
s.trace.RemoveConn(s.name, dir, usefd, s.rc.nconnsIn, s.rc.nconnsOut, s.rc.nfd)
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) removeConnForEdges(dir network.Direction, usefd bool) {
|
||
|
if s.owner != nil {
|
||
|
s.owner.RemoveConn(dir, usefd)
|
||
|
}
|
||
|
|
||
|
for _, e := range s.edges {
|
||
|
e.RemoveConnForChild(dir, usefd)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) RemoveConnForChild(dir network.Direction, usefd bool) {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
if s.done {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
s.rc.removeConn(dir, usefd)
|
||
|
s.trace.RemoveConn(s.name, dir, usefd, s.rc.nconnsIn, s.rc.nconnsOut, s.rc.nfd)
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) ReserveForChild(st network.ScopeStat) error {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
if s.done {
|
||
|
return s.wrapError(network.ErrResourceScopeClosed)
|
||
|
}
|
||
|
|
||
|
if err := s.rc.reserveMemory(st.Memory, network.ReservationPriorityAlways); err != nil {
|
||
|
s.trace.BlockReserveMemory(s.name, 255, st.Memory, s.rc.memory)
|
||
|
return s.wrapError(err)
|
||
|
}
|
||
|
|
||
|
if err := s.rc.addStreams(st.NumStreamsInbound, st.NumStreamsOutbound); err != nil {
|
||
|
s.trace.BlockAddStreams(s.name, st.NumStreamsInbound, st.NumStreamsOutbound, s.rc.nstreamsIn, s.rc.nstreamsOut)
|
||
|
s.rc.releaseMemory(st.Memory)
|
||
|
return s.wrapError(err)
|
||
|
}
|
||
|
|
||
|
if err := s.rc.addConns(st.NumConnsInbound, st.NumConnsOutbound, st.NumFD); err != nil {
|
||
|
s.trace.BlockAddConns(s.name, st.NumConnsInbound, st.NumConnsOutbound, st.NumFD, s.rc.nconnsIn, s.rc.nconnsOut, s.rc.nfd)
|
||
|
|
||
|
s.rc.releaseMemory(st.Memory)
|
||
|
s.rc.removeStreams(st.NumStreamsInbound, st.NumStreamsOutbound)
|
||
|
return s.wrapError(err)
|
||
|
}
|
||
|
|
||
|
s.trace.ReserveMemory(s.name, 255, st.Memory, s.rc.memory)
|
||
|
s.trace.AddStreams(s.name, st.NumStreamsInbound, st.NumStreamsOutbound, s.rc.nstreamsIn, s.rc.nstreamsOut)
|
||
|
s.trace.AddConns(s.name, st.NumConnsInbound, st.NumConnsOutbound, st.NumFD, s.rc.nconnsIn, s.rc.nconnsOut, s.rc.nfd)
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) ReleaseForChild(st network.ScopeStat) {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
if s.done {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
s.rc.releaseMemory(st.Memory)
|
||
|
s.rc.removeStreams(st.NumStreamsInbound, st.NumStreamsOutbound)
|
||
|
s.rc.removeConns(st.NumConnsInbound, st.NumConnsOutbound, st.NumFD)
|
||
|
|
||
|
s.trace.ReleaseMemory(s.name, st.Memory, s.rc.memory)
|
||
|
s.trace.RemoveStreams(s.name, st.NumStreamsInbound, st.NumStreamsOutbound, s.rc.nstreamsIn, s.rc.nstreamsOut)
|
||
|
s.trace.RemoveConns(s.name, st.NumConnsInbound, st.NumConnsOutbound, st.NumFD, s.rc.nconnsIn, s.rc.nconnsOut, s.rc.nfd)
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) ReleaseResources(st network.ScopeStat) {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
if s.done {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
s.rc.releaseMemory(st.Memory)
|
||
|
s.rc.removeStreams(st.NumStreamsInbound, st.NumStreamsOutbound)
|
||
|
s.rc.removeConns(st.NumConnsInbound, st.NumConnsOutbound, st.NumFD)
|
||
|
|
||
|
if s.owner != nil {
|
||
|
s.owner.ReleaseResources(st)
|
||
|
} else {
|
||
|
for _, e := range s.edges {
|
||
|
e.ReleaseForChild(st)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
s.trace.ReleaseMemory(s.name, st.Memory, s.rc.memory)
|
||
|
s.trace.RemoveStreams(s.name, st.NumStreamsInbound, st.NumStreamsOutbound, s.rc.nstreamsIn, s.rc.nstreamsOut)
|
||
|
s.trace.RemoveConns(s.name, st.NumConnsInbound, st.NumConnsOutbound, st.NumFD, s.rc.nconnsIn, s.rc.nconnsOut, s.rc.nfd)
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) nextSpanID() int {
|
||
|
s.spanID++
|
||
|
return s.spanID
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) BeginSpan() (network.ResourceScopeSpan, error) {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
if s.done {
|
||
|
return nil, s.wrapError(network.ErrResourceScopeClosed)
|
||
|
}
|
||
|
|
||
|
s.refCnt++
|
||
|
return newResourceScopeSpan(s, s.nextSpanID()), nil
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) Done() {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
if s.done {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
stat := s.rc.stat()
|
||
|
if s.owner != nil {
|
||
|
s.owner.ReleaseResources(stat)
|
||
|
s.owner.DecRef()
|
||
|
} else {
|
||
|
for _, e := range s.edges {
|
||
|
e.ReleaseForChild(stat)
|
||
|
e.DecRef()
|
||
|
}
|
||
|
}
|
||
|
|
||
|
s.rc.nstreamsIn = 0
|
||
|
s.rc.nstreamsOut = 0
|
||
|
s.rc.nconnsIn = 0
|
||
|
s.rc.nconnsOut = 0
|
||
|
s.rc.nfd = 0
|
||
|
s.rc.memory = 0
|
||
|
|
||
|
s.done = true
|
||
|
|
||
|
s.trace.DestroyScope(s.name)
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) Stat() network.ScopeStat {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
return s.rc.stat()
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) IncRef() {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
s.refCnt++
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) DecRef() {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
s.refCnt--
|
||
|
}
|
||
|
|
||
|
func (s *resourceScope) IsUnused() bool {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
if s.done {
|
||
|
return true
|
||
|
}
|
||
|
|
||
|
if s.refCnt > 0 {
|
||
|
return false
|
||
|
}
|
||
|
|
||
|
st := s.rc.stat()
|
||
|
return st.NumStreamsInbound == 0 &&
|
||
|
st.NumStreamsOutbound == 0 &&
|
||
|
st.NumConnsInbound == 0 &&
|
||
|
st.NumConnsOutbound == 0 &&
|
||
|
st.NumFD == 0
|
||
|
}
|