mirror of
https://github.com/onsonr/sonr.git
synced 2025-03-10 13:07:09 +00:00
53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
package proxy
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"time"
|
|
|
|
"github.com/labstack/echo/v4"
|
|
"github.com/labstack/gommon/log"
|
|
"github.com/onsonr/sonr/nebula"
|
|
"github.com/onsonr/sonr/nebula/pages"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func NewProxyCmd() *cobra.Command {
|
|
return &cobra.Command{
|
|
Use: "dwn-proxy",
|
|
Short: "Starts the DWN proxy server for the local IPFS node",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
// Echo instance
|
|
e := echo.New()
|
|
e.Logger.SetLevel(log.INFO)
|
|
|
|
// Configure the server
|
|
nebula.UseAssets(e)
|
|
e.GET("/", pages.Home)
|
|
e.GET("/allocate", pages.Profile)
|
|
|
|
// Start server
|
|
ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt)
|
|
defer stop()
|
|
// Start server
|
|
go func() {
|
|
if err := e.Start(":1323"); err != nil && err != http.ErrServerClosed {
|
|
e.Logger.Fatal("shutting down the server")
|
|
}
|
|
}()
|
|
|
|
// Wait for interrupt signal to gracefully shutdown the server with a timeout of 10 seconds.
|
|
<-ctx.Done()
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
|
|
// Shutdown the server with 10 seconds timeout.
|
|
if err := e.Shutdown(ctx); err != nil {
|
|
e.Logger.Fatal(err)
|
|
}
|
|
},
|
|
}
|
|
}
|