49 lines
911 B
Go
49 lines
911 B
Go
|
package server
|
||
|
|
|
||
|
|
import (
|
||
|
|
"crypto/tls"
|
||
|
|
"net"
|
||
|
|
"net/http"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Asynchronous HTTP server that can be started and stopped asynchronously.
|
||
|
|
type AsyncServer struct {
|
||
|
|
http.Server
|
||
|
|
listener net.Listener
|
||
|
|
stopped chan bool
|
||
|
|
}
|
||
|
|
|
||
|
|
// Create a new server instance. Note that Start() must be called before the
|
||
|
|
// server will begin accepting new connections.
|
||
|
|
func New(addr string) *AsyncServer {
|
||
|
|
a := &AsyncServer{
|
||
|
|
stopped: make(chan bool),
|
||
|
|
}
|
||
|
|
a.Addr = addr
|
||
|
|
return a
|
||
|
|
}
|
||
|
|
|
||
|
|
// Start the server.
|
||
|
|
func (a *AsyncServer) Start() error {
|
||
|
|
l, err := net.Listen("tcp", a.Addr)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
a.Addr = l.Addr().String()
|
||
|
|
if a.TLSConfig != nil {
|
||
|
|
l = tls.NewListener(l, a.TLSConfig)
|
||
|
|
}
|
||
|
|
a.listener = l
|
||
|
|
go func() {
|
||
|
|
a.Serve(a.listener)
|
||
|
|
a.stopped <- true
|
||
|
|
}()
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Stop the server. This method blocks until the server is stopped.
|
||
|
|
func (a *AsyncServer) Stop() {
|
||
|
|
a.listener.Close()
|
||
|
|
<-a.stopped
|
||
|
|
}
|