Added two new filters to statistics generator

Author: acz-icm

statistics-generator/statistics-generator-impl/src/main/java/pl/edu/icm/coansys/statisticsgenerator/conf/ComponentsMapping.java

@@ -23,8 +23,10 @@
 import pl.edu.icm.coansys.statisticsgenerator.operationcomponents.CountSummary;
 import pl.edu.icm.coansys.statisticsgenerator.operationcomponents.DateRangesPartitioner;
 import pl.edu.icm.coansys.statisticsgenerator.operationcomponents.EqualsPartitioner;
+import pl.edu.icm.coansys.statisticsgenerator.operationcomponents.FilterContains;
 import pl.edu.icm.coansys.statisticsgenerator.operationcomponents.FilterEq;
 import pl.edu.icm.coansys.statisticsgenerator.operationcomponents.FilterNotEmpty;
+import pl.edu.icm.coansys.statisticsgenerator.operationcomponents.FilterTwoFieldsEqual;
 import pl.edu.icm.coansys.statisticsgenerator.operationcomponents.FirstCharsPartitioner;
 import pl.edu.icm.coansys.statisticsgenerator.operationcomponents.LastCharsPartitioner;
 import pl.edu.icm.coansys.statisticsgenerator.operationcomponents.OperationComponent;
@@ -43,6 +45,8 @@ private ComponentsMapping() {}
     static {
         mapping.put("EQFILTER", FilterEq.class);
         mapping.put("NONEMPTYFILTER", FilterNotEmpty.class);
+        mapping.put("TWOFIELDSEQFILTER", FilterTwoFieldsEqual.class);
+        mapping.put("CONTAINSFILTER", FilterContains.class);
         mapping.put("EQUALS", EqualsPartitioner.class);
         mapping.put("FIRSTCHARS", FirstCharsPartitioner.class);
         mapping.put("LASTCHARS", LastCharsPartitioner.class);

statistics-generator/statistics-generator-impl/src/main/java/pl/edu/icm/coansys/statisticsgenerator/operationcomponents/FilterContains.java

@@ -0,0 +1,50 @@
+/*
+ * This file is part of CoAnSys project.
+ * Copyright (c) 2012-2013 ICM-UW
+ * 
+ * CoAnSys is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+
+ * CoAnSys is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU Affero General Public License for more details.
+ * 
+ * You should have received a copy of the GNU Affero General Public License
+ * along with CoAnSys. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package pl.edu.icm.coansys.statisticsgenerator.operationcomponents;
+
+import pl.edu.icm.coansys.models.StatisticsProtos;
+import pl.edu.icm.coansys.models.StatisticsProtos.KeyValue;
+
+/**
+ *
+ * @author Artur Czeczko <a.czeczko@icm.edu.pl>
+ */
+public class FilterContains implements FilterComponent {
+    private String fieldName;
+    private String value;
+    
+    @Override
+    public boolean filter(StatisticsProtos.InputEntry entry) {
+        for (KeyValue kv : entry.getFieldList()) {
+            if (kv.getKey().equals(fieldName)) {
+                return kv.getValue().contains(value);
+            }
+        }
+        return false;
+    }
+
+    @Override
+    public void setup(String... params) {
+        if (params.length != 2) {
+            throw new IllegalArgumentException("setup requires 2 parameters: field name and value");
+        }
+        fieldName = params[0];
+        value = params[1];
+    }
+}

statistics-generator/statistics-generator-impl/src/main/java/pl/edu/icm/coansys/statisticsgenerator/operationcomponents/FilterTwoFieldsEqual.java

@@ -0,0 +1,60 @@
+/*
+ * This file is part of CoAnSys project.
+ * Copyright (c) 2012-2013 ICM-UW
+ * 
+ * CoAnSys is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+
+ * CoAnSys is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU Affero General Public License for more details.
+ * 
+ * You should have received a copy of the GNU Affero General Public License
+ * along with CoAnSys. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package pl.edu.icm.coansys.statisticsgenerator.operationcomponents;
+
+import java.util.ArrayList;
+import java.util.List;
+import pl.edu.icm.coansys.models.StatisticsProtos;
+import pl.edu.icm.coansys.models.StatisticsProtos.KeyValue;
+
+/**
+ *
+ * @author Artur Czeczko <a.czeczko@icm.edu.pl>
+ */
+public class FilterTwoFieldsEqual implements FilterComponent {
+
+    private List<String> fieldNames = new ArrayList<String>(2);
+
+    @Override
+    public boolean filter(StatisticsProtos.InputEntry entry) {
+        String firstKey = null;
+        String firstValue = null;
+        
+        for (KeyValue kv : entry.getFieldList()) {
+            if (fieldNames.contains(kv.getKey()) && !kv.getKey().equals(firstKey)) {
+                if (firstValue == null) {
+                    firstKey = kv.getKey();
+                    firstValue = kv.getValue();
+                } else {
+                    return firstValue.equals(kv.getValue());
+                }
+            }
+        }
+        return false;
+    }
+
+    @Override
+    public void setup(String... params) {
+        if (params.length != 2) {
+            throw new IllegalArgumentException("setup requires 2 parameters: field name and value");
+        }
+        fieldNames.add(params[0]);
+        fieldNames.add(params[1]);
+    }
+}