aljaz/khatru
1
0
Fork 0
mirror of https://github.com/aljazceru/khatru.git synced 2025-12-25 01:04:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

expose Router so servers can inject their own routes.

Browse Source
This commit is contained in:
fiatjaf
2021-12-26 20:56:42 -03:00
parent d9a6c6a1ad
commit 80e643c492
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
start.go
View File

@@ -19,7 +19,7 @@ type Settings struct {
var s Settings
var log = zerolog.New(os.Stderr).Output(zerolog.ConsoleWriter{Out: os.Stderr})
var router = mux.NewRouter()
var Router = mux.NewRouter()
func Start(relay Relay) {
if err := envconfig.Process("", &s); err != nil {
@@ -32,7 +32,8 @@ func Start(relay Relay) {
Log.Fatal().Err(err).Msg("failed to start")
}
router.Path("/").Methods("GET").HandlerFunc(handleWebsocket(relay))
Router.Path("/").Methods("GET").Headers("Upgrade", "websocket").
HandlerFunc(handleWebsocket(relay))
if inj, ok := relay.(Injector); ok {
go func() {
@@ -43,7 +44,7 @@ func Start(relay Relay) {
}
srv := &http.Server{
Handler: cors.Default().Handler(router),
Handler: cors.Default().Handler(Router),
Addr: s.Host + ":" + s.Port,
WriteTimeout: 2 * time.Second,
ReadTimeout: 2 * time.Second,