chore: updated examples

Author: Marvin Zhang

src/main/java/io/crawlab/example/Main.java

@@ -5,6 +5,7 @@
 import java.util.HashMap;
 import java.util.Arrays;
 
+import com.fasterxml.jackson.databind.annotation.JsonSerialize;
 import io.crawlab.sdk.CrawlabSDK;
 
 public class Main {
@@ -24,20 +25,15 @@ public static void main(String[] args) {
         dataList.add(Map.of("name", "Jane", "age", 30));
         CrawlabSDK.saveItem(dataList);
 
-        // Passing Dynamic Arguments
+        // Passing a Map directly
         Map<String, Object> dataMap = new HashMap<>();
         dataMap.put("name", "John");
         dataMap.put("age", 25);
         dataMap.put("skills", Arrays.asList("Java", "Python"));
         CrawlabSDK.saveItem(dataMap);
 
-        // Using varargs for multiple items
-        CrawlabSDK.saveItem(
-                Map.of("name", "John", "age", 25),
-                Map.of("name", "Jane", "age", 30)
-        );
-
         // Using custom objects
+        @JsonSerialize
         class Person {
             private String name;
             private int age;
@@ -46,8 +42,15 @@ public Person(String name, int age) {
                 this.name = name;
                 this.age = age;
             }
-        }
 
+            public String getName() {
+                return name;
+            }
+
+            public int getAge() {
+                return age;
+            }
+        }
         Person person1 = new Person("John", 25);
         Person person2 = new Person("Jane", 30);
         CrawlabSDK.saveItem(person1, person2);

src/test/java/io/crawlab/sdk/CrawlabSDKTest.java

@@ -1,5 +1,6 @@
 package io.crawlab.sdk;
 
+import com.fasterxml.jackson.databind.annotation.JsonSerialize;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.BeforeEach;
 
@@ -19,31 +20,53 @@ void setUp() {
 
     @Test
     void testSaveItemWithMap() {
-        Map<String, Object> data = Map.of("name", "John", "age", 25);
-        CrawlabSDK.saveItem(data);
+        Map<String, Object> data1 = Map.of("name", "John", "age", 25);
+        Map<String, Object> data2 = Map.of("name", "Jane", "age", 30);
+        CrawlabSDK.saveItem(data1, data2);
         String output = outputStreamCaptor.toString().trim();
 
         assertTrue(output.contains("\"type\":\"data\""));
         assertTrue(output.contains("\"payload\":["));
         assertTrue(output.contains("\"name\":\"John\""));
         assertTrue(output.contains("\"age\":25"));
+        assertTrue(output.contains("\"name\":\"Jane\""));
+        assertTrue(output.contains("\"age\":30"));
         assertTrue(output.contains("\"ipc\":true"));
     }
 
     @Test
     void testSaveItemWithClass() {
-        TestData data = new TestData("John", 25);
-        CrawlabSDK.saveItem(data);
+        Person data1 = new Person("John", 25);
+        Person data2 = new Person("Jane", 30);
+        CrawlabSDK.saveItem(data1, data2);
         String output = outputStreamCaptor.toString().trim();
 
         assertTrue(output.contains("\"type\":\"data\""));
         assertTrue(output.contains("\"payload\":["));
         assertTrue(output.contains("\"name\":\"John\""));
         assertTrue(output.contains("\"age\":25"));
+        assertTrue(output.contains("\"name\":\"Jane\""));
+        assertTrue(output.contains("\"age\":30"));
         assertTrue(output.contains("\"ipc\":true"));
     }
 
     // Helper class for testing object serialization
-    private record TestData(String name, int age) {
+    @JsonSerialize
+    class Person {
+        private String name;
+        private int age;
+
+        public Person(String name, int age) {
+            this.name = name;
+            this.age = age;
+        }
+
+        public String getName() {
+            return name;
+        }
+
+        public int getAge() {
+            return age;
+        }
     }
-} 
+}