From 88377b6889fbe0baa2deca98997f175ccdb2b07b Mon Sep 17 00:00:00 2001
From: ascrutae <ascrutae@gmail.com>
Date: Fri, 3 Jun 2016 18:04:59 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4agent-test=E5=B7=A5=E7=A8=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 test/skywalking-agent-test/pom.xml            | 37 ++++++++++
 .../skywalking/agent/test/AgentTest.java      | 20 ++++++
 .../agent/test/utils/JedisUtils.java          | 72 +++++++++++++++++++
 3 files changed, 129 insertions(+)
 create mode 100644 test/skywalking-agent-test/pom.xml
 create mode 100644 test/skywalking-agent-test/src/main/java/com/ai/cloud/skywalking/agent/test/AgentTest.java
 create mode 100644 test/skywalking-agent-test/src/main/java/com/ai/cloud/skywalking/agent/test/utils/JedisUtils.java

diff --git a/test/skywalking-agent-test/pom.xml b/test/skywalking-agent-test/pom.xml
new file mode 100644
index 000000000..10ceb19a9
--- /dev/null
+++ b/test/skywalking-agent-test/pom.xml
@@ -0,0 +1,37 @@
+<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <modelVersion>4.0.0</modelVersion>
+
+    <groupId>com.ai.cloud</groupId>
+    <artifactId>skywalking-agent-test</artifactId>
+    <version>1.0-Final</version>
+    <packaging>jar</packaging>
+
+    <name>skywalking-agent-test</name>
+    <url>http://maven.apache.org</url>
+
+    <properties>
+        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
+    </properties>
+
+    <dependencies>
+        <dependency>
+            <groupId>junit</groupId>
+            <artifactId>junit</artifactId>
+            <version>3.8.1</version>
+            <scope>test</scope>
+        </dependency>
+        <!-- http://mvnrepository.com/artifact/org.apache.logging.log4j/log4j-core -->
+        <dependency>
+            <groupId>org.apache.logging.log4j</groupId>
+            <artifactId>log4j-core</artifactId>
+            <version>2.6</version>
+        </dependency>
+        <!-- http://mvnrepository.com/artifact/redis.clients/jedis -->
+        <dependency>
+            <groupId>redis.clients</groupId>
+            <artifactId>jedis</artifactId>
+            <version>2.8.1</version>
+        </dependency>
+    </dependencies>
+</project>
diff --git a/test/skywalking-agent-test/src/main/java/com/ai/cloud/skywalking/agent/test/AgentTest.java b/test/skywalking-agent-test/src/main/java/com/ai/cloud/skywalking/agent/test/AgentTest.java
new file mode 100644
index 000000000..eaeb25f27
--- /dev/null
+++ b/test/skywalking-agent-test/src/main/java/com/ai/cloud/skywalking/agent/test/AgentTest.java
@@ -0,0 +1,20 @@
+package com.ai.cloud.skywalking.agent.test;
+
+import com.ai.cloud.skywalking.agent.test.utils.JedisUtils;
+
+/**
+ * Created by xin on 16-6-3.
+ */
+public class AgentTest {
+
+    public static void main(String[] args) {
+        System.out.println("Begin...");
+        JedisUtils.setData("testKey1", "testKey2");
+
+        System.out.println("testKey1 = " + JedisUtils.getData("testKey1"));
+
+        JedisUtils.expire("testKey1");
+
+        System.out.println("End.....");
+    }
+}
diff --git a/test/skywalking-agent-test/src/main/java/com/ai/cloud/skywalking/agent/test/utils/JedisUtils.java b/test/skywalking-agent-test/src/main/java/com/ai/cloud/skywalking/agent/test/utils/JedisUtils.java
new file mode 100644
index 000000000..40e8d4bfe
--- /dev/null
+++ b/test/skywalking-agent-test/src/main/java/com/ai/cloud/skywalking/agent/test/utils/JedisUtils.java
@@ -0,0 +1,72 @@
+package com.ai.cloud.skywalking.agent.test.utils;
+
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
+import redis.clients.jedis.Jedis;
+import redis.clients.jedis.JedisPool;
+
+import java.io.InputStream;
+import java.util.Properties;
+
+public class JedisUtils {
+    private static JedisPool jedisPool;
+    private static Logger logger = LogManager.getLogger(JedisUtils.class);
+
+    static {
+        InputStream redisConfigFileStream = JedisUtils.class.getResourceAsStream("/redis.conf");
+        Properties jedisConfig = new Properties();
+        try {
+            jedisConfig.load(redisConfigFileStream);
+        } catch (Exception e) {
+            System.err.print("Failed to load redis.conf");
+            System.exit(-1);
+        }
+
+        jedisPool = new JedisPool(jedisConfig.getProperty("redis.ip", "127.0.0.1"),
+                Integer.parseInt(jedisConfig.getProperty("redis.port", "6379")));
+    }
+
+    public static void setData(String key, String value) {
+        Jedis jedis = null;
+        try {
+            jedis = jedisPool.getResource();
+            jedis.set(key, value);
+        } catch (Exception e) {
+            logger.error("Failed to set data", e);
+        } finally {
+            if (jedis != null) {
+                jedis.close();
+            }
+        }
+    }
+
+    public static String getData(String key) {
+        Jedis jedis = null;
+        try {
+            jedis = jedisPool.getResource();
+            return jedis.get(key);
+        } catch (Exception e) {
+            logger.error("Failed to set data", e);
+        } finally {
+            if (jedis != null) {
+                jedis.close();
+            }
+        }
+
+        return null;
+    }
+
+    public static void expire(String key) {
+        Jedis jedis = null;
+        try {
+            jedis = jedisPool.getResource();
+            jedis.expire(key, 0);
+        } catch (Exception e) {
+            logger.error("Failed to set data", e);
+        } finally {
+            if (jedis != null) {
+                jedis.close();
+            }
+        }
+    }
+}