Create & Init Project...

app/service/main/dapper-query/util/BUILD

@@ -0,0 +1,32 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["sessionid.go"],
+    importpath = "go-common/app/service/main/dapper-query/util",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//library/ecode:go_default_library",
+        "//library/net/http/blademaster:go_default_library",
+    ],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+    visibility = ["//visibility:private"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [":package-srcs"],
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+)

app/service/main/dapper-query/util/sessionid.go

@@ -0,0 +1,42 @@
+package util
+
+import (
+	"context"
+
+	"go-common/library/ecode"
+	bm "go-common/library/net/http/blademaster"
+)
+
+const (
+	_ajsSessioID = "_AJSESSIONID"
+)
+
+type sessionKeyT string
+
+var sessionKey sessionKeyT = "sessionID"
+
+// SessionIDMiddleware extrace session from cookie set to context
+func SessionIDMiddleware(c *bm.Context) {
+	cookie, err := c.Request.Cookie(_ajsSessioID)
+	if err != nil {
+		c.JSON(nil, ecode.AccessDenied)
+		c.Abort()
+		return
+	}
+	c.Context = SessionIDWithContext(c.Context, cookie.Value)
+}
+
+// SessionIDFromContext get session id from context
+func SessionIDFromContext(ctx context.Context) string {
+	if val := ctx.Value(sessionKey); val != nil {
+		if sessionID, ok := val.(string); ok {
+			return sessionID
+		}
+	}
+	return ""
+}
+
+// SessionIDWithContext set session id to context
+func SessionIDWithContext(ctx context.Context, sessionID string) context.Context {
+	return context.WithValue(ctx, sessionKey, sessionID)
+}