mirror of
https://github.com/mochi-mqtt/server.git
synced 2025-09-27 04:26:23 +08:00
44bac0adc5
* Begin adding new slog calls * Fixed unit tests * Add leveler example * Add debug log level to Redis example * Change location of server.Close() and add logs to example/hooks * Begin removing references to zerolog * Removed final references to zerolog * Change where server.Close() occurs in main * Change to 1.21 to remove x dependency * Add slog * Update references to 1.21 * Begin change of LogAttrs to standard logging interface * Change the rest of LogAttrs to default * Fix bad log * Update badger.go Changing "data" to "key" or "id" here might be more appropriate. * Update badger.go Changing "data" to "key" or "id" here might be more appropriate. * Update server.go Not checking if err is equal to nil * Update server.go printing information for ID or error is missing. * Change references of err.Error() to err in slog * Remove missed removal of Error() references for logging --------- Co-authored-by: Derek Duncan <dduncan@atlassian.com> Co-authored-by: Derek Duncan <derekduncan@gmail.com> Co-authored-by: JB <28275108+mochi-co@users.noreply.github.com> Co-authored-by: werbenhu <werben@qq.com>
99 lines
2.0 KiB
Go
99 lines
2.0 KiB
Go
// SPDX-License-Identifier: MIT
|
|
// SPDX-FileCopyrightText: 2022 mochi-mqtt, mochi-co
|
|
// SPDX-FileContributor: jason@zgwit.com
|
|
|
|
package listeners
|
|
|
|
import (
|
|
"net"
|
|
"os"
|
|
"sync"
|
|
"sync/atomic"
|
|
|
|
"log/slog"
|
|
)
|
|
|
|
// UnixSock is a listener for establishing client connections on basic UnixSock protocol.
|
|
type UnixSock struct {
|
|
sync.RWMutex
|
|
id string // the internal id of the listener.
|
|
address string // the network address to bind to.
|
|
listen net.Listener // a net.Listener which will listen for new clients.
|
|
log *slog.Logger // server logger
|
|
end uint32 // ensure the close methods are only called once.
|
|
}
|
|
|
|
// NewUnixSock initialises and returns a new UnixSock listener, listening on an address.
|
|
func NewUnixSock(id, address string) *UnixSock {
|
|
return &UnixSock{
|
|
id: id,
|
|
address: address,
|
|
}
|
|
}
|
|
|
|
// ID returns the id of the listener.
|
|
func (l *UnixSock) ID() string {
|
|
return l.id
|
|
}
|
|
|
|
// Address returns the address of the listener.
|
|
func (l *UnixSock) Address() string {
|
|
return l.address
|
|
}
|
|
|
|
// Protocol returns the address of the listener.
|
|
func (l *UnixSock) Protocol() string {
|
|
return "unix"
|
|
}
|
|
|
|
// Init initializes the listener.
|
|
func (l *UnixSock) Init(log *slog.Logger) error {
|
|
l.log = log
|
|
|
|
var err error
|
|
_ = os.Remove(l.address)
|
|
l.listen, err = net.Listen("unix", l.address)
|
|
return err
|
|
}
|
|
|
|
// Serve starts waiting for new UnixSock connections, and calls the establish
|
|
// connection callback for any received.
|
|
func (l *UnixSock) Serve(establish EstablishFn) {
|
|
for {
|
|
if atomic.LoadUint32(&l.end) == 1 {
|
|
return
|
|
}
|
|
|
|
conn, err := l.listen.Accept()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
if atomic.LoadUint32(&l.end) == 0 {
|
|
go func() {
|
|
err = establish(l.id, conn)
|
|
if err != nil {
|
|
l.log.Warn("", "error", err)
|
|
}
|
|
}()
|
|
}
|
|
}
|
|
}
|
|
|
|
// Close closes the listener and any client connections.
|
|
func (l *UnixSock) Close(closeClients CloseFn) {
|
|
l.Lock()
|
|
defer l.Unlock()
|
|
|
|
if atomic.CompareAndSwapUint32(&l.end, 0, 1) {
|
|
closeClients(l.id)
|
|
}
|
|
|
|
if l.listen != nil {
|
|
err := l.listen.Close()
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
}
|