1 package com.atlassian.vcache.internal.memcached;
2
3 import com.atlassian.marshalling.jdk.JavaSerializationMarshalling;
4 import com.atlassian.vcache.DirectExternalCache;
5 import com.atlassian.vcache.ExternalCacheSettings;
6 import com.atlassian.vcache.internal.RequestContext;
7 import com.atlassian.vcache.internal.RequestMetrics;
8 import com.atlassian.vcache.internal.core.DefaultRequestContext;
9 import com.atlassian.vcache.internal.core.Sha1ExternalCacheKeyGenerator;
10 import com.atlassian.vcache.internal.core.metrics.DefaultMetricsCollector;
11 import com.atlassian.vcache.internal.core.metrics.MetricsCollector;
12 import com.atlassian.vcache.internal.test.AbstractDirectExternalCacheIT;
13 import net.spy.memcached.AddrUtil;
14 import net.spy.memcached.BinaryConnectionFactory;
15 import net.spy.memcached.MemcachedClient;
16 import org.junit.BeforeClass;
17
18 import java.io.IOException;
19 import java.time.Duration;
20
21 public class MemcachedDirectExternalCacheIT extends AbstractDirectExternalCacheIT {
22 private static MemcachedClient client;
23 private final RequestContext requestContext = new DefaultRequestContext("tenant-id");
24 private final MetricsCollector metricsCollector = new DefaultMetricsCollector(() -> requestContext);
25
26 @BeforeClass
27 public static void init_client() throws IOException {
28 client = new MemcachedClient(
29 new BinaryConnectionFactory(),
30 AddrUtil.getAddresses(System.getProperty("memcached.svr")));
31 }
32
33 @Override
34 protected DirectExternalCache<String> createCache(String name, ExternalCacheSettings settings) {
35 return metricsCollector.wrap(
36 new MemcachedDirectExternalCache<>(
37 () -> client,
38 () -> requestContext,
39 new Sha1ExternalCacheKeyGenerator("it-test-42"),
40 name,
41 JavaSerializationMarshalling.pair(String.class),
42 settings,
43 Duration.ofSeconds(1)));
44 }
45
46 @Override
47 protected RequestMetrics requestMetrics() {
48 return metricsCollector.obtainRequestMetrics(requestContext);
49 }
50 }