Initial commit

library/queue/databus/metadata/BUILD

@@ -0,0 +1,29 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["metadata.go"],
+    importpath = "go-common/library/queue/databus/metadata",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = ["//library/net/metadata: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"],
+)

library/queue/databus/metadata/metadata.go

@@ -0,0 +1,28 @@
+package metadata
+
+import (
+	"context"
+
+	"go-common/library/net/metadata"
+)
+
+// FromContext get metadata from context.
+func FromContext(c context.Context) map[string]string {
+	return map[string]string{
+		metadata.Color:    metadata.String(c, metadata.Color),
+		metadata.Caller:   metadata.String(c, metadata.Caller),
+		metadata.Mirror:   metadata.String(c, metadata.Mirror),
+		metadata.RemoteIP: metadata.String(c, metadata.RemoteIP),
+	}
+}
+
+// NewContext new metadata context.
+func NewContext(c context.Context, meta map[string]string) context.Context {
+	md := metadata.MD{
+		metadata.Color:    meta[metadata.Color],
+		metadata.Caller:   meta[metadata.Caller],
+		metadata.Mirror:   meta[metadata.Mirror],
+		metadata.RemoteIP: meta[metadata.RemoteIP],
+	}
+	return metadata.NewContext(c, md)
+}