highperfocused/khatru
1
0
Fork 0
mirror of https://github.com/fiatjaf/khatru.git synced 2026-06-04 09:41:28 +02: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 s Settings
var log = zerolog.New(os.Stderr).Output(zerolog.ConsoleWriter{Out: os.Stderr}) var log = zerolog.New(os.Stderr).Output(zerolog.ConsoleWriter{Out: os.Stderr})
var router = mux.NewRouter() var Router = mux.NewRouter()
func Start(relay Relay) { func Start(relay Relay) {
if err := envconfig.Process("", &s); err != nil { if err := envconfig.Process("", &s); err != nil {
@@ -32,7 +32,8 @@ func Start(relay Relay) {
Log.Fatal().Err(err).Msg("failed to start") 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 { if inj, ok := relay.(Injector); ok {
go func() { go func() {
@@ -43,7 +44,7 @@ func Start(relay Relay) {
} }
srv := &http.Server{ srv := &http.Server{
Handler: cors.Default().Handler(router), Handler: cors.Default().Handler(Router),
Addr: s.Host + ":" + s.Port, Addr: s.Host + ":" + s.Port,
WriteTimeout: 2 * time.Second, WriteTimeout: 2 * time.Second,
ReadTimeout: 2 * time.Second, ReadTimeout: 2 * time.Second,