Merge pull request #2541 from lifenjoiner/chan

Use blocking channel instead of looping sleep for less CPU usage
This commit is contained in:
Frank Denis 2023-12-03 17:24:10 +01:00 committed by GitHub
commit 42b6ae9052
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 3 deletions

View File

@ -17,14 +17,13 @@ type CaptivePortalMap map[string]CaptivePortalEntryIPs
type CaptivePortalHandler struct {
cancelChannel chan struct{}
countChannel chan struct{}
waitChannel chan struct{}
channelCount int
}
func (captivePortalHandler *CaptivePortalHandler) Stop() {
close(captivePortalHandler.cancelChannel)
for len(captivePortalHandler.countChannel) < captivePortalHandler.channelCount {
time.Sleep(10 * time.Millisecond)
}
<-captivePortalHandler.waitChannel
close(captivePortalHandler.countChannel)
}
@ -138,6 +137,9 @@ func addColdStartListener(
}
clientPc.Close()
captivePortalHandler.countChannel <- struct{}{}
if len(captivePortalHandler.countChannel) == captivePortalHandler.channelCount {
close(captivePortalHandler.waitChannel)
}
}()
return nil
}
@ -186,6 +188,7 @@ func ColdStart(proxy *Proxy) (*CaptivePortalHandler, error) {
captivePortalHandler := CaptivePortalHandler{
cancelChannel: make(chan struct{}),
countChannel: make(chan struct{}, len(listenAddrStrs)),
waitChannel: make(chan struct{}),
channelCount: 0,
}
for _, listenAddrStr := range listenAddrStrs {