001/** 002 * Copyright (c) 2011, The University of Southampton and the individual contributors. 003 * All rights reserved. 004 * 005 * Redistribution and use in source and binary forms, with or without modification, 006 * are permitted provided that the following conditions are met: 007 * 008 * * Redistributions of source code must retain the above copyright notice, 009 * this list of conditions and the following disclaimer. 010 * 011 * * Redistributions in binary form must reproduce the above copyright notice, 012 * this list of conditions and the following disclaimer in the documentation 013 * and/or other materials provided with the distribution. 014 * 015 * * Neither the name of the University of Southampton nor the names of its 016 * contributors may be used to endorse or promote products derived from this 017 * software without specific prior written permission. 018 * 019 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND 020 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 021 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 022 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR 023 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 024 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 025 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON 026 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 027 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 028 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 029 */ 030package org.openimaj.hadoop.tools.twitter; 031 032import java.util.ArrayList; 033import java.util.Collection; 034 035import org.hamcrest.Matcher; 036import static org.hamcrest.Matchers.*; 037 038import org.openimaj.util.pair.IndependentPair; 039 040import com.jayway.jsonassert.JsonAsserter; 041import static com.jayway.jsonassert.JsonAssert.*; 042 043/** 044 * Given a collection of JSONpath, treat each path as a filter and return true if 045 * a given json input matches every filter 046 * 047 * @author Sina Samangooei (ss@ecs.soton.ac.uk) 048 * 049 */ 050public class JsonPathFilterSet { 051 private ArrayList<IndependentPair<String,Matcher<?>>> filters; 052 053 054 /** 055 * @param paths the json paths against which to filter 056 */ 057 public JsonPathFilterSet(Collection<String> paths) { 058 this(); 059 if(paths == null) return; 060 for (String string : paths) { 061 addString(string); 062 } 063 } 064 065 private void addString(String string) { 066 System.out.println("Adding String: " + string); 067 String[] split = string.split(":=="); 068 IndependentPair<String, Matcher<?>> r = null; 069 if(split.length == 1) 070 { 071 r = new IndependentPair<String,Matcher<?>>(split[0], null); 072 } 073 else{ 074 r = new IndependentPair<String,Matcher<?>>(split[0],equalTo(split[1])); 075 } 076 this.filters.add(r); 077 } 078 079 /** 080 * @param paths the json paths against which to filter 081 */ 082 public JsonPathFilterSet(String ... paths) { 083 this(); 084 for (String string : paths) { 085 addString(string); 086 } 087 } 088 089 private JsonPathFilterSet() { 090 this.filters = new ArrayList<IndependentPair<String,Matcher<?>>>(); 091 } 092 093 /** 094 * @param json the input to be checked 095 * @return true if the inputed json matches all filters 096 */ 097 public boolean filter(String json){ 098 JsonAsserter jass = with(json); 099 for (IndependentPair<String, Matcher<?>> filter : this.filters) { 100 String path = filter.firstObject(); 101 Matcher<?> matcher = filter.secondObject(); 102 try{ 103 if(matcher==null) 104 jass = jass.assertNotNull(path); 105 else 106 jass = jass.assertThat(path, matcher); 107 }catch(AssertionError e){ 108 return false; 109 } 110 } 111 return true; 112 } 113}