-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
48 lines (40 loc) · 1.28 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
package main
import (
"flag"
"net"
"os"
"strconv"
"github.com/rs/zerolog"
"github.com/spangenberg/arrowflight-server-reference-implementation/handler"
"github.com/spangenberg/arrowflight-server-reference-implementation/internal"
"github.com/spangenberg/arrowflight-server-reference-implementation/internal/noophandler"
)
func main() {
var (
authToken = flag.String("auth-token", "foobar", "auth token to use for the server")
host = flag.String("host", "localhost", "hostname to bind to")
port = flag.Int("port", 9090, "port to bind to")
)
flag.Parse()
var logger zerolog.Logger
{
var closer func()
out := os.Stdout
logger = zerolog.New(zerolog.ConsoleWriter{Out: out}).With().Timestamp().Logger()
closer = func() {
if err := out.Close(); err != nil {
logger.Error().Err(err).Msg("failed to close stdout")
}
}
defer closer()
}
// Create a new handler with no-op implementations.
// Swap these out with your own implementations.
myHandler := handler.Handler{
CloudQueryHandler: &noophandler.CloudQuery{},
ArrowFlightHandler: &noophandler.ArrowFlight{},
}
if err := internal.Serve(logger, net.JoinHostPort(*host, strconv.Itoa(*port)), *authToken, myHandler); err != nil {
logger.Fatal().Err(err).Msg("failed to start flight server")
}
}