mirror of
https://github.com/openziti/ziti.git
synced 2026-10-12 15:28:05 +00:00
Paul Lorenz
549260d650
* Persist terminator precedence. Fix link cost use/scaling. Fix link removal related panic. Fix service cache clearing on terminator change. Fixes GH-112 Fixes GH-121 Fixes GH-122 Fixes GH-123 Fixes GH-124
130 lines
2.7 KiB
Go
130 lines
2.7 KiB
Go
/*
|
|
Copyright NetFoundry, Inc.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
https://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package network
|
|
|
|
import (
|
|
"github.com/openziti/foundation/identity/identity"
|
|
"github.com/openziti/foundation/util/info"
|
|
"sync/atomic"
|
|
)
|
|
|
|
type Link struct {
|
|
Id *identity.TokenId
|
|
Src *Router
|
|
Dst *Router
|
|
state []*LinkState
|
|
Down bool
|
|
StaticCost int32
|
|
SrcLatency int64
|
|
DstLatency int64
|
|
Cost int64
|
|
}
|
|
|
|
func newLink(id *identity.TokenId) *Link {
|
|
l := &Link{
|
|
Id: id,
|
|
state: make([]*LinkState, 0),
|
|
Down: false,
|
|
StaticCost: 1,
|
|
Cost: 1,
|
|
}
|
|
l.addState(&LinkState{Mode: Pending, Timestamp: info.NowInMilliseconds()})
|
|
return l
|
|
}
|
|
|
|
func (link *Link) CurrentState() *LinkState {
|
|
if link.state == nil || len(link.state) < 1 {
|
|
return nil
|
|
}
|
|
return link.state[0]
|
|
}
|
|
|
|
func (link *Link) addState(s *LinkState) {
|
|
if link.state == nil {
|
|
link.state = make([]*LinkState, 0)
|
|
}
|
|
link.state = append([]*LinkState{s}, link.state...)
|
|
}
|
|
|
|
func (link *Link) GetStaticCost() int32 {
|
|
return atomic.LoadInt32(&link.StaticCost)
|
|
}
|
|
|
|
func (link *Link) SetStaticCost(cost int32) {
|
|
atomic.StoreInt32(&link.StaticCost, cost)
|
|
link.recalculateCost()
|
|
}
|
|
|
|
func (link *Link) GetSrcLatency() int64 {
|
|
return atomic.LoadInt64(&link.SrcLatency)
|
|
}
|
|
|
|
func (link *Link) SetSrcLatency(latency int64) {
|
|
atomic.StoreInt64(&link.SrcLatency, latency)
|
|
link.recalculateCost()
|
|
}
|
|
|
|
func (link *Link) GetDstLatency() int64 {
|
|
return atomic.LoadInt64(&link.DstLatency)
|
|
}
|
|
|
|
func (link *Link) SetDstLatency(latency int64) {
|
|
atomic.StoreInt64(&link.DstLatency, latency)
|
|
link.recalculateCost()
|
|
}
|
|
|
|
func (link *Link) recalculateCost() {
|
|
cost := int64(link.GetStaticCost()) + link.GetSrcLatency()/1_000_000 + link.GetDstLatency()/1_000_000
|
|
atomic.StoreInt64(&link.Cost, cost)
|
|
}
|
|
|
|
func (link *Link) GetCost() int64 {
|
|
return atomic.LoadInt64(&link.Cost)
|
|
}
|
|
|
|
type LinkMode byte
|
|
|
|
const (
|
|
Pending LinkMode = iota
|
|
Connected
|
|
Failed
|
|
)
|
|
|
|
func (t LinkMode) String() string {
|
|
if t == Pending {
|
|
return "Pending"
|
|
} else if t == Connected {
|
|
return "Connected"
|
|
} else if t == Failed {
|
|
return "Failed"
|
|
} else {
|
|
return ""
|
|
}
|
|
}
|
|
|
|
type LinkState struct {
|
|
Mode LinkMode
|
|
Timestamp int64
|
|
}
|
|
|
|
func newLinkState(mode LinkMode) *LinkState {
|
|
return &LinkState{
|
|
Mode: mode,
|
|
Timestamp: info.NowInMilliseconds(),
|
|
}
|
|
}
|