mirror of
https://github.com/hanzozt/dilithium.git
synced 2026-10-30 20:38:17 +00:00
zeekay
5ee674cbcf
Global rebrand of all OpenZiti references to Hanzo ZT: - Go module paths: github.com/openziti/* → github.com/hanzozt/* - Docker images: openziti/* → hanzozt/* - Java packages: org.openziti.* → org.hanzozt.* - Brand: OpenZiti → Hanzo ZT - URLs: openziti.io → hanzozt.dev
119 lines
2.5 KiB
Go
119 lines
2.5 KiB
Go
package westworld3
|
|
|
|
import (
|
|
"github.com/hanzozt/dilithium/util"
|
|
"github.com/sirupsen/logrus"
|
|
"time"
|
|
)
|
|
|
|
const notClosed = int32(-33)
|
|
|
|
type closer struct {
|
|
seq *util.Sequence
|
|
rxCloseSeq int32
|
|
rxCloseSeqIn chan int32
|
|
txCloseSeq int32
|
|
txCloseSeqIn chan int32
|
|
txPortal *txPortal
|
|
rxPortal *rxPortal
|
|
lastEvent time.Time
|
|
profile *Profile
|
|
closeHook func()
|
|
}
|
|
|
|
func newCloser(seq *util.Sequence, profile *Profile, closeHook func()) *closer {
|
|
return &closer{
|
|
seq: seq,
|
|
rxCloseSeq: notClosed,
|
|
rxCloseSeqIn: make(chan int32, 1),
|
|
txCloseSeq: notClosed,
|
|
txCloseSeqIn: make(chan int32, 1),
|
|
profile: profile,
|
|
closeHook: closeHook,
|
|
}
|
|
}
|
|
|
|
func (self *closer) emergencyStop() {
|
|
logrus.Infof("broken glass")
|
|
|
|
self.txPortal.close()
|
|
self.rxPortal.close()
|
|
|
|
if self.closeHook != nil {
|
|
self.closeHook()
|
|
}
|
|
}
|
|
|
|
func (self *closer) timeout() {
|
|
logrus.Infof("timeout")
|
|
|
|
self.txPortal.close()
|
|
self.rxPortal.close()
|
|
|
|
if self.closeHook != nil {
|
|
self.closeHook()
|
|
}
|
|
}
|
|
|
|
func (self *closer) run() {
|
|
logrus.Info("started")
|
|
defer logrus.Info("exited")
|
|
|
|
closeWait:
|
|
for {
|
|
select {
|
|
case rxCloseSeq, ok := <-self.rxCloseSeqIn:
|
|
if !ok {
|
|
logrus.Info("unexpected closed rx close seq")
|
|
break closeWait
|
|
}
|
|
self.rxCloseSeq = rxCloseSeq
|
|
self.lastEvent = time.Now()
|
|
logrus.Infof("got rx close seq: %d", rxCloseSeq)
|
|
if self.txCloseSeq == notClosed {
|
|
if err := self.txPortal.sendClose(self.seq); err != nil {
|
|
logrus.Errorf("error sending close (%v)", err)
|
|
}
|
|
}
|
|
if self.readyToClose() {
|
|
break closeWait
|
|
}
|
|
|
|
case txCloseSeq, ok := <-self.txCloseSeqIn:
|
|
if !ok {
|
|
logrus.Infof("unexpected closed tx close seq")
|
|
break closeWait
|
|
}
|
|
self.txCloseSeq = txCloseSeq
|
|
self.lastEvent = time.Now()
|
|
logrus.Infof("got tx close seq: %d", txCloseSeq)
|
|
if self.readyToClose() {
|
|
break closeWait
|
|
}
|
|
|
|
case <-time.After(time.Duration(self.profile.CloseCheckMs) * time.Millisecond):
|
|
if self.readyToClose() {
|
|
break closeWait
|
|
}
|
|
}
|
|
}
|
|
logrus.Info("ready to close")
|
|
|
|
self.txPortal.close()
|
|
self.rxPortal.close()
|
|
|
|
if self.closeHook != nil {
|
|
self.closeHook()
|
|
}
|
|
|
|
logrus.Info("close complete")
|
|
}
|
|
|
|
func (self *closer) readyToClose() bool {
|
|
if (self.txCloseSeq != notClosed || self.rxCloseSeq != notClosed) && time.Since(self.lastEvent).Milliseconds() > 15000 {
|
|
return true
|
|
} else {
|
|
return self.txCloseSeq != notClosed && self.rxCloseSeq != notClosed && time.Since(self.lastEvent).Milliseconds() > int64(self.profile.CloseWaitMs)
|
|
}
|
|
}
|