aboutsummaryrefslogtreecommitdiffstats
path: root/pkg/server/server.go
diff options
context:
space:
mode:
Diffstat (limited to 'pkg/server/server.go')
-rw-r--r--pkg/server/server.go8
1 files changed, 7 insertions, 1 deletions
diff --git a/pkg/server/server.go b/pkg/server/server.go
index c08d5a4..a7a5395 100644
--- a/pkg/server/server.go
+++ b/pkg/server/server.go
@@ -2,6 +2,7 @@ package server
import (
"bytes"
+ "context"
"io"
"log/slog"
"net/http"
@@ -14,6 +15,11 @@ import (
"github.com/jpappel/atlas/pkg/query"
)
+type Server interface {
+ ListenAndServe() error
+ Shutdown(context.Context) error
+}
+
func info(w http.ResponseWriter, r *http.Request) {
w.Write([]byte(`
<h1>Atlas Server</h1>
@@ -22,7 +28,7 @@ func info(w http.ResponseWriter, r *http.Request) {
`))
}
-func New(db *data.Query) *http.ServeMux {
+func NewMux(db *data.Query) *http.ServeMux {
mux := http.NewServeMux()
outputBufPool := &sync.Pool{}