63 lines
1.3 KiB
Go
63 lines
1.3 KiB
Go
|
package web
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"net/http"
|
|||
|
|
"os"
|
|||
|
|
"os/signal"
|
|||
|
|
"syscall"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type Web struct {
|
|||
|
|
*WebConfig
|
|||
|
|
httpServer *http.Server
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type WebConfig struct {
|
|||
|
|
Port, ReadTimeout, WriteTimeout, IdleTimeout int
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type WebServer func(config *WebConfig) *Web
|
|||
|
|
|
|||
|
|
func (this *Web) stop() {
|
|||
|
|
c := make(chan os.Signal)
|
|||
|
|
signal.Notify(c, os.Interrupt, os.Kill, syscall.SIGHUP, syscall.SIGTERM)
|
|||
|
|
|
|||
|
|
go func() {
|
|||
|
|
defer func() {
|
|||
|
|
if err := recover(); err != nil {
|
|||
|
|
err = fmt.Errorf("internal error: %v", err)
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
s := <-c
|
|||
|
|
defer close(c)
|
|||
|
|
signal.Stop(c)
|
|||
|
|
fmt.Printf("Http Server Stop：Closed - %d\n", s)
|
|||
|
|
os.Exit(0)
|
|||
|
|
}()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (this *Web) Run(handler http.Handler) {
|
|||
|
|
this.httpServer = &http.Server{
|
|||
|
|
Addr: fmt.Sprintf(":%d", this.Port),
|
|||
|
|
Handler: handler,
|
|||
|
|
ReadTimeout: time.Duration(this.ReadTimeout) * time.Second,
|
|||
|
|
WriteTimeout: time.Duration(this.WriteTimeout) * time.Second,
|
|||
|
|
IdleTimeout: time.Duration(this.IdleTimeout) * time.Second,
|
|||
|
|
MaxHeaderBytes: 1 << 20,
|
|||
|
|
}
|
|||
|
|
this.stop()
|
|||
|
|
|
|||
|
|
fmt.Println("Http Server Start")
|
|||
|
|
fmt.Printf("Http Server Address - %v\n", fmt.Sprintf("http://127.0.0.1:%d", this.Port))
|
|||
|
|
|
|||
|
|
_ = this.httpServer.ListenAndServe()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewWeb() WebServer {
|
|||
|
|
return func(config *WebConfig) *Web {
|
|||
|
|
return &Web{WebConfig: config}
|
|||
|
|
}
|
|||
|
|
}
|