diff src/main/java/org/msgpack/template/CollectionTemplate.java @ 0:cb825acd883a

first commit
author sugi
date Sat, 18 Oct 2014 15:06:15 +0900
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/main/java/org/msgpack/template/CollectionTemplate.java	Sat Oct 18 15:06:15 2014 +0900
@@ -0,0 +1,69 @@
+//
+// MessagePack for Java
+//
+// Copyright (C) 2009 - 2013 FURUHASHI Sadayuki
+//
+//    Licensed under the Apache License, Version 2.0 (the "License");
+//    you may not use this file except in compliance with the License.
+//    You may obtain a copy of the License at
+//
+//        http://www.apache.org/licenses/LICENSE-2.0
+//
+//    Unless required by applicable law or agreed to in writing, software
+//    distributed under the License is distributed on an "AS IS" BASIS,
+//    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+//    See the License for the specific language governing permissions and
+//    limitations under the License.
+//
+package org.msgpack.template;
+
+import java.io.IOException;
+import java.util.Collection;
+import java.util.LinkedList;
+import org.msgpack.packer.Packer;
+import org.msgpack.unpacker.Unpacker;
+import org.msgpack.MessageTypeException;
+
+public class CollectionTemplate<E> extends AbstractTemplate<Collection<E>> {
+    private Template<E> elementTemplate;
+
+    public CollectionTemplate(Template<E> elementTemplate) {
+        this.elementTemplate = elementTemplate;
+    }
+
+    public void write(Packer pk, Collection<E> target, boolean required)
+            throws IOException {
+        if (target == null) {
+            if (required) {
+                throw new MessageTypeException("Attempted to write null");
+            }
+            pk.writeNil();
+            return;
+        }
+        Collection<E> col = target;
+        pk.writeArrayBegin(col.size());
+        for (E e : col) {
+            elementTemplate.write(pk, e);
+        }
+        pk.writeArrayEnd();
+    }
+
+    public Collection<E> read(Unpacker u, Collection<E> to, boolean required)
+            throws IOException {
+        if (!required && u.trySkipNil()) {
+            return null;
+        }
+        int n = u.readArrayBegin();
+        if (to == null) {
+            to = new LinkedList<E>();
+        } else {
+            to.clear();
+        }
+        for (int i = 0; i < n; i++) {
+            E e = elementTemplate.read(u, null);
+            to.add(e);
+        }
+        u.readArrayEnd();
+        return to;
+    }
+}