|
|
|
|
@@ -0,0 +1,246 @@
|
|
|
|
|
/*
|
|
|
|
|
* Copyright 2002-2013 the original author or authors.
|
|
|
|
|
*
|
|
|
|
|
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
|
|
|
|
|
* the License. You may obtain a copy of the License at
|
|
|
|
|
*
|
|
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
|
*
|
|
|
|
|
* Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
|
|
|
|
|
* an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
|
|
|
|
* specific language governing permissions and limitations under the License.
|
|
|
|
|
*/
|
|
|
|
|
package org.springframework.data.gemfire.support;
|
|
|
|
|
|
|
|
|
|
import java.util.ArrayList;
|
|
|
|
|
import java.util.Arrays;
|
|
|
|
|
import java.util.Collection;
|
|
|
|
|
import java.util.HashMap;
|
|
|
|
|
import java.util.List;
|
|
|
|
|
import java.util.Map;
|
|
|
|
|
import java.util.Map.Entry;
|
|
|
|
|
|
|
|
|
|
import org.apache.commons.logging.Log;
|
|
|
|
|
import org.apache.commons.logging.LogFactory;
|
|
|
|
|
import org.aspectj.lang.ProceedingJoinPoint;
|
|
|
|
|
import org.aspectj.lang.annotation.Around;
|
|
|
|
|
import org.aspectj.lang.annotation.Aspect;
|
|
|
|
|
import org.codehaus.jackson.map.ObjectMapper;
|
|
|
|
|
import org.springframework.util.CollectionUtils;
|
|
|
|
|
|
|
|
|
|
import com.gemstone.gemfire.cache.Region;
|
|
|
|
|
import com.gemstone.gemfire.pdx.JSONFormatter;
|
|
|
|
|
import com.gemstone.gemfire.pdx.PdxInstance;
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* @author David Turanski
|
|
|
|
|
*
|
|
|
|
|
*/
|
|
|
|
|
@Aspect
|
|
|
|
|
public class JSONRegionAdvice {
|
|
|
|
|
private static Log log = LogFactory.getLog(JSONRegionAdvice.class);
|
|
|
|
|
private List<String> includedRegions;
|
|
|
|
|
private boolean convertReturnedCollections = true;
|
|
|
|
|
private boolean prettyPrint = false;
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Sets names of regions to be included for JSON conversion. By default, all regions will be included
|
|
|
|
|
* @param regions a List of region names to include
|
|
|
|
|
*/
|
|
|
|
|
public void setIncludedRegionNames(List<String> regionNames) {
|
|
|
|
|
this.includedRegions = regionNames;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Sets regions to be included for JSON conversion. By default, all regions will be included
|
|
|
|
|
* @param regions a List of region names to include
|
|
|
|
|
*/
|
|
|
|
|
public void setIncludedRegions(List<Region<?,?>> regions) {
|
|
|
|
|
this.includedRegions = new ArrayList<String>();
|
|
|
|
|
for (Region<?,?> region: regions) {
|
|
|
|
|
includedRegions.add(region.getName());
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
/**
|
|
|
|
|
* Flag to convert collections returned from cache from @{link PdxInstance} to JSON String. If the returned
|
|
|
|
|
* collections are very large, overhead will be incurred to covert all the values from from
|
|
|
|
|
* Region.getAll() and Region.values()
|
|
|
|
|
* @param convertReturnedCollections true by default
|
|
|
|
|
*/
|
|
|
|
|
public void setConvertReturnedCollections(boolean convertReturnedCollections) {
|
|
|
|
|
this.convertReturnedCollections = convertReturnedCollections;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Flag to print JSON Strings with proper indentation, etc.
|
|
|
|
|
* @param prettyPrint false be default
|
|
|
|
|
*/
|
|
|
|
|
public void setPrettyPrint(boolean prettyPrint) {
|
|
|
|
|
this.prettyPrint = prettyPrint;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Around("execution(* com.gemstone.gemfire.cache.Region.put(..)) || "
|
|
|
|
|
+ "execution(* com.gemstone.gemfire.cache.Region.create(..)) ||"
|
|
|
|
|
+ "execution(* com.gemstone.gemfire.cache.Region.putIfAbsent(..)) ||"
|
|
|
|
|
+ "execution(* com.gemstone.gemfire.cache.Region.replace(..))")
|
|
|
|
|
public Object put(ProceedingJoinPoint pjp) {
|
|
|
|
|
System.out.println("intercepted " + pjp.getSignature().getName());
|
|
|
|
|
|
|
|
|
|
boolean JSONRegion = isIncludedSONRegion(pjp.getTarget());
|
|
|
|
|
Object retVal = null;
|
|
|
|
|
try {
|
|
|
|
|
if (JSONRegion) {
|
|
|
|
|
Object[] newArgs = Arrays.copyOf(pjp.getArgs(), pjp.getArgs().length);
|
|
|
|
|
Object val = newArgs[1];
|
|
|
|
|
newArgs[1] = convertArgumentToPdxInstance(val);
|
|
|
|
|
retVal = pjp.proceed(newArgs);
|
|
|
|
|
} else {
|
|
|
|
|
retVal = pjp.proceed();
|
|
|
|
|
}
|
|
|
|
|
} catch (Throwable t) {
|
|
|
|
|
handleThrowable(t);
|
|
|
|
|
}
|
|
|
|
|
return retVal;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Around("execution(* com.gemstone.gemfire.cache.Region.putAll(..))")
|
|
|
|
|
public Object putAll(ProceedingJoinPoint pjp) {
|
|
|
|
|
boolean JSONRegion = isIncludedSONRegion(pjp.getTarget());
|
|
|
|
|
|
|
|
|
|
Object retVal = null;
|
|
|
|
|
try {
|
|
|
|
|
if (JSONRegion) {
|
|
|
|
|
Object[] newArgs = Arrays.copyOf(pjp.getArgs(), pjp.getArgs().length);
|
|
|
|
|
Map<?, ?> val = (Map<?, ?>) newArgs[0];
|
|
|
|
|
Map<Object, Object> newArg = new HashMap<Object, Object>();
|
|
|
|
|
for (Entry<?, ?> entry : val.entrySet()) {
|
|
|
|
|
newArg.put(entry.getKey(), convertArgumentToPdxInstance(entry.getValue()));
|
|
|
|
|
}
|
|
|
|
|
newArgs[0] = newArg;
|
|
|
|
|
retVal = pjp.proceed(newArgs);
|
|
|
|
|
} else {
|
|
|
|
|
retVal = pjp.proceed();
|
|
|
|
|
}
|
|
|
|
|
} catch (Throwable t) {
|
|
|
|
|
handleThrowable(t);
|
|
|
|
|
}
|
|
|
|
|
return retVal;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Around("execution(* com.gemstone.gemfire.cache.Region.get(..)) "
|
|
|
|
|
+ "|| execution(* com.gemstone.gemfire.cache.Region.selectValue(..))")
|
|
|
|
|
public Object get(ProceedingJoinPoint pjp) {
|
|
|
|
|
Object retVal = null;
|
|
|
|
|
try {
|
|
|
|
|
if (isIncludedSONRegion(pjp.getTarget())) {
|
|
|
|
|
retVal = pjp.proceed();
|
|
|
|
|
log.debug("converting " + retVal + " to JSON string");
|
|
|
|
|
return convertPdxInstanceToJSONString(retVal);
|
|
|
|
|
} else {
|
|
|
|
|
return pjp.proceed();
|
|
|
|
|
}
|
|
|
|
|
} catch (Throwable t) {
|
|
|
|
|
handleThrowable(t);
|
|
|
|
|
}
|
|
|
|
|
return retVal;
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@SuppressWarnings("unchecked")
|
|
|
|
|
@Around("execution(* com.gemstone.gemfire.cache.Region.getAll(..))")
|
|
|
|
|
public Map<Object, Object> getAll(ProceedingJoinPoint pjp) {
|
|
|
|
|
Map<Object, Object> result = null;
|
|
|
|
|
try {
|
|
|
|
|
Map<Object, Object> retVal = (Map<Object, Object>) pjp.proceed();
|
|
|
|
|
if (!convertReturnedCollections || CollectionUtils.isEmpty(retVal) || !isIncludedSONRegion(pjp.getTarget())) {
|
|
|
|
|
result = retVal;
|
|
|
|
|
} else {
|
|
|
|
|
result = new HashMap<Object, Object>();
|
|
|
|
|
for (Entry<Object, Object> entry : retVal.entrySet()) {
|
|
|
|
|
result.put(entry.getKey(), convertPdxInstanceToJSONString(entry.getValue()));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
} catch (Throwable t) {
|
|
|
|
|
handleThrowable(t);
|
|
|
|
|
}
|
|
|
|
|
return result;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@SuppressWarnings("unchecked")
|
|
|
|
|
@Around("execution(* com.gemstone.gemfire.cache.Region.values(..))")
|
|
|
|
|
public Collection<Object> values(ProceedingJoinPoint pjp) {
|
|
|
|
|
Collection<Object> result = null;
|
|
|
|
|
try {
|
|
|
|
|
Collection<Object> retVal = (Collection<Object>) pjp.proceed();
|
|
|
|
|
if (!convertReturnedCollections || CollectionUtils.isEmpty(retVal) || !isIncludedSONRegion(pjp.getTarget())) {
|
|
|
|
|
result = retVal;
|
|
|
|
|
} else {
|
|
|
|
|
result = new ArrayList<Object>();
|
|
|
|
|
for (Object obj : retVal) {
|
|
|
|
|
result.add(convertArgumentToPdxInstance(obj));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
} catch (Throwable t) {
|
|
|
|
|
handleThrowable(t);
|
|
|
|
|
}
|
|
|
|
|
return result;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private PdxInstance convertArgumentToPdxInstance(Object value) {
|
|
|
|
|
PdxInstance val = null;
|
|
|
|
|
if (value instanceof PdxInstance) {
|
|
|
|
|
val = (PdxInstance) value;
|
|
|
|
|
} else if (value instanceof String) {
|
|
|
|
|
val = JSONFormatter.fromJSON((String) value);
|
|
|
|
|
} else {
|
|
|
|
|
ObjectMapper mapper = new ObjectMapper();
|
|
|
|
|
try {
|
|
|
|
|
String json = mapper.writeValueAsString(value);
|
|
|
|
|
val = JSONFormatter.fromJSON(json);
|
|
|
|
|
} catch (Throwable t) {
|
|
|
|
|
handleThrowable(t);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return val;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private boolean isIncludedSONRegion(Object target) {
|
|
|
|
|
Region<?, ?> region = (Region<?, ?>) target;
|
|
|
|
|
boolean result = false;
|
|
|
|
|
if (includedRegions == null || includedRegions.contains(region.getName())) {
|
|
|
|
|
if (log.isDebugEnabled()) {
|
|
|
|
|
log.debug(region.getName() + " is included for JSON conversion");
|
|
|
|
|
}
|
|
|
|
|
result = true;
|
|
|
|
|
}
|
|
|
|
|
return result;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private Object convertPdxInstanceToJSONString(Object retVal) {
|
|
|
|
|
Object result = retVal;
|
|
|
|
|
if (retVal != null && retVal instanceof PdxInstance) {
|
|
|
|
|
result = JSONFormatter.toJSON((PdxInstance) retVal);
|
|
|
|
|
}
|
|
|
|
|
if (!prettyPrint) {
|
|
|
|
|
result = flattenString(result);
|
|
|
|
|
}
|
|
|
|
|
return result;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private Object flattenString(Object result) {
|
|
|
|
|
if (result instanceof String) {
|
|
|
|
|
String json = (String) result;
|
|
|
|
|
return json.replaceAll("\\s*", "");
|
|
|
|
|
}
|
|
|
|
|
return result;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private void handleThrowable(Throwable t) {
|
|
|
|
|
if (t instanceof RuntimeException) {
|
|
|
|
|
throw (RuntimeException) t;
|
|
|
|
|
} else {
|
|
|
|
|
throw new RuntimeException(t);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
}
|
David Turanski