[update] add plugin config (#1180)

Signed-off-by: lihaowei <haoweili35@gmail.com>
This commit is contained in:
Howie
2021-10-31 12:56:25 +08:00
committed by GitHub
parent 8230474667
commit cd1f8da13f
4 changed files with 58 additions and 5 deletions

View File

@@ -2,6 +2,7 @@ package internal
import (
"context"
"crypto/tls"
"fmt"
"net/http"
@@ -10,24 +11,27 @@ import (
// StartHttp starts a http server.
func StartHttp(host string, port int, handler http.Handler) error {
return start(host, port, handler, func(srv *http.Server) error {
return start(host, port, handler, nil, func(srv *http.Server) error {
return srv.ListenAndServe()
})
}
// StartHttps starts a https server.
func StartHttps(host string, port int, certFile, keyFile string, handler http.Handler) error {
return start(host, port, handler, func(srv *http.Server) error {
func StartHttps(host string, port int, certFile, keyFile string, tlsConfig *tls.Config, handler http.Handler) error {
return start(host, port, handler, tlsConfig, func(srv *http.Server) error {
// certFile and keyFile are set in buildHttpsServer
return srv.ListenAndServeTLS(certFile, keyFile)
})
}
func start(host string, port int, handler http.Handler, run func(srv *http.Server) error) (err error) {
func start(host string, port int, handler http.Handler, tlsConfig *tls.Config, run func(srv *http.Server) error) (err error) {
server := &http.Server{
Addr: fmt.Sprintf("%s:%d", host, port),
Handler: handler,
}
if tlsConfig != nil {
server.TLSConfig = tlsConfig
}
waitForCalled := proc.AddWrapUpListener(func() {
server.Shutdown(context.Background())
})