add memoizeObject method to RedisService, fixes #34

grails-app/services/grails/plugin/redis/RedisService.groovy

@@ -14,6 +14,7 @@
  */
 package grails.plugin.redis
 
+import com.google.gson.Gson
 import redis.clients.jedis.Jedis
 import redis.clients.jedis.Pipeline
 import redis.clients.jedis.Transaction
@@ -321,10 +322,6 @@ class RedisService {
  domainClass.load(domainId)
  }
 
-// Long persistDomainId(Object domainInstance, String key, Map options) {
- // return persistDomainId(domainInstance?.id as Long, key, options.expire)
- // }
-
  Long persistDomainId(Long domainId, String key, Integer expire) {
  if(domainId) {
  withOptionalPipeline { pipeline ->
@@ -334,10 +331,24 @@ class RedisService {
  }
  }
  }
-// if (domainId) withRedis { Jedis redis -> redis.set(key, domainId.toString()) }
  domainId
  }
 
+ def memoizeObject(Class clazz, String key, Integer expire, Closure closure) {
+ memoizeObject(clazz, key, [expire: expire], closure)
+ }
+
+ def memoizeObject(Class clazz, String key, Map options = [:], Closure closure) {
+ Gson gson = new Gson()
+
+ String memoizedJson = memoize(key, options) { ->
+ def original = closure()
+ gson.toJson(original)
+ }
+
+ gson.fromJson(memoizedJson, clazz)
+ }
+
  // deletes all keys matching a pattern (see redis "keys" documentation for more)
  // OK for low traffic methods, but expensive compared to other redis commands
  // perf test before relying on this rather than storing your own set of keys to 

src/groovy/grails/plugin/redis/ast/MemoizeObjectASTTransformation.groovy

@@ -5,32 +5,24 @@ import java.util.Map;
 import org.codehaus.groovy.ast.ASTNode
 import org.codehaus.groovy.ast.ClassHelper;
 import org.codehaus.groovy.ast.ClassNode
-import org.codehaus.groovy.ast.FieldNode;
 import org.codehaus.groovy.ast.MethodNode
-import org.codehaus.groovy.ast.PropertyNode;
 import org.codehaus.groovy.ast.VariableScope;
-import org.codehaus.groovy.ast.builder.AstBuilder;
 import org.codehaus.groovy.ast.expr.ArgumentListExpression
 import org.codehaus.groovy.ast.expr.ClassExpression;
-import org.codehaus.groovy.ast.expr.ClosureExpression
 import org.codehaus.groovy.ast.expr.ConstantExpression
 import org.codehaus.groovy.ast.expr.ConstructorCallExpression;
 import org.codehaus.groovy.ast.expr.Expression;
 import org.codehaus.groovy.ast.expr.MethodCallExpression
-import org.codehaus.groovy.ast.expr.VariableExpression
 import org.codehaus.groovy.control.CompilePhase
 import org.codehaus.groovy.control.SourceUnit
 import org.codehaus.groovy.ast.stmt.BlockStatement
-import org.codehaus.groovy.ast.stmt.ExpressionStatement
 import org.codehaus.groovy.ast.stmt.ReturnStatement
 import org.codehaus.groovy.ast.stmt.Statement
 import org.codehaus.groovy.classgen.VariableScopeVisitor
 
 import com.google.gson.Gson
 
 import grails.plugin.redis.RedisService
-import static org.springframework.asm.Opcodes.ACC_PRIVATE
-import static org.springframework.asm.Opcodes.ACC_PUBLIC
 
 import org.codehaus.groovy.transform.GroovyASTTransformation
 
@@ -132,7 +124,6 @@ class MemoizeObjectASTTransformation extends AbstractMemoizeASTTransformation {
  if(!validateMemoizeProperties(astNodes, sourceUnit, keyString, expire, clazz)) {
  return
  }
- //***************************************************************************
 
  memoizeProperties.put(KEY, keyString)
  memoizeProperties.put(CLAZZ, clazz)

test/integration/grails/plugin/redis/RedisServiceTests.groovy

@@ -334,6 +334,43 @@ class RedisServiceTests {
  assert NO_EXPIRATION_TTL < redisService.ttl("mykey")
  }
 
+ @Test
+ public void testMemoizeObject_simpleMapOfStrings() {
+ Map<String, String> map = [foo: "bar", baz: "qux"]
+
+ def calledCount = 0
+ def cacheMissClosure = {
+ calledCount += 1
+ map
+ }
+
+ def cacheMissValue = redisService.memoizeObject(Map.class, "mykey", cacheMissClosure)
+
+ assert 1 == calledCount
+ assert "bar" == cacheMissValue.foo
+ assert "qux" == cacheMissValue.baz
+ assert NO_EXPIRATION_TTL == redisService.ttl("mykey")
+
+ def cacheHitValue = redisService.memoizeObject(Map.class, "mykey", cacheMissClosure)
+
+ assert 1 == calledCount
+ assert "bar" == cacheHitValue.foo
+ assert "qux" == cacheHitValue.baz
+ }
+
+
+ @Test
+ public void testMemoizeObject_withTTL() {
+ Map<String, String> map = [foo: "bar", baz: "qux"]
+ assert 0 > redisService.ttl("mykey")
+
+ def cacheMissValue = redisService.memoizeObject(Map.class, "mykey", 60) { -> map }
+
+ assert "bar" == cacheMissValue.foo
+ assert "qux" == cacheMissValue.baz
+ assert NO_EXPIRATION_TTL < redisService.ttl("mykey")
+ }
+
 
  @Test
  public void testDeleteKeysWithPattern() {