mirror of
https://github.com/featurebasedb/featurebase.git
synced 2026-10-15 23:38:05 +00:00
4c53f86e82
i used this script, a little clunky but it got the job done
```bash
for file in `find . -type f -print | grep '\.go'`; do
sed '1,/^\/\/ limitations under the License.$/d' $file > $file.tmp;
result=`cat $file.tmp`
if [[ result != "" ]]; then
gofmt $file.tmp &> /dev/null;
if [[ $? == 0 ]]; then
mv $file.tmp $file && gofmt -w $file;
else
rm $file.tmp;
fi
else
rm $file.tmp;
fi
done
```
57 lines
1 KiB
Go
57 lines
1 KiB
Go
package pgtest
|
|
|
|
import (
|
|
"errors"
|
|
"net"
|
|
"sync"
|
|
)
|
|
|
|
// errListenerClosed is an error returned when the listener is closed.
|
|
var errListenerClosed = errors.New("listener closed")
|
|
|
|
type inMemoryListener struct {
|
|
ch chan net.Conn
|
|
closed chan struct{}
|
|
once sync.Once
|
|
}
|
|
|
|
func (l *inMemoryListener) Accept() (net.Conn, error) {
|
|
select {
|
|
case <-l.closed:
|
|
return nil, errListenerClosed
|
|
default:
|
|
}
|
|
select {
|
|
case conn := <-l.ch:
|
|
return conn, nil
|
|
case <-l.closed:
|
|
return nil, errListenerClosed
|
|
}
|
|
}
|
|
|
|
func (l *inMemoryListener) Close() error {
|
|
l.once.Do(func() { close(l.closed) })
|
|
|
|
return nil
|
|
}
|
|
|
|
type memAddr struct{}
|
|
|
|
func (a memAddr) Network() string { return "memory" }
|
|
func (a memAddr) String() string { return "memory" }
|
|
|
|
func (l *inMemoryListener) Addr() net.Addr {
|
|
return memAddr{}
|
|
}
|
|
|
|
func (l *inMemoryListener) Dial() (net.Conn, error) {
|
|
serverConn, clientConn := net.Pipe()
|
|
select {
|
|
case l.ch <- serverConn:
|
|
return clientConn, nil
|
|
case <-l.closed:
|
|
serverConn.Close()
|
|
clientConn.Close()
|
|
return nil, errListenerClosed
|
|
}
|
|
}
|