context.go

@@ -2,7 +2,10 @@ package yaml
 
 import "context"
 
-type ctxMergeKey struct{}
+type (
+	ctxMergeKey  struct{}
+	ctxAnchorKey struct{}
+)
 
 func withMerge(ctx context.Context) context.Context {
 	return context.WithValue(ctx, ctxMergeKey{}, true)
@@ -15,3 +18,20 @@ func isMerge(ctx context.Context) bool {
 	}
 	return v
 }
+
+func withAnchor(ctx context.Context, name string) context.Context {
+	anchorMap := getAnchorMap(ctx)
+	if anchorMap == nil {
+		anchorMap = make(map[string]struct{})
+	}
+	anchorMap[name] = struct{}{}
+	return context.WithValue(ctx, ctxAnchorKey{}, anchorMap)
+}
+
+func getAnchorMap(ctx context.Context) map[string]struct{} {
+	v, ok := ctx.Value(ctxAnchorKey{}).(map[string]struct{})
+	if !ok {
+		return nil
+	}
+	return v
+}