Lesson10 DataStream

Tanechka · Tanechka · commit 6f699cb5d804 · 2017-08-05T23:52:43.000-07:00
diff --git a/src/ru/javawebinar/basejava/model/Resume.java b/src/ru/javawebinar/basejava/model/Resume.java
@@ -40,6 +40,18 @@ public String getUuid() {
         return uuid;
     }
 
+    public String getFullName() {
+        return fullName;
+    }
+
+    public Map<ContactType, String> getContacts() {
+        return contacts;
+    }
+
+    public Map<SectionType, Section> getSections() {
+        return sections;
+    }
+
     public String getContact(ContactType type) {
         return contacts.get(type);
     }
diff --git a/src/ru/javawebinar/basejava/storage/serializer/DataStreamSerializer.java b/src/ru/javawebinar/basejava/storage/serializer/DataStreamSerializer.java
@@ -0,0 +1,40 @@
+package ru.javawebinar.basejava.storage.serializer;
+
+import ru.javawebinar.basejava.model.ContactType;
+import ru.javawebinar.basejava.model.Resume;
+
+import java.io.*;
+import java.util.Map;
+
+public class DataStreamSerializer implements StreamSerializer {
+
+    @Override
+    public void doWrite(Resume r, OutputStream os) throws IOException {
+        try (DataOutputStream dos = new DataOutputStream(os)) {
+            dos.writeUTF(r.getUuid());
+            dos.writeUTF(r.getFullName());
+            Map<ContactType, String> contacts = r.getContacts();
+            dos.writeInt(contacts.size());
+            for (Map.Entry<ContactType, String> entry : contacts.entrySet()) {
+                dos.writeUTF(entry.getKey().name());
+                dos.writeUTF(entry.getValue());
+            }
+            // TODO implements sections
+        }
+    }
+
+    @Override
+    public Resume doRead(InputStream is) throws IOException {
+        try (DataInputStream dis = new DataInputStream(is)) {
+            String uuid = dis.readUTF();
+            String fullName = dis.readUTF();
+            Resume resume = new Resume(uuid, fullName);
+            int size = dis.readInt();
+            for (int i = 0; i < size; i++) {
+                resume.addContact(ContactType.valueOf(dis.readUTF()), dis.readUTF());
+            }
+            // TODO implements sections
+            return resume;
+        }
+    }
+}
diff --git a/test/ru/javawebinar/basejava/storage/AllStorageTest.java b/test/ru/javawebinar/basejava/storage/AllStorageTest.java
@@ -14,7 +14,8 @@
                 ObjectFileStorageTest.class,
                 ObjectPathStorageTest.class,
                 XmlPathStorageTest.class,
-                JsonPathStorageTest.class
+                JsonPathStorageTest.class,
+                DataPathStorageTest.class
         })
 public class AllStorageTest {
 }
diff --git a/test/ru/javawebinar/basejava/storage/DataPathStorageTest.java b/test/ru/javawebinar/basejava/storage/DataPathStorageTest.java
@@ -0,0 +1,10 @@
+package ru.javawebinar.basejava.storage;
+
+import ru.javawebinar.basejava.storage.serializer.DataStreamSerializer;
+
+public class DataPathStorageTest extends AbstractStorageTest {
+
+    public DataPathStorageTest() {
+        super(new PathStorage(STORAGE_DIR.getAbsolutePath(), new DataStreamSerializer()));
+    }
+}
