58 lines
1.0 KiB
Go
58 lines
1.0 KiB
Go
package admin
|
|
|
|
import (
|
|
"net"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/rakyll/statik/fs"
|
|
|
|
// Register static assets via statik.
|
|
_ "github.com/influxdb/influxdb/statik"
|
|
)
|
|
|
|
// Server manages InfluxDB's admin web server.
|
|
type Server struct {
|
|
port string
|
|
listener net.Listener
|
|
closed bool
|
|
}
|
|
|
|
// NewServer constructs a new admin web server. The "port" argument should be a
|
|
// string that looks like ":8083" or whatever port to serve on.
|
|
func NewServer(port string) *Server {
|
|
return &Server{port: port, closed: true}
|
|
}
|
|
|
|
// ListenAndServe starts the admin web server and serves requests until
|
|
// s.Close() is called.
|
|
func (s *Server) ListenAndServe() {
|
|
if s.port == "" {
|
|
return
|
|
}
|
|
|
|
var err error
|
|
s.listener, err = net.Listen("tcp", s.port)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
s.closed = false
|
|
statikFS, _ := fs.New()
|
|
|
|
err = http.Serve(s.listener, http.FileServer(statikFS))
|
|
if !strings.Contains(err.Error(), "closed") {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
// Close stops the admin web server.
|
|
func (s *Server) Close() {
|
|
if s.closed {
|
|
return
|
|
}
|
|
|
|
s.closed = true
|
|
s.listener.Close()
|
|
}
|