com.bigdata.rdf.sail.webapp.client.EntityContentProvider.java Source code

Introduction

Here is the source code for com.bigdata.rdf.sail.webapp.client.EntityContentProvider.java

Source

/**
    
Copyright (C) SYSTAP, LLC DBA Blazegraph 2006-2016.  All rights reserved.
    
Contact:
 SYSTAP, LLC DBA Blazegraph
 2501 Calvert ST NW #106
 Washington, DC 20008
 licenses@blazegraph.com
    
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; version 2 of the License.
    
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.
    
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*/
/*
 * Created on Oct 13, 2011
 */
package com.bigdata.rdf.sail.webapp.client;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.ByteBuffer;
import java.util.Iterator;

import org.apache.http.HttpEntity;
import org.eclipse.jetty.client.api.ContentProvider;

public class EntityContentProvider implements ContentProvider {

    private final HttpEntity m_entity;

    public EntityContentProvider(final HttpEntity entity) {
        m_entity = entity;
    }

    @Override
    public long getLength() {
        return m_entity.getContentLength();
    }

    public String getContentType() {
        return m_entity.getContentType().getValue();
    }

    @Override
    public Iterator<ByteBuffer> iterator() {
        try {
            final InputStream instr;

            if (m_entity.isStreaming()) {
                instr = m_entity.getContent();
            } else {
                // If Apache Entity does not stream then we need to double buffer to an output stream - not ideal
                final ByteArrayOutputStream streambuf = new ByteArrayOutputStream();
                m_entity.writeTo(streambuf);

                instr = new ByteArrayInputStream(streambuf.toByteArray());
            }

            return new Iterator<ByteBuffer>() {

                boolean eof = false;
                int bufindex = 0;

                @Override
                public boolean hasNext() {
                    return !eof;
                }

                @Override
                public ByteBuffer next() {
                    final byte[] buf = new byte[1024];
                    try {
                        final int rdlen = instr.read(buf);

                        // System.out.println("Returning ByteBuffer[" + bufindex++ + "] length: " + rdlen);

                        if (rdlen == -1) {
                            eof = true;
                            return ByteBuffer.wrap(buf, 0, 0);
                        } else {
                            return ByteBuffer.wrap(buf, 0, rdlen);
                        }
                    } catch (IOException e) {
                        throw new RuntimeException(e);
                    }
                }

                @Override
                public void remove() {
                    throw new UnsupportedOperationException();
                }
            };
        } catch (Exception e) {
            throw new RuntimeException("Unexpected", e);
        }
    }
}