Mercurial Hosting > luan
view src/org/eclipse/jetty/io/AbstractBuffer.java @ 1038:b71ad168fe34
rename Buffer.length() to remaining()
author | Franklin Schmidt <fschmidt@gmail.com> |
---|---|
date | Thu, 03 Nov 2016 22:16:11 -0600 |
parents | 3c4c7cc7904f |
children | a7319f14ba1e |
line wrap: on
line source
// // ======================================================================== // Copyright (c) 1995-2014 Mort Bay Consulting Pty. Ltd. // ------------------------------------------------------------------------ // All rights reserved. This program and the accompanying materials // are made available under the terms of the Eclipse Public License v1.0 // and Apache License v2.0 which accompanies this distribution. // // The Eclipse Public License is available at // http://www.eclipse.org/legal/epl-v10.html // // The Apache License v2.0 is available at // http://www.opensource.org/licenses/apache2.0.php // // You may elect to redistribute this code under either of these licenses. // ======================================================================== // package org.eclipse.jetty.io; import java.io.IOException; import java.io.InputStream; import java.io.OutputStream; import java.nio.charset.Charset; import org.eclipse.jetty.util.TypeUtil; import org.slf4j.Logger; import org.slf4j.LoggerFactory; /** * * */ public abstract class AbstractBuffer implements Buffer { private static final Logger LOG = LoggerFactory.getLogger(AbstractBuffer.class); private final static boolean __boundsChecking = Boolean.getBoolean("org.eclipse.jetty.io.AbstractBuffer.boundsChecking"); protected final static String __READONLY = "READONLY"; protected int _access; protected int _get; protected int _put; protected int _hash; protected int _hashGet; protected int _hashPut; protected int _mark; protected String _string; protected View _view; /** * Constructor for BufferView * * @param access 0==IMMUTABLE, 1==READONLY, 2==READWRITE */ protected AbstractBuffer(int access) { setMarkIndex(-1); _access = access; } public byte[] asArray() { byte[] bytes = new byte[remaining()]; byte[] array = array(); if (array != null) System.arraycopy(array, getIndex(), bytes, 0, bytes.length); else peek(getIndex(), bytes, 0, remaining()); return bytes; } public Buffer buffer() { return this; } public void clear() { setMarkIndex(-1); setGetIndex(0); setPutIndex(0); } public void compact() { if (isReadOnly()) throw new IllegalStateException(__READONLY); int s = markIndex() >= 0 ? markIndex() : getIndex(); if (s > 0) { byte array[] = array(); int length = putIndex() - s; if (length > 0) { if (array != null) System.arraycopy(array(), s, array(), 0, length); else poke(0, peek(s, length)); } if (markIndex() > 0) setMarkIndex(markIndex() - s); setGetIndex(getIndex() - s); setPutIndex(putIndex() - s); } } @Override public boolean equals(Object obj) { if (obj==this) return true; // reject non buffers; if (obj == null || !(obj instanceof Buffer)) return false; Buffer b = (Buffer) obj; // reject different lengths if (b.remaining() != remaining()) return false; // reject AbstractBuffer with different hash value if (_hash != 0 && obj instanceof AbstractBuffer) { AbstractBuffer ab = (AbstractBuffer) obj; if (ab._hash != 0 && _hash != ab._hash) return false; } // Nothing for it but to do the hard grind. int get=getIndex(); int bi=b.putIndex(); for (int i = putIndex(); i-->get;) { byte b1 = peek(i); byte b2 = b.peek(--bi); if (b1 != b2) return false; } return true; } public byte get() { return peek(_get++); } public int get(byte[] b, int offset, int length) { int gi = getIndex(); int l=remaining(); if (l==0) return -1; if (length>l) length=l; length = peek(gi, b, offset, length); if (length>0) setGetIndex(gi + length); return length; } public Buffer get(int length) { int gi = getIndex(); Buffer view = peek(gi, length); setGetIndex(gi + length); return view; } public final int getIndex() { return _get; } public boolean hasRemaining() { return _put > _get; } @Override public int hashCode() { if (_hash == 0 || _hashGet!=_get || _hashPut!=_put) { int get=getIndex(); byte[] array = array(); if (array==null) { for (int i = putIndex(); i-- >get;) { byte b = peek(i); if ('a' <= b && b <= 'z') b = (byte) (b - 'a' + 'A'); _hash = 31 * _hash + b; } } else { for (int i = putIndex(); i-- >get;) { byte b = array[i]; if ('a' <= b && b <= 'z') b = (byte) (b - 'a' + 'A'); _hash = 31 * _hash + b; } } if (_hash == 0) _hash = -1; _hashGet=_get; _hashPut=_put; } return _hash; } public boolean isImmutable() { return _access <= IMMUTABLE; } public boolean isReadOnly() { return _access <= READONLY; } public int remaining() { return _put - _get; } public void mark() { setMarkIndex(_get - 1); } public void mark(int offset) { setMarkIndex(_get + offset); } public int markIndex() { return _mark; } public byte peek() { return peek(_get); } private Buffer peek(int index, int length) { if (_view == null) { _view = new View(this, -1, index, index + length, isReadOnly() ? READONLY : READWRITE); } else { _view.update(this.buffer()); _view.setMarkIndex(-1); _view.setGetIndex(0); _view.setPutIndex(index + length); _view.setGetIndex(index); } return _view; } @Override public int poke(int index, Buffer src) { _hash=0; int length=src.remaining(); if (index + length > capacity()) { length=capacity()-index; /* if (length<0) throw new IllegalArgumentException("index>capacity(): " + index + ">" + capacity()); */ } byte[] src_array = src.array(); byte[] dst_array = array(); if (src_array != null && dst_array != null) System.arraycopy(src_array, src.getIndex(), dst_array, index, length); else if (src_array != null) { int s=src.getIndex(); for (int i=0;i<length;i++) poke(index++,src_array[s++]); } else if (dst_array != null) { int s=src.getIndex(); for (int i=0;i<length;i++) dst_array[index++]=src.peek(s++); } else { int s=src.getIndex(); for (int i=0;i<length;i++) poke(index++,src.peek(s++)); } return length; } public int poke(int index, byte[] b, int offset, int length) { _hash=0; if (index + length > capacity()) { length=capacity()-index; /* if (length<0) throw new IllegalArgumentException("index>capacity(): " + index + ">" + capacity()); */ } byte[] dst_array = array(); if (dst_array != null) System.arraycopy(b, offset, dst_array, index, length); else { int s=offset; for (int i=0;i<length;i++) poke(index++,b[s++]); } return length; } public int put(Buffer src) { int pi = putIndex(); int l = poke(pi, src); setPutIndex(pi + l); return l; } public void put(byte b) { int pi = putIndex(); poke(pi, b); setPutIndex(pi + 1); } public int put(byte[] b, int offset, int length) { int pi = putIndex(); int l = poke(pi, b, offset, length); setPutIndex(pi + l); return l; } public int put(byte[] b) { int pi = putIndex(); int l = poke(pi, b, 0, b.length); setPutIndex(pi + l); return l; } public final int putIndex() { return _put; } public void reset() { if (markIndex() >= 0) setGetIndex(markIndex()); } public void rewind() { setGetIndex(0); setMarkIndex(-1); } public void setGetIndex(int getIndex) { _get = getIndex; _hash=0; } public void setMarkIndex(int index) { _mark = index; } public void setPutIndex(int putIndex) { _put = putIndex; _hash=0; } public int skip(int n) { if (remaining() < n) n = remaining(); setGetIndex(getIndex() + n); return n; } public Buffer slice() { return peek(getIndex(), remaining()); } public Buffer sliceFromMark() { return sliceFromMark(getIndex() - markIndex() - 1); } public Buffer sliceFromMark(int length) { if (markIndex() < 0) return null; Buffer view = peek(markIndex(), length); setMarkIndex(-1); return view; } public int space() { return capacity() - _put; } public String toDetailString() { StringBuilder buf = new StringBuilder(); buf.append("["); buf.append(super.hashCode()); buf.append(","); buf.append(this.buffer().hashCode()); buf.append(",m="); buf.append(markIndex()); buf.append(",g="); buf.append(getIndex()); buf.append(",p="); buf.append(putIndex()); buf.append(",c="); buf.append(capacity()); buf.append("]={"); if (markIndex() >= 0) { for (int i = markIndex(); i < getIndex(); i++) { byte b = peek(i); TypeUtil.toHex(b,buf); } buf.append("}{"); } int count = 0; for (int i = getIndex(); i < putIndex(); i++) { byte b = peek(i); TypeUtil.toHex(b,buf); if (count++ == 50) { if (putIndex() - i > 20) { buf.append(" ... "); i = putIndex() - 20; } } } buf.append('}'); return buf.toString(); } @Override public String toString() { /* if (isImmutable()) { if (_string == null) _string = new String(asArray(), 0, remaining()); return _string; } return new String(asArray(), 0, remaining()); */ return toString("ISO-8859-1"); } @Override public final String toString(int index, int length) { return peek(index,length).toString(); } @Override public final String toString(String charset) { try { byte[] bytes=array(); if (bytes!=null) return new String(bytes,getIndex(),remaining(),charset); return new String(asArray(), 0, remaining(),charset); } catch(Exception e) { LOG.warn("",e); return new String(asArray(), 0, remaining()); } } @Override public final String toString(Charset charset) { try { byte[] bytes=array(); if (bytes!=null) return new String(bytes,getIndex(),remaining(),charset); return new String(asArray(), 0, remaining(),charset); } catch(Exception e) { LOG.warn("",e); return new String(asArray(), 0, remaining()); } } /* ------------------------------------------------------------ */ public String toDebugString() { return getClass()+"@"+super.hashCode(); } /* ------------------------------------------------------------ */ public int readFrom(InputStream in,int max) throws IOException { byte[] array = array(); int s=space(); if (s>max) s=max; if (array!=null) { int l=in.read(array,_put,s); if (l>0) _put+=l; return l; } else { byte[] buf=new byte[s>1024?1024:s]; int total=0; while (s>0) { int l=in.read(buf,0,buf.length); if (l<0) return total>0?total:-1; int p=put(buf,0,l); assert l==p; s-=l; } return total; } } }