add client to query data

Author: ethendev

pom.xml

@@ -19,7 +19,8 @@
         <jackson.version>2.9.6</jackson.version>
         <lombok.version>1.18.2</lombok.version>
         <junit.version>4.12</junit.version>
-        <logback.version>1.2.1</logback.version>
+        <slf4j.version>1.7.25</slf4j.version>
+        <okhttp.version>2.7.5</okhttp.version>
     </properties>
 
     <dependencies>
@@ -36,6 +37,18 @@
             <scope>provided</scope>
         </dependency>
 
+        <dependency>
+            <groupId>com.squareup.okhttp</groupId>
+            <artifactId>okhttp</artifactId>
+            <version>${okhttp.version}</version>
+        </dependency>
+
+        <dependency>
+            <groupId>org.slf4j</groupId>
+            <artifactId>slf4j-api</artifactId>
+            <version>${slf4j.version}</version>
+        </dependency>
+
         <dependency>
             <groupId>junit</groupId>
             <artifactId>junit</artifactId>

src/main/java/com/github/ethendev/jdruid/DruidClient.java

@@ -0,0 +1,52 @@
+package com.github.ethendev.jdruid;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.github.ethendev.jdruid.query.BaseQuery;
+import com.squareup.okhttp.*;
+import lombok.extern.slf4j.Slf4j;
+
+import java.io.IOException;
+
+/**
+ * @author: Ethan
+ * @Date: 2018/9/18
+ */
+@Slf4j
+public class DruidClient {
+
+    private static final MediaType JSON = MediaType.parse("application/json; charset=utf-8");
+    private final OkHttpClient client;
+    private final ObjectMapper mapper;
+
+    public DruidClient() {
+        this.client = new OkHttpClient();
+        this.mapper = new ObjectMapper();
+    }
+
+    public String query(String url, String param) throws IOException {
+        Request request = new Request.Builder()
+                .url(url)
+                .post(RequestBody.create(JSON, param))
+                .build();
+        Response response = client.newCall(request).execute();
+        if (!response.isSuccessful()) {
+            throw new IOException("query error: " + response);
+        }
+        return response.body().toString();
+    }
+
+    public String query(String url, BaseQuery param) throws IOException {
+        return query(url, mapper.writeValueAsString(param));
+    }
+
+    public Object query(String url, String param, Class<?> classType) throws IOException {
+        String result = query(url, param);
+        return mapper.readValue(result, classType);
+    }
+
+    public Object query(String url, BaseQuery param, Class<?> classType) throws IOException {
+        String result = query(url, mapper.writeValueAsString(param));
+        return mapper.readValue(result, classType);
+    }
+
+}

src/main/java/com/github/ethendev/jdruid/filter/BoundFilter.java

@@ -14,7 +14,7 @@
 @EqualsAndHashCode(callSuper = true)
 @JsonInclude(JsonInclude.Include.NON_NULL)
 public class BoundFilter extends BaseFilter {
-    private static String FILTER_TYPE = "bound";
+    private static final String FILTER_TYPE = "bound";
 
     private String dimension;
     private String lower;

src/main/java/com/github/ethendev/jdruid/filter/InFilter.java

@@ -13,7 +13,7 @@
 @Getter
 @EqualsAndHashCode(callSuper = true)
 public class InFilter extends BaseFilter {
-    private static String FILTER_TYPE = "in";
+    private static final String FILTER_TYPE = "in";
 
     private String dimension;
     private List<Object> values;

src/main/java/com/github/ethendev/jdruid/filter/JavaScriptFilter.java

@@ -11,7 +11,7 @@
 @Getter
 @EqualsAndHashCode(callSuper = true)
 public class JavaScriptFilter extends BaseFilter {
-    private static String FILTER_TYPE = "javascript";
+    private static final String FILTER_TYPE = "javascript";
 
     private String dimension;
     private String function;

src/main/java/com/github/ethendev/jdruid/filter/RegexFilter.java

@@ -11,7 +11,7 @@
 @Getter
 @EqualsAndHashCode(callSuper = true)
 public class RegexFilter extends BaseFilter {
-    private static String FILTER_TYPE = "regex";
+    private static final String FILTER_TYPE = "regex";
 
     private String dimension;
     private String pattern;

src/main/java/com/github/ethendev/jdruid/filter/SearchFilter.java

@@ -11,7 +11,7 @@
 @Getter
 @EqualsAndHashCode(callSuper = true)
 public class SearchFilter extends BaseFilter {
-    private static String FILTER_TYPE = "search";
+    private static final String FILTER_TYPE = "search";
 
     private String dimension;
 

src/main/java/com/github/ethendev/jdruid/filter/SelectorFilter.java

@@ -11,7 +11,7 @@
 @Getter
 @EqualsAndHashCode(callSuper = true)
 public class SelectorFilter extends BaseFilter {
-    private static String FILTER_TYPE = "regex";
+    private static String FILTER_TYPE = "selector";
 
     private String dimension;
     private String value;

src/main/java/com/github/ethendev/jdruid/query/GranularityType.java

@@ -1,6 +1,6 @@
 package com.github.ethendev.jdruid.query;
 
-import com.fasterxml.jackson.annotation.JsonGetter;
+import com.fasterxml.jackson.annotation.JsonValue;
 
 /**
  * @author: Ethan
@@ -26,7 +26,7 @@ public enum GranularityType {
         this.value = value;
     }
 
-    @JsonGetter
+    @JsonValue
     public String getValue() {
         return value;
     }

src/main/java/com/github/ethendev/jdruid/query/GroupByQuery.java

@@ -18,6 +18,7 @@
 @Getter
 @JsonInclude(JsonInclude.Include.NON_NULL)
 public class GroupByQuery extends BaseQuery {
+    private static final String QUERY_TYPE = "groupBy";
     private List<String> dimensions;
     private LimitSpec limitSpec;
     // TODO change type
@@ -28,10 +29,10 @@ public class GroupByQuery extends BaseQuery {
     private Context context;
 
     @Builder
-    public GroupByQuery(String queryType, String dataSource, List<String> intervals, GranularityType granularity,
+    public GroupByQuery(String dataSource, List<String> intervals, GranularityType granularity,
                         @NonNull List<String> dimensions, LimitSpec limitSpec, Object having, BaseFilter filter,
                         List<BaseAggregator> aggregators, List<BasePostAggregator> postAggregations, Context context) {
-        super(queryType, dataSource, intervals, granularity);
+        super(QUERY_TYPE, dataSource, intervals, granularity);
         this.dimensions = dimensions;
         this.limitSpec = limitSpec;
         this.having = having;