乱七八糟的改了不少，心烦意乱,妈的睡觉

Author: zhangyingwei

cockroach-core/src/main/java/com/zhangyingwei/cockroach/executer/response/TaskErrorResponse.java

@@ -22,7 +22,7 @@ public TaskErrorResponse(TaskResponse response) throws IOException {
     }
 
     @Override
-    public String getContent() throws IOException {
+    public ResponseContent getContent() throws IOException {
         return this.response.getContent();
     }
 

cockroach-core/src/main/java/com/zhangyingwei/cockroach/executer/response/TaskResponse.java

@@ -61,6 +61,11 @@ public TaskResponse setTask(Task task) {
         return this;
     }
 
+    public TaskResponse charset(String charset) {
+        this.content.charset(charset);
+        return this;
+    }
+
     public Elements select(String cssSelect) throws IOException {
         return this.content.toDocument().select(cssSelect);
     }

cockroach-core/src/main/java/com/zhangyingwei/cockroach/store/DescribeStore.java

@@ -13,7 +13,7 @@ public void store(TaskResponse response) throws Exception {
         logger.info("==================desc=================");
         logger.info(String.format("task id: %s",response.getTask().getId()));
         logger.info(String.format("thread name: %s",Thread.currentThread().getName()));
-        logger.info(String.format("bytes: %d",response.getContentBytes().length));
+        logger.info(String.format("bytes: %d",response.getContent().bytes().length));
         logger.info("=======================================");
     }
 }

cockroach-samples/cockroach-samples.iml

@@ -1,5 +1,5 @@
 <?xml version="1.0" encoding="UTF-8"?>
-<module org.jetbrains.idea.maven.project.MavenProjectsManager.isMavenModule="true" type="JAVA_MODULE" version="4">
+<module type="JAVA_MODULE" version="4">
   <component name="NewModuleRootManager" LANGUAGE_LEVEL="JDK_1_8" inherit-compiler-output="false">
     <output url="file://$MODULE_DIR$/target/classes" />
     <output-test url="file://$MODULE_DIR$/target/test-classes" />

cockroach-samples/src/main/java/com/zhangyingwei/cockroach/samples/juejin/JueJinApplication.java

@@ -0,0 +1,30 @@
+package com.zhangyingwei.cockroach.samples.juejin;
+
+import com.zhangyingwei.cockroach.CockroachApplication;
+import com.zhangyingwei.cockroach.annotation.AppName;
+import com.zhangyingwei.cockroach.annotation.EnableAutoConfiguration;
+import com.zhangyingwei.cockroach.annotation.Store;
+import com.zhangyingwei.cockroach.annotation.ThreadConfig;
+import com.zhangyingwei.cockroach.executer.task.Task;
+import com.zhangyingwei.cockroach.queue.CockroachQueue;
+import com.zhangyingwei.cockroach.queue.TaskQueue;
+import com.zhangyingwei.cockroach.samples.juejin.store.JueJinStore;
+
+/**
+ * @author: zhangyw
+ * @date: 2018/2/26
+ * @time: 下午8:43
+ * @desc:
+ */
+
+@EnableAutoConfiguration
+@AppName("掘金")
+@ThreadConfig(num = 1,sleep = 500)
+@Store(JueJinStore.class)
+public class JueJinApplication {
+    public static void main(String[] args) throws Exception {
+        CockroachQueue queue = TaskQueue.of();
+        queue.push(new Task("https://timeline-merger-ms.juejin.im/v1/get_entry_by_rank?src=web&uid=58368a0461ff4b475bd600bc&device_id=1519648660286&token=eyJhY2Nlc3NfdG9rZW4iOiJqQVlaSWxIN1U3aGJnV3YzIiwicmVmcmVzaF90b2tlbiI6IlV5RnA3eDFaVWp4bk9jRVEiLCJ0b2tlbl90eXBlIjoibWFjIiwiZXhwaXJlX2luIjoyNTkyMDAwfQ%3D%3D&limit=20&category=all&recomment=1"));
+        CockroachApplication.run(JueJinApplication.class, queue);
+    }
+}

cockroach-samples/src/main/java/com/zhangyingwei/cockroach/samples/juejin/store/JueJinStore.java

@@ -0,0 +1,24 @@
+package com.zhangyingwei.cockroach.samples.juejin.store;
+
+import com.zhangyingwei.cockroach.executer.response.TaskResponse;
+import com.zhangyingwei.cockroach.store.IStore;
+import net.sf.json.JSONObject;
+
+/**
+ * @author: zhangyw
+ * @date: 2018/2/26
+ * @time: 下午8:49
+ * @desc:
+ */
+public class JueJinStore implements IStore{
+    @Override
+    public void store(TaskResponse response) throws Exception {
+        response.getContent().toJsonObject()
+                .getJSONObject("d")
+                .getJSONArray("entrylist")
+                .forEach(item -> {
+                    JSONObject itemJson = JSONObject.fromObject(item);
+                    System.out.println(itemJson.getString("title"));
+                });
+    }
+}

cockroach-test/cockroach-test.iml

@@ -1,5 +1,5 @@
 <?xml version="1.0" encoding="UTF-8"?>
-<module org.jetbrains.idea.maven.project.MavenProjectsManager.isMavenModule="true" type="JAVA_MODULE" version="4">
+<module type="JAVA_MODULE" version="4">
   <component name="NewModuleRootManager" LANGUAGE_LEVEL="JDK_1_8" inherit-compiler-output="false">
     <output url="file://$MODULE_DIR$/target/classes" />
     <output-test url="file://$MODULE_DIR$/target/test-classes" />

cockroach-test/src/test/java/com/zhangyingwei/cockroach/store/ImageStore.java

@@ -11,7 +11,7 @@ public class ImageStore implements IStore {
 
     @Override
     public void store(TaskResponse response) throws Exception {
-        byte[] bytes = response.getContentBytes();
+        byte[] bytes = response.getContent().bytes();
         File file = new File("image.jpeg");
         DataOutputStream outputStream = new DataOutputStream(new FileOutputStream(file));
         outputStream.write(bytes);

cockroach-test/src/test/java/com/zhangyingwei/cockroach/store/MeiZhiStore.java

@@ -32,7 +32,7 @@ public void store(TaskResponse response) throws Exception {
             });
         } else {
             File file = new File("D://meizitu/"+UUID.randomUUID()+".png");
-            byte[] bytes = response.getContentBytes();
+            byte[] bytes = response.getContent().bytes();
             DataOutputStream outputStream = new DataOutputStream(new FileOutputStream(file));
             outputStream.write(bytes);
             outputStream.close();

cockroach-test/src/test/java/com/zhangyingwei/cockroach/store/ZhiHuMeiZhi.java

@@ -32,7 +32,7 @@ public void store(TaskResponse response) throws Exception {
             });
         } else {
             File file = new File("D://zhihumeizitu2/"+ UUID.randomUUID()+".jpg");
-            byte[] bytes = response.getContentBytes();
+            byte[] bytes = response.getContent().bytes();
             DataOutputStream outputStream = new DataOutputStream(new FileOutputStream(file));
             outputStream.write(bytes);
             outputStream.close();