Skip to content

Commit 02be14f

Browse files
mikereichemikereiche
committed
Add transcoder support on Cache get, putIfAbsent.
Closes #1966.
1 parent 10f71c9 commit 02be14f

File tree

6 files changed

+312
-33
lines changed

6 files changed

+312
-33
lines changed

‎src/main/java/org/springframework/data/couchbase/cache/CouchbaseCache.java

Lines changed: 66 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
*/
1616
package org.springframework.data.couchbase.cache;
1717

18+
1819
import java.lang.reflect.Method;
1920
import java.util.Arrays;
2021
import java.util.Collection;
@@ -23,14 +24,21 @@
2324
import java.util.concurrent.Callable;
2425

2526
import org.springframework.cache.support.AbstractValueAdaptingCache;
26-
import org.springframework.cache.support.SimpleValueWrapper;
2727
import org.springframework.core.convert.ConversionFailedException;
2828
import org.springframework.core.convert.ConversionService;
2929
import org.springframework.core.convert.TypeDescriptor;
30+
import org.springframework.lang.NonNull;
3031
import org.springframework.util.Assert;
3132
import org.springframework.util.ObjectUtils;
3233
import org.springframework.util.ReflectionUtils;
3334

35+
/**
36+
* Couchbase-backed Cache Methods that take a Class return non-wrapped objects - cache-miss cannot be distinguished from
37+
* cached null - this is what AbstractValueAdaptingCache does Methods that do not take a Class return wrapped objects -
38+
* the wrapper is null for cache-miss - the exception is T get(final Object key, final Callable<T> valueLoader), which
39+
* does not return a wrapper because if there is a cache-miss, it gets the value from valueLoader (and caches it). There
40+
* are anomalies with get(key, ValueLoader) - which returns non-wrapped object.
41+
*/
3442
public class CouchbaseCache extends AbstractValueAdaptingCache {
3543

3644
private final String name;
@@ -60,21 +68,30 @@ private static <T> T valueFromLoader(Object key, Callable<T> valueLoader) {
6068
}
6169
}
6270

71+
@NonNull
6372
@Override
6473
public String getName() {
6574
return name;
6675
}
6776

77+
@NonNull
6878
@Override
6979
public CouchbaseCacheWriter getNativeCache() {
7080
return cacheWriter;
7181
}
7282

73-
@Override
74-
protected Object lookup(final Object key) {
75-
return cacheWriter.get(cacheConfig.getCollectionName(), createCacheKey(key), cacheConfig.getValueTranscoder());
83+
/**
84+
* same as inherited, but passes clazz for transcoder
85+
*/
86+
protected Object lookup(final Object key, Class<?> clazz) {
87+
return cacheWriter.get(cacheConfig.getCollectionName(), createCacheKey(key), cacheConfig.getValueTranscoder(),
88+
clazz);
7689
}
7790

91+
@Override
92+
protected Object lookup(@NonNull final Object key) {
93+
return lookup(key, Object.class);
94+
}
7895
/**
7996
* Returns the configuration for this {@link CouchbaseCache}.
8097
*/
@@ -84,7 +101,7 @@ public CouchbaseCacheConfiguration getCacheConfiguration() {
84101

85102
@Override
86103
@SuppressWarnings("unchecked")
87-
public synchronized <T> T get(final Object key, final Callable<T> valueLoader) {
104+
public synchronized <T> T get(@NonNull final Object key, @NonNull final Callable<T> valueLoader) {
88105
ValueWrapper result = get(key);
89106

90107
if (result != null) {
@@ -97,42 +114,65 @@ public synchronized <T> T get(final Object key, final Callable<T> valueLoader) {
97114
}
98115

99116
@Override
100-
public void put(final Object key, final Object value) {
101-
if (!isAllowNullValues() && value == null) {
117+
@SuppressWarnings("unchecked")
118+
public <T> T get(@NonNull final Object key, Class<T> type) {
119+
Object value = this.fromStoreValue(this.lookup(key, type));
120+
if (value != null && type != null && !type.isInstance(value)) {
121+
throw new IllegalStateException("Cached value is not of required type [" + type.getName() + "]: " + value);
122+
} else {
123+
return (T) value;
124+
}
125+
}
102126

103-
throw new IllegalArgumentException(String.format(
104-
"Cache '%s' does not allow 'null' values. Avoid storing null via '@Cacheable(unless=\"#result == null\")' or "
105-
+ "configure CouchbaseCache to allow 'null' via CouchbaseCacheConfiguration.",
106-
name));
127+
public synchronized <T> T get(@NonNull final Object key, final Callable<T> valueLoader, Class<T> type) {
128+
T value = get(key, type);
129+
if (value == null) { // cannot distinguish between cache miss and cached null
130+
value = valueFromLoader(key, valueLoader);
131+
put(key, value);
107132
}
133+
return value;
134+
}
108135

136+
@Override
137+
public void put(@NonNull final Object key, final Object value) {
109138
cacheWriter.put(cacheConfig.getCollectionName(), createCacheKey(key), toStoreValue(value), cacheConfig.getExpiry(),
110139
cacheConfig.getValueTranscoder());
111140
}
112141

113142
@Override
114-
public ValueWrapper putIfAbsent(final Object key, final Object value) {
115-
if (!isAllowNullValues() && value == null) {
116-
return get(key);
117-
}
143+
public ValueWrapper putIfAbsent(@NonNull final Object key, final Object value) {
118144

119145
Object result = cacheWriter.putIfAbsent(cacheConfig.getCollectionName(), createCacheKey(key), toStoreValue(value),
120146
cacheConfig.getExpiry(), cacheConfig.getValueTranscoder());
121147

122-
if (result == null) {
123-
return null;
124-
}
148+
return toValueWrapper(result);
149+
}
125150

126-
return new SimpleValueWrapper(result);
151+
/**
152+
* Not sure why this isn't in AbstractValueAdaptingCache
153+
*
154+
* @param key
155+
* @param value
156+
* @param clazz
157+
* @return
158+
* @param <T>
159+
*/
160+
@SuppressWarnings("unchecked")
161+
public <T> T putIfAbsent(@NonNull final Object key, final Object value, final Class<T> clazz) {
162+
163+
Object result = cacheWriter.putIfAbsent(cacheConfig.getCollectionName(), createCacheKey(key),
164+
toStoreValue(value), cacheConfig.getExpiry(), cacheConfig.getValueTranscoder(), clazz);
165+
166+
return (T) result;
127167
}
128168

129169
@Override
130-
public void evict(final Object key) {
170+
public void evict(@NonNull final Object key) {
131171
cacheWriter.remove(cacheConfig.getCollectionName(), createCacheKey(key));
132172
}
133173

134174
@Override
135-
public boolean evictIfPresent(final Object key) {
175+
public boolean evictIfPresent(@NonNull final Object key) {
136176
return cacheWriter.remove(cacheConfig.getCollectionName(), createCacheKey(key));
137177
}
138178

@@ -152,7 +192,7 @@ public void clear() {
152192
* @param key will never be {@literal null}.
153193
* @return never {@literal null}.
154194
*/
155-
protected String createCacheKey(final Object key) {
195+
protected String createCacheKey(@NonNull final Object key) {
156196
String convertedKey = convertKey(key);
157197
if (!cacheConfig.usePrefix()) {
158198
return convertedKey;
@@ -167,7 +207,10 @@ protected String createCacheKey(final Object key) {
167207
* @return never {@literal null}.
168208
* @throws IllegalStateException if {@code key} cannot be converted to {@link String}.
169209
*/
170-
protected String convertKey(final Object key) {
210+
protected String convertKey(@NonNull final Object key) {
211+
if (key == null) {
212+
throw new IllegalArgumentException(String.format("Cache '%s' does not allow 'null' key.", name));
213+
}
171214
if (key instanceof String) {
172215
return (String) key;
173216
}

‎src/main/java/org/springframework/data/couchbase/cache/CouchbaseCacheWriter.java

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,20 @@ public interface CouchbaseCacheWriter {
4848
Object putIfAbsent(String collectionName, String key, Object value, @Nullable Duration expiry,
4949
@Nullable Transcoder transcoder);
5050

51+
/**
52+
* Write the given value to Couchbase if the key does not already exist.
53+
*
54+
* @param collectionName The cache name must not be {@literal null}.
55+
* @param key The key for the cache entry. Must not be {@literal null}.
56+
* @param value The value stored for the key. Must not be {@literal null}.
57+
* @param expiry Optional expiration time. Can be {@literal null}.
58+
* @param transcoder Optional transcoder to use. Can be {@literal null}.
59+
* @param clazz Optional class for contentAs(clazz)
60+
*/
61+
@Nullable
62+
Object putIfAbsent(String collectionName, String key, Object value, @Nullable Duration expiry,
63+
@Nullable Transcoder transcoder, @Nullable Class<?> clazz);
64+
5165
/**
5266
* Get the binary value representation from Couchbase stored for the given key.
5367
*
@@ -59,6 +73,18 @@ Object putIfAbsent(String collectionName, String key, Object value, @Nullable Du
5973
@Nullable
6074
Object get(String collectionName, String key, @Nullable Transcoder transcoder);
6175

76+
/**
77+
* Get the binary value representation from Couchbase stored for the given key.
78+
*
79+
* @param collectionName must not be {@literal null}.
80+
* @param key must not be {@literal null}.
81+
* @param transcoder Optional transcoder to use. Can be {@literal null}.
82+
* @param clazz Optional class for contentAs(clazz)
83+
* @return {@literal null} if key does not exist.
84+
*/
85+
@Nullable
86+
Object get(String collectionName, String key, @Nullable Transcoder transcoder, @Nullable Class<?> clazz);
87+
6288
/**
6389
* Remove the given key from Couchbase.
6490
*

‎src/main/java/org/springframework/data/couchbase/cache/DefaultCouchbaseCacheWriter.java

Lines changed: 30 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -18,12 +18,14 @@
1818

1919
import static com.couchbase.client.core.io.CollectionIdentifier.DEFAULT_COLLECTION;
2020
import static com.couchbase.client.core.io.CollectionIdentifier.DEFAULT_SCOPE;
21-
import static com.couchbase.client.java.kv.GetOptions.*;
22-
import static com.couchbase.client.java.kv.InsertOptions.*;
23-
import static com.couchbase.client.java.kv.UpsertOptions.*;
24-
import static com.couchbase.client.java.query.QueryOptions.*;
21+
import static com.couchbase.client.java.kv.GetOptions.getOptions;
22+
import static com.couchbase.client.java.kv.InsertOptions.insertOptions;
23+
import static com.couchbase.client.java.kv.UpsertOptions.upsertOptions;
24+
import static com.couchbase.client.java.query.QueryOptions.queryOptions;
2525
import static com.couchbase.client.java.query.QueryScanConsistency.REQUEST_PLUS;
2626

27+
import io.micrometer.common.lang.Nullable;
28+
2729
import java.time.Duration;
2830

2931
import org.springframework.data.couchbase.CouchbaseClientFactory;
@@ -65,6 +67,22 @@ public void put(final String collectionName, final String key, final Object valu
6567
@Override
6668
public Object putIfAbsent(final String collectionName, final String key, final Object value, final Duration expiry,
6769
final Transcoder transcoder) {
70+
return putIfAbsent(collectionName, key, value, expiry, transcoder, Object.class);
71+
}
72+
73+
/**
74+
* same as above, plus clazz
75+
*
76+
* @param collectionName
77+
* @param key
78+
* @param value
79+
* @param expiry
80+
* @param transcoder
81+
* @param clazz
82+
*/
83+
@Override
84+
public Object putIfAbsent(final String collectionName, final String key, final Object value, final Duration expiry,
85+
final Transcoder transcoder, @Nullable final Class<?> clazz) {
6886
InsertOptions options = insertOptions();
6987

7088
if (expiry != null) {
@@ -79,15 +97,20 @@ public Object putIfAbsent(final String collectionName, final String key, final O
7997
return null;
8098
} catch (final DocumentExistsException ex) {
8199
// If the document exists, return the current one per contract
82-
return get(collectionName, key, transcoder);
100+
return get(collectionName, key, transcoder, clazz);
83101
}
84102
}
85103

86104
@Override
87105
public Object get(final String collectionName, final String key, final Transcoder transcoder) {
88-
// TODO .. the decoding side transcoding needs to be figured out?
106+
return get(collectionName, key, transcoder, Object.class);
107+
}
108+
109+
@Override
110+
public Object get(final String collectionName, final String key, final Transcoder transcoder,
111+
final Class<?> clazz) {
89112
try {
90-
return getCollection(collectionName).get(key, getOptions().transcoder(transcoder)).contentAs(Object.class);
113+
return getCollection(collectionName).get(key, getOptions().transcoder(transcoder)).contentAs(clazz);
91114
} catch (DocumentNotFoundException ex) {
92115
return null;
93116
}

‎src/test/java/org/springframework/data/couchbase/cache/CacheUser.java

Lines changed: 24 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,9 +26,11 @@
2626
*/
2727
class CacheUser implements Serializable {
2828
// private static final long serialVersionUID = 8817717605659870262L;
29-
String firstname;
30-
String lastname;
31-
String id;
29+
String firstname; // must have getter/setter for Serialize/Deserialize
30+
String lastname; // must have getter/setter for Serialize/Deserialize
31+
String id; // must have getter/setter for Serialize/Deserialize
32+
33+
public CacheUser() {};
3234

3335
public CacheUser(String id, String firstname, String lastname) {
3436
this.id = id;
@@ -40,6 +42,25 @@ public String getId() {
4042
return id;
4143
}
4244

45+
public String getFirstname() {
46+
return firstname;
47+
}
48+
49+
public String getLastname() {
50+
return lastname;
51+
}
52+
53+
public void setId(String id) {
54+
this.id = id;
55+
}
56+
57+
public void setFirstname(String firstname) {
58+
this.firstname = firstname;
59+
}
60+
61+
public void setLastname(String lastname) {
62+
this.lastname = lastname;
63+
}
4364
// define equals for assertEquals()
4465
public boolean equals(Object o) {
4566
if (o == null) {

‎src/test/java/org/springframework/data/couchbase/cache/CouchbaseCacheCollectionIntegrationTests.java

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,32 @@ void cachePutGet() {
6868
assertEquals(user2, cache.get(user2.getId()).get()); // get user2
6969
}
7070

71+
@Test
72+
void cacheGetValueLoaderWithClass() {
73+
CacheUser user1 = new CacheUser(UUID.randomUUID().toString(), "first1", "last1");
74+
assertNull(cache.get(user1.getId(), CacheUser.class)); // was not put -> cacheMiss
75+
assertEquals(user1, cache.get(user1.getId(), () -> user1)); // put and get user1
76+
assertEquals(user1, cache.get(user1.getId(), () -> user1, CacheUser.class)); // already put, get user1
77+
78+
CacheUser user2 = new CacheUser(UUID.randomUUID().toString(), "first2", "last2");
79+
assertNull(cache.get(user2.getId(), CacheUser.class)); // was not put -> cacheMiss
80+
assertEquals(user2, cache.get(user2.getId(), () -> user2, CacheUser.class)); // put and get user2
81+
assertEquals(user2, cache.get(user2.getId(), () -> user2, CacheUser.class)); // already put, get user2
82+
}
83+
84+
@Test
85+
void cacheGetValueLoaderNoClass() {
86+
CacheUser user1 = new CacheUser(UUID.randomUUID().toString(), "first1", "last1");
87+
assertNull(cache.get(user1.getId())); // was not put -> cacheMiss
88+
assertEquals(user1, cache.get(user1.getId(), () -> user1)); // put and get user1
89+
assertEquals(user1, cache.get(user1.getId(), () -> user1)); // already put, get user1
90+
91+
CacheUser user2 = new CacheUser(UUID.randomUUID().toString(), "first2", "last2");
92+
assertNull(cache.get(user2.getId())); // was not put -> cacheMiss
93+
assertEquals(user2, cache.get(user2.getId(), () -> user2)); // put and get user2
94+
assertEquals(user2, cache.get(user2.getId(), () -> user2)); // already put, get user2
95+
}
96+
7197
@Test
7298
void cacheEvict() {
7399
CacheUser user1 = new CacheUser(UUID.randomUUID().toString(), "first1", "last1");

0 commit comments

Comments
 (0)