Add example for httpserver

test/test-httpserver/main.go

@@ -1 +1,145 @@
-package test_httpserver
+/***********************************************************************************************************************
+ *
+ *   MIT License
+ *
+ *   Copyright (c) 2021 Nicolas JUHEL
+ *
+ *   Permission is hereby granted, free of charge, to any person obtaining a copy
+ *   of this software and associated documentation files (the "Software"), to deal
+ *   in the Software without restriction, including without limitation the rights
+ *   to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ *   copies of the Software, and to permit persons to whom the Software is
+ *   furnished to do so, subject to the following conditions:
+ *
+ *   The above copyright notice and this permission notice shall be included in all
+ *   copies or substantial portions of the Software.
+ *
+ *   THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ *   IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ *   FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ *   AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ *   LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ *   OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ *   SOFTWARE.
+ *
+ *
+ **********************************************************************************************************************/
+
+package main
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+	"time"
+
+	liblog "github.com/nabbar/golib/logger"
+
+	libtls "github.com/nabbar/golib/certificates"
+	liberr "github.com/nabbar/golib/errors"
+	libsrv "github.com/nabbar/golib/httpserver"
+)
+
+var tlsConfigSrv = libtls.Config{
+	InheritDefault: true,
+	VersionMin:     "1.2",
+}
+
+var cfgSrv01 = libsrv.ServerConfig{
+	Name:   "test-01",
+	Listen: "0.0.0.0:61001",
+	Expose: "0.0.0.0:61000",
+	TLS:    tlsConfigSrv,
+}
+
+var cfgSrv02 = libsrv.ServerConfig{
+	Name:   "test-02",
+	Listen: "0.0.0.0:61002",
+	Expose: "0.0.0.0:61000",
+	TLS:    tlsConfigSrv,
+}
+
+var (
+	cfgPool libsrv.PoolServerConfig
+	ctx     context.Context
+	cnl     context.CancelFunc
+)
+
+func init() {
+	liblog.EnableColor()
+	liblog.SetLevel(liblog.DebugLevel)
+	liblog.AddGID(true)
+	liblog.FileTrace(true)
+	liblog.SetFormat(liblog.TextFormat)
+	liblog.Timestamp(true)
+
+	liberr.SetModeReturnError(liberr.ErrorReturnCodeErrorTraceFull)
+
+	ctx, cnl = context.WithCancel(context.Background())
+
+	cfgPool = libsrv.PoolServerConfig{cfgSrv01, cfgSrv02}
+	cfgPool.MapUpdate(func(cfg libsrv.ServerConfig) libsrv.ServerConfig {
+		cfg.SetParentContext(func() context.Context {
+			return ctx
+		})
+		return cfg
+	})
+
+}
+
+func main() {
+	var (
+		pool libsrv.PoolServer
+		lerr liberr.Error
+	)
+
+	if pool, lerr = cfgPool.PoolServer(); lerr != nil {
+		panic(lerr)
+	}
+
+	mux := http.NewServeMux()
+	mux.HandleFunc("/hello", hello)
+	mux.HandleFunc("/headers", headers)
+
+	if lerr = pool.Listen(mux); lerr != nil {
+		panic(lerr)
+	}
+
+	defer func() {
+		pool.Shutdown()
+		cnl()
+	}()
+
+	go pool.WaitNotify(ctx, cnl)
+
+	var i = 0
+	for {
+
+		time.Sleep(5 * time.Second)
+
+		if ctx.Err() != nil {
+			return
+		}
+
+		fmt.Printf("Srv Name : %v\n", pool.List(libsrv.FieldBind, libsrv.FieldName, "", ".*"))
+		i++
+
+		if i%3 == 0 {
+			fmt.Printf("Reloading Server : %v\n", pool.List(libsrv.FieldBind, libsrv.FieldName, "", ".*"))
+			pool.Restart()
+			i = 0
+		}
+	}
+}
+
+func hello(w http.ResponseWriter, req *http.Request) {
+	_, _ = fmt.Fprintf(w, "hello\n")
+}
+
+func headers(w http.ResponseWriter, req *http.Request) {
+	for name, headers := range req.Header {
+		for _, h := range headers {
+			_, _ = fmt.Fprintf(w, "%v: %v\n", name, h)
+		}
+	}
+}