Add new function for count pending (#43)

pom.xml

@@ -11,7 +11,7 @@
     </parent>
 
     <artifactId>fraudo</artifactId>
-    <version>1.0.4</version>
+    <version>1.0.5</version>
 
     <name>Fraudo</name>
     <description>Language for describing antifraud patterns</description>

src/main/antlr4/dev.vality.fraudo/FraudoPayment.g4

@@ -50,6 +50,7 @@ unaryExpression
 
 integerExpression
     : count
+    | count_pending
     | count_success
     | count_error
     | count_chargeback
@@ -72,6 +73,10 @@ count_success
  : 'countSuccess' LPAREN STRING time_window (group_by)? RPAREN
  ;
 
+count_pending
+ : 'countPending' LPAREN STRING time_window (group_by)? RPAREN
+ ;
+
 count_error
  : 'countError' LPAREN STRING time_window (DELIMITER STRING)? (group_by)? RPAREN
  ;

src/main/java/dev/vality/fraudo/payment/aggregator/CountPaymentAggregator.java

@@ -13,6 +13,8 @@ public interface CountPaymentAggregator<T, U> extends CountAggregator<T, U> {
 
     Integer countError(U checkedField, T model, TimeWindow timeWindow, List<U> fields);
 
+    Integer countPending(U checkedField, T model, TimeWindow timeWindow, List<U> fields);
+
     Integer countChargeback(U checkedField, T model, TimeWindow timeWindow, List<U> fields);
 
     Integer countRefund(U checkedField, T model, TimeWindow timeWindow, List<U> fields);

src/main/java/dev/vality/fraudo/payment/visitor/CountVisitor.java

@@ -8,6 +8,8 @@ public interface CountVisitor<T> {
 
     Integer visitCountSuccess(Count_successContext ctx, T model);
 
+    Integer visitCountPending(Count_pendingContext ctx, T model);
+
     Integer visitCountError(Count_errorContext ctx, T model);
 
     Integer visitCountChargeback(Count_chargebackContext ctx, T model);

src/main/java/dev/vality/fraudo/payment/visitor/impl/CountVisitorImpl.java

@@ -39,6 +39,17 @@ public class CountVisitorImpl<T, U> implements CountVisitor<T> {
         );
     }
 
+    @Override
+    public Integer visitCountPending(FraudoPaymentParser.Count_pendingContext ctx, T model) {
+        String countTarget = TextUtil.safeGetText(ctx.STRING());
+        return countPaymentAggregator.countPending(
+                fieldResolver.resolveName(countTarget),
+                model,
+                timeWindowResolver.resolve(ctx.time_window()),
+                groupResolver.resolve(ctx.group_by())
+        );
+    }
+
     @Override
     public Integer visitCountError(FraudoPaymentParser.Count_errorContext ctx, T model) {
         String countTarget = TextUtil.safeGetText(ctx.STRING(0));

src/main/java/dev/vality/fraudo/payment/visitor/impl/FirstFindVisitorImpl.java

@@ -296,6 +296,15 @@ public class FirstFindVisitorImpl<T extends BaseModel, U> extends FraudoPaymentB
         );
     }
 
+    @Override
+    public Integer visitCount_pending(Count_pendingContext ctx) {
+        String key = CountKeyGenerator.generatePendingKey(ctx, fieldResolver::resolveName);
+        return (Integer) localFuncCache.get().computeIfAbsent(
+                key,
+                s -> countVisitor.visitCountPending(ctx, threadLocalModel.get())
+        );
+    }
+
     @Override
     public Integer visitCount_error(Count_errorContext ctx) {
         String key = CountKeyGenerator.generateErrorKey(ctx, fieldResolver::resolveName);

src/main/java/dev/vality/fraudo/utils/key/generator/CountKeyGenerator.java

@@ -26,6 +26,14 @@ public class CountKeyGenerator {
                 resolve);
     }
 
+    public static <T> String generatePendingKey(Count_pendingContext ctx, Function<String, T> resolve) {
+        return CommonKeyGenerator.generateKeyGroupedFunction(ctx.STRING(),
+                ctx.children.get(0),
+                ctx.time_window(),
+                ctx.group_by(),
+                resolve);
+    }
+
     public static <T> String generateErrorKey(Count_errorContext ctx, Function<String, T> resolve) {
         if (ctx.STRING().size() == 2) {
             return CommonKeyGenerator.generateKeyGroupedTwoFieldFunction(ctx.STRING(0),

src/test/java/dev/vality/fraudo/CountTest.java

@@ -53,6 +53,16 @@ public class CountTest extends AbstractPaymentTest {
         assertEquals("0", ResultUtils.findFirstNotNotifyStatus(result).get().getRuleChecked());
     }
 
+    @Test
+    void countPendingTest() throws Exception {
+        InputStream resourceAsStream = CountTest.class.getResourceAsStream("/rules/countPending.frd");
+        when(countPaymentAggregator.countPending(anyObject(), any(), any(), any())).thenReturn(6);
+        ParseContext parseContext = getParseContext(resourceAsStream);
+        ResultModel result = invokeParse(parseContext);
+        assertEquals(ResultStatus.DECLINE, ResultUtils.findFirstNotNotifyStatus(result).get().getResultStatus());
+        assertEquals("0", ResultUtils.findFirstNotNotifyStatus(result).get().getRuleChecked());
+    }
+
     @Test
     void countCargeRefundTest() throws Exception {
         InputStream resourceAsStream = CountTest.class.getResourceAsStream("/rules/count_chargeback_refund.frd");

src/test/resources/rules/countPending.frd

@@ -0,0 +1,2 @@
+rule: countPending("fingerprint", 1444) > 5
+-> decline;