From 204839957794ad3c3e62c76700ed8ef716f2f1a5 Mon Sep 17 00:00:00 2001
From: Jonathan Shook <jshook@gmail.com>
Date: Thu, 24 Feb 2022 16:13:30 -0600
Subject: [PATCH] add new parsed op enhancer function

---
 .../nosqlbench/engine/api/templating/ParsedOp.java | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/adapters-api/src/main/java/io/nosqlbench/engine/api/templating/ParsedOp.java b/adapters-api/src/main/java/io/nosqlbench/engine/api/templating/ParsedOp.java
index c830278e0..af8c46671 100644
--- a/adapters-api/src/main/java/io/nosqlbench/engine/api/templating/ParsedOp.java
+++ b/adapters-api/src/main/java/io/nosqlbench/engine/api/templating/ParsedOp.java
@@ -460,6 +460,20 @@ public class ParsedOp implements LongFunction<Map<String, ?>>, StaticFieldReader
         return getOptionalEnumFromField(enumClass,fieldName).orElse(defaultEnum);
     }
 
+    public <FA,FE> LongFunction<FA> enhance(
+        LongFunction<FA> func,
+        String field,
+        Class<FE> type,
+        FE defaultFe,
+        BiFunction<FA,FE,FA> combiner
+    ) {
+        LongFunction<FE> fieldEnhancerFunc = getAsFunctionOr(field, defaultFe);
+        LongFunction<FA> faLongFunction = func;
+        LongFunction<FA> lfa = l -> combiner.apply(faLongFunction.apply(l),fieldEnhancerFunc.apply(l));
+        return lfa;
+
+    }
+
     public <FA,FE> Optional<LongFunction<FA>> enhance(
         Optional<LongFunction<FA>> func,
         String field,