initial import

example/graceful/etcd/api/handler/gracefulhandler.go

@@ -0,0 +1,49 @@
+package handler
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+	"os"
+	"time"
+
+	"zero/core/executors"
+	"zero/core/httpx"
+	"zero/core/logx"
+	"zero/example/graceful/etcd/api/svc"
+	"zero/example/graceful/etcd/api/types"
+	"zero/example/graceful/etcd/rpc/graceful"
+)
+
+func gracefulHandler(ctx *svc.ServiceContext) http.HandlerFunc {
+	logger := executors.NewLessExecutor(time.Second)
+	return func(w http.ResponseWriter, r *http.Request) {
+		var resp types.Response
+
+		conn, ok := ctx.Client.Next()
+		if !ok {
+			http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
+			return
+		}
+
+		host, err := os.Hostname()
+		if err != nil {
+			http.Error(w, http.StatusText(http.StatusNotImplemented), http.StatusNotImplemented)
+			return
+		}
+
+		client := graceful.NewGraceServiceClient(conn)
+		rp, err := client.Grace(context.Background(), &graceful.Request{From: host})
+		if err != nil {
+			logx.Error(err)
+			http.Error(w, http.StatusText(http.StatusBadGateway), http.StatusBadGateway)
+			return
+		}
+
+		resp.Host = rp.Host
+		logger.DoOrDiscard(func() {
+			fmt.Printf("%s from host: %s\n", time.Now().Format("15:04:05"), rp.Host)
+		})
+		httpx.OkJson(w, resp)
+	}
+}

example/graceful/etcd/api/handler/routes.go

@@ -0,0 +1,19 @@
+// DO NOT EDIT, generated by goctl
+package handler
+
+import (
+	"net/http"
+
+	"zero/example/graceful/etcd/api/svc"
+	"zero/ngin"
+)
+
+func RegisterHandlers(engine *ngin.Engine, ctx *svc.ServiceContext) {
+	engine.AddRoutes([]ngin.Route{
+		{
+			Method:  http.MethodGet,
+			Path:    "/api/graceful",
+			Handler: gracefulHandler(ctx),
+		},
+	})
+}