forked from TrueCloudLab/frostfs-node
40 lines
644 B
Go
40 lines
644 B
Go
|
package transport
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync/atomic"
|
||
|
|
|
||
|
|
manet "github.com/multiformats/go-multiaddr-net"
|
||
|
|
)
|
||
|
|
|
||
|
|
type (
|
||
|
|
// Connection is an interface of network connection.
|
||
|
|
Connection interface {
|
||
|
|
manet.Conn
|
||
|
|
Closed() bool
|
||
|
|
}
|
||
|
|
|
||
|
|
conn struct {
|
||
|
|
manet.Conn
|
||
|
|
closed *int32
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
func newConnection(con manet.Conn) Connection {
|
||
|
|
return &conn{
|
||
|
|
Conn: con,
|
||
|
|
closed: new(int32),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Closed checks that connection closed.
|
||
|
|
func (c *conn) Closed() bool { return atomic.LoadInt32(c.closed) == 1 }
|
||
|
|
|
||
|
|
// Close connection and write state.
|
||
|
|
func (c *conn) Close() error {
|
||
|
|
if atomic.CompareAndSwapInt32(c.closed, 0, 1) {
|
||
|
|
return c.Conn.Close()
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|