yyfbaloot/public/ByteArray.go

package public

import (
	"encoding/binary"
	"io"
)

type ByteArray struct {
	buf      []byte
	posWrite int
	posRead  int
	endian   binary.ByteOrder
}

var ByteArrayEndian binary.ByteOrder = binary.LittleEndian

func CreateByteArray(bytes []byte) *ByteArray {
	var ba *ByteArray
	if bytes != nil && len(bytes) > 0 {
		ba = &ByteArray{buf: bytes}
	} else {
		ba = &ByteArray{}
	}

	ba.endian = binary.LittleEndian

	return ba
}

func (this *ByteArray) Length() int {
	return len(this.buf)
}

func (this *ByteArray) Available() int {
	return this.Length() - this.posRead
}

func (this *ByteArray) SetEndian(endian binary.ByteOrder) {
	this.endian = endian
}

func (this *ByteArray) GetEndian() binary.ByteOrder {
	if this.endian == nil {
		return ByteArrayEndian
	}
	return this.endian
}

func (this *ByteArray) grow(l int) {
	if l == 0 {
		return
	}
	space := len(this.buf) - this.posWrite
	if space >= l {
		return
	}

	needGrow := l - space
	bufGrow := make([]byte, needGrow)

	this.buf = append(this.buf, bufGrow...)
}

func (this *ByteArray) SetWritePos(pos int) error {
	if pos > this.Length() {
		this.posWrite = this.Length()
		return io.EOF
	} else {
		this.posWrite = pos
	}
	return nil
}

func (this *ByteArray) SetWriteEnd() {
	this.SetWritePos(this.Length())
}

func (this *ByteArray) GetWritePos() int {
	return this.posWrite
}

func (this *ByteArray) SetReadPos(pos int) error {
	if pos > this.Length() {
		this.posRead = this.Length()
		return io.EOF
	} else {
		this.posRead = pos
	}
	return nil
}

func (this *ByteArray) SetReadEnd() {
	this.SetReadPos(this.Length())
}

func (this *ByteArray) GetReadPos() int {
	return this.posRead
}

func (this *ByteArray) Seek(pos int) error {
	err := this.SetWritePos(pos)
	this.SetReadPos(pos)

	return err
}

func (this *ByteArray) Reset() {
	this.buf = []byte{}
	this.Seek(0)
}

func (this *ByteArray) Bytes() []byte {
	return this.buf
}

func (this *ByteArray) BytesAvailable() []byte {
	return this.buf[this.posRead:]
}

//==========write
func (this *ByteArray) Write(bytes []byte) (l int, err error) {
	this.grow(len(bytes))

	l = copy(this.buf[this.posWrite:], bytes)
	this.posWrite += l

	return l, nil
}

func (this *ByteArray) WriteBytes(bytes []byte) (l int, err error) {
	return this.Write(bytes)
}

func (this *ByteArray) WriteByte(b byte) error {
	bytes := make([]byte, 1)
	bytes[0] = b
	_, err := this.WriteBytes(bytes)
	return err
}

func (this *ByteArray) WriteInt8(value int8) {
	binary.Write(this, this.endian, &value)
}

func (this *ByteArray) WriteUInt8(value uint8) {
	binary.Write(this, this.endian, &value)
}

func (this *ByteArray) WriteInt16(value int16) {
	binary.Write(this, this.endian, &value)
}

func (this *ByteArray) WriteUInt16(value uint16) {
	binary.Write(this, this.endian, &value)
}

func (this *ByteArray) WriteInt32(value int32) {
	binary.Write(this, this.endian, &value)
}

func (this *ByteArray) WriteUInt32(value uint32) {
	binary.Write(this, this.endian, &value)
}

/*func (this *ByteArray) WriteInt(value int) {
	binary.Write(this, this.endian, &value)
}

func (this *ByteArray) WriteUInt(value uint) {
	binary.Write(this, this.endian, &value)
}
*/
func (this *ByteArray) WriteInt64(value int64) {
	binary.Write(this, this.endian, &value)
}

func (this *ByteArray) WriteFloat32(value float32) {
	binary.Write(this, this.endian, &value)
}

func (this *ByteArray) WriteFloat64(value float64) {
	binary.Write(this, this.endian, &value)
}

func (this *ByteArray) WriteBool(value bool) {
	var bb byte
	if value {
		bb = 1
	} else {
		bb = 0
	}

	this.WriteByte(bb)
}

func (this *ByteArray) WriteString(value string) {
	this.WriteBytes([]byte(value))
}

func (this *ByteArray) WriteStringWithLen(value string, stringLen int) {
	byString := []byte(value)
	this.WriteBytes(byString)
	for i := 0; i < stringLen-len(byString); i++ {
		this.WriteUInt8(0)
	}
}

func (this *ByteArray) WriteUTF(value string) {
	this.WriteInt16(int16(len(value)))
	this.WriteBytes([]byte(value))
}

//==========read

func (this *ByteArray) Read(bytes []byte) (l int, err error) {
	if len(bytes) == 0 {
		return
	}
	if len(bytes) > this.Length()-this.posRead {
		return 0, io.EOF
	}
	l = copy(bytes, this.buf[this.posRead:])
	this.posRead += l

	return l, nil
}

func (this *ByteArray) ReadBytes(bytes []byte, length int, offset int) (l int, err error) {
	return this.Read(bytes[offset : offset+length])
}

func (this *ByteArray) ReadByte() (b byte, err error) {
	bytes := make([]byte, 1)
	_, err = this.ReadBytes(bytes, 1, 0)

	if err == nil {
		b = bytes[0]
	}

	return
}

func (this *ByteArray) ReadInt8() (ret int8, err error) {
	err = binary.Read(this, this.endian, &ret)
	return
}

func (this *ByteArray) ReadUInt8() (ret uint8, err error) {
	err = binary.Read(this, this.endian, &ret)
	return
}

func (this *ByteArray) ReadInt16() (ret int16, err error) {
	err = binary.Read(this, this.endian, &ret)
	return
}

func (this *ByteArray) ReadUInt16() (ret uint16, err error) {
	err = binary.Read(this, this.endian, &ret)
	return
}

func (this *ByteArray) ReadInt32() (ret int32, err error) {
	err = binary.Read(this, this.endian, &ret)
	return
}

func (this *ByteArray) ReadUInt32() (ret uint32, err error) {
	err = binary.Read(this, this.endian, &ret)
	return
}

func (this *ByteArray) ReadInt64() (ret int64, err error) {
	err = binary.Read(this, this.endian, &ret)
	return
}

func (this *ByteArray) ReadFloat32() (ret float32, err error) {
	err = binary.Read(this, this.endian, &ret)
	return
}

func (this *ByteArray) ReadFloat64() (ret float64, err error) {
	err = binary.Read(this, this.endian, &ret)
	return
}

func (this *ByteArray) ReadBool() (ret bool, err error) {
	var bb byte
	bb, err = this.ReadByte()
	if err == nil {
		if bb == 1 {
			ret = true
		} else {
			ret = false
		}
	} else {
		ret = false
	}
	return
}

func (this *ByteArray) ReadString(length int) (ret string, err error) {
	bytes := make([]byte, length)
	_, err = this.ReadBytes(bytes, length, 0)
	if err == nil {
		// 有可能有末尾的多余空格
		for i := 0; i < length; i++ {
			if bytes[i] == 0 {
				bytes = bytes[:i]
				break
			}
		}
		ret = string(bytes)
	} else {
		ret = ""
	}
	return
}

func (this *ByteArray) ReadUTF() (ret string, err error) {
	var l int16
	l, err = this.ReadInt16()

	if err != nil {
		return "", err
	}

	return this.ReadString(int(l))
}