set/get operation in context: used for route operation names

logctx/context.go

@@ -7,8 +7,8 @@ import (
 type ContextKey string
 
 const (
-	opID      ContextKey = "request_id"
-	opMessage ContextKey = "request_message"
+	operationKey ContextKey = "operation"
+	opID         ContextKey = "request_id"
 )
 
 // SetID sets the request ID logged to the context.
@@ -25,3 +25,18 @@ func GetID(ctx context.Context) string {
 
 	return ""
 }
+
+// SetOperation sets the operation logged to the context.
+func SetOperation(ctx context.Context, operation string) context.Context {
+	return context.WithValue(ctx, operationKey, operation)
+}
+
+// GetOperation returns the operation logged to the context, otherwise an empty string.
+func GetOperation(ctx context.Context) string {
+	s, ok := ctx.Value(operationKey).(string)
+	if ok {
+		return s
+	}
+
+	return ""
+}

logctx/context_test.go

@@ -17,3 +17,12 @@ func TestID(t *testing.T) {
 	ctx = logctx.SetID(ctx, "123")
 	assert.Equal(t, "123", logctx.GetID(ctx))
 }
+
+func TestOperation(t *testing.T) {
+	ctx := context.Background()
+
+	assert.Empty(t, logctx.GetOperation(ctx))
+
+	ctx = logctx.SetOperation(ctx, "getTest")
+	assert.Equal(t, "getTest", logctx.GetOperation(ctx))
+}