X-Git-Url: https://git.argeo.org/?a=blobdiff_plain;f=jcr%2Forg.argeo.cms.jcr%2Fsrc%2Forg%2Fargeo%2Fcms%2Fjcr%2Ftabular%2FJcrTabularWriter.java;fp=jcr%2Forg.argeo.cms.jcr%2Fsrc%2Forg%2Fargeo%2Fcms%2Fjcr%2Ftabular%2FJcrTabularWriter.java;h=cc3e0d7a9f428621b3998fc8e3f283037dbd4c5f;hb=8282011b0e20e80704b209ad55fa9fb132e16280;hp=0000000000000000000000000000000000000000;hpb=633a8acd189cc22f06944d278879601189be1bc8;p=lgpl%2Fargeo-commons.git

diff --git a/jcr/org.argeo.cms.jcr/src/org/argeo/cms/jcr/tabular/JcrTabularWriter.java b/jcr/org.argeo.cms.jcr/src/org/argeo/cms/jcr/tabular/JcrTabularWriter.java
new file mode 100644
index 000000000..cc3e0d7a9
--- /dev/null
+++ b/jcr/org.argeo.cms.jcr/src/org/argeo/cms/jcr/tabular/JcrTabularWriter.java
@@ -0,0 +1,82 @@
+package org.argeo.cms.jcr.tabular;
+
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.InputStream;
+import java.util.List;
+
+import javax.jcr.Binary;
+import javax.jcr.Node;
+import javax.jcr.Property;
+import javax.jcr.PropertyType;
+import javax.jcr.RepositoryException;
+
+import org.apache.commons.io.IOUtils;
+import org.argeo.cms.ArgeoTypes;
+import org.argeo.cms.tabular.TabularColumn;
+import org.argeo.cms.tabular.TabularWriter;
+import org.argeo.jcr.JcrException;
+import org.argeo.jcr.JcrUtils;
+import org.argeo.util.CsvWriter;
+
+/** Write / reference tabular content in a JCR repository. */
+public class JcrTabularWriter implements TabularWriter {
+	private Node contentNode;
+	private ByteArrayOutputStream out;
+	private CsvWriter csvWriter;
+	
+	@SuppressWarnings("unused")
+	private final List<TabularColumn> columns;
+
+	/** Creates a table node */
+	public JcrTabularWriter(Node tableNode, List<TabularColumn> columns,
+			String contentNodeType) {
+		try {
+			this.columns = columns;
+			for (TabularColumn column : columns) {
+				String normalized = JcrUtils.replaceInvalidChars(column
+						.getName());
+				Node columnNode = tableNode.addNode(normalized,
+						ArgeoTypes.ARGEO_COLUMN);
+				columnNode.setProperty(Property.JCR_TITLE, column.getName());
+				if (column.getType() != null)
+					columnNode.setProperty(Property.JCR_REQUIRED_TYPE,
+							PropertyType.nameFromValue(column.getType()));
+				else
+					columnNode.setProperty(Property.JCR_REQUIRED_TYPE,
+							PropertyType.TYPENAME_STRING);
+			}
+			contentNode = tableNode.addNode(Property.JCR_CONTENT,
+					contentNodeType);
+			if (contentNodeType.equals(ArgeoTypes.ARGEO_CSV)) {
+				contentNode.setProperty(Property.JCR_MIMETYPE, "text/csv");
+				contentNode.setProperty(Property.JCR_ENCODING, "UTF-8");
+				out = new ByteArrayOutputStream();
+				csvWriter = new CsvWriter(out);
+			}
+		} catch (RepositoryException e) {
+			throw new JcrException("Cannot create table node " + tableNode, e);
+		}
+	}
+
+	public void appendRow(Object[] row) {
+		csvWriter.writeLine(row);
+	}
+
+	public void close() {
+		Binary binary = null;
+		InputStream in = null;
+		try {
+			// TODO parallelize with pipes and writing from another thread
+			in = new ByteArrayInputStream(out.toByteArray());
+			binary = contentNode.getSession().getValueFactory()
+					.createBinary(in);
+			contentNode.setProperty(Property.JCR_DATA, binary);
+		} catch (RepositoryException e) {
+			throw new JcrException("Cannot store data in " + contentNode, e);
+		} finally {
+			IOUtils.closeQuietly(in);
+			JcrUtils.closeQuietly(binary);
+		}
+	}
+}