mirror of
https://github.com/strukturag/nextcloud-spreed-signaling
synced 2024-05-04 14:53:10 +02:00
Add helper class to handle close channels.
This commit is contained in:
parent
2a47829164
commit
e6b2d1e0aa
47
closer.go
Normal file
47
closer.go
Normal file
|
@ -0,0 +1,47 @@
|
||||||
|
/**
|
||||||
|
* Standalone signaling server for the Nextcloud Spreed app.
|
||||||
|
* Copyright (C) 2023 struktur AG
|
||||||
|
*
|
||||||
|
* @author Joachim Bauch <bauch@struktur.de>
|
||||||
|
*
|
||||||
|
* @license GNU AGPL version 3 or any later version
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU Affero General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU Affero General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU Affero General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
package signaling
|
||||||
|
|
||||||
|
import (
|
||||||
|
"sync/atomic"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Closer struct {
|
||||||
|
closed uint32
|
||||||
|
C chan struct{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewCloser() *Closer {
|
||||||
|
return &Closer{
|
||||||
|
C: make(chan struct{}),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Closer) IsClosed() bool {
|
||||||
|
return atomic.LoadUint32(&c.closed) != 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Closer) Close() {
|
||||||
|
if atomic.CompareAndSwapUint32(&c.closed, 0, 1) {
|
||||||
|
close(c.C)
|
||||||
|
}
|
||||||
|
}
|
62
closer_test.go
Normal file
62
closer_test.go
Normal file
|
@ -0,0 +1,62 @@
|
||||||
|
/**
|
||||||
|
* Standalone signaling server for the Nextcloud Spreed app.
|
||||||
|
* Copyright (C) 2023 struktur AG
|
||||||
|
*
|
||||||
|
* @author Joachim Bauch <bauch@struktur.de>
|
||||||
|
*
|
||||||
|
* @license GNU AGPL version 3 or any later version
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU Affero General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU Affero General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU Affero General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
package signaling
|
||||||
|
|
||||||
|
import (
|
||||||
|
"sync"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestCloserMulti(t *testing.T) {
|
||||||
|
closer := NewCloser()
|
||||||
|
|
||||||
|
var wg sync.WaitGroup
|
||||||
|
count := 10
|
||||||
|
for i := 0; i < count; i++ {
|
||||||
|
wg.Add(1)
|
||||||
|
go func() {
|
||||||
|
defer wg.Done()
|
||||||
|
<-closer.C
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
|
||||||
|
if closer.IsClosed() {
|
||||||
|
t.Error("should not be closed")
|
||||||
|
}
|
||||||
|
closer.Close()
|
||||||
|
if !closer.IsClosed() {
|
||||||
|
t.Error("should be closed")
|
||||||
|
}
|
||||||
|
wg.Wait()
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCloserCloseBeforeWait(t *testing.T) {
|
||||||
|
closer := NewCloser()
|
||||||
|
closer.Close()
|
||||||
|
if !closer.IsClosed() {
|
||||||
|
t.Error("should be closed")
|
||||||
|
}
|
||||||
|
<-closer.C
|
||||||
|
if !closer.IsClosed() {
|
||||||
|
t.Error("should be closed")
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in a new issue