mirror of
https://github.com/turbot/steampipe.git
synced 2026-04-11 07:00:06 -04:00
51 lines
1.0 KiB
Go
51 lines
1.0 KiB
Go
package reportserver
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"time"
|
|
|
|
"github.com/gin-contrib/static"
|
|
"github.com/gin-gonic/gin"
|
|
"gopkg.in/olahol/melody.v1"
|
|
)
|
|
|
|
func StartAPI(ctx context.Context, webSocket *melody.Melody) {
|
|
router := gin.Default()
|
|
|
|
router.Use(static.Serve("/", static.LocalFile("./static", true)))
|
|
|
|
router.GET("/ws", func(c *gin.Context) {
|
|
webSocket.HandleRequest(c.Writer, c.Request)
|
|
})
|
|
|
|
srv := &http.Server{
|
|
Addr: ":5000",
|
|
Handler: router,
|
|
}
|
|
|
|
go func() {
|
|
// service connections
|
|
if err := srv.ListenAndServe(); err != nil {
|
|
log.Printf("listen: %s\n", err)
|
|
}
|
|
}()
|
|
|
|
// Wait for interrupt signal to gracefully shutdown the server with
|
|
// a timeout of 5 seconds.
|
|
quit := make(chan os.Signal)
|
|
signal.Notify(quit, os.Interrupt)
|
|
<-quit
|
|
log.Println("Shutdown Server ...")
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
if err := srv.Shutdown(ctx); err != nil {
|
|
log.Fatal("Server Shutdown:", err)
|
|
}
|
|
log.Println("Server exiting")
|
|
}
|