package binary

import (
	"bytes"
	"compress/gzip"
	"compress/zlib"
	"sync"
)

var bufferPool = sync.Pool{
	New: func() interface{} {
		return new(Writer)
	},
}

// NewWriter 从池中取出一个 Writer
func NewWriter() *Writer {
	w := bufferPool.Get().(*Writer)
	if w == nil {
		return new(Writer)
	}
	return w
}

// PutBuffer 将 Writer 放回池中
func PutBuffer(w *Writer) {
	// See https://golang.org/issue/23199
	const maxSize = 1 << 16
	if w.Cap() < maxSize { // 对于大Buffer直接丢弃
		w.Reset()
		bufferPool.Put(w)
	}
}

type gzipWriter struct {
	w   *gzip.Writer
	buf *bytes.Buffer
}

var gzipPool = sync.Pool{
	New: func() interface{} {
		buf := new(bytes.Buffer)
		w := gzip.NewWriter(buf)
		return &gzipWriter{
			w:   w,
			buf: buf,
		}
	},
}

func acquireGzipWriter() *gzipWriter {
	ret := gzipPool.Get().(*gzipWriter)
	ret.buf.Reset()
	ret.w.Reset(ret.buf)
	return ret
}

func releaseGzipWriter(w *gzipWriter) {
	// See https://golang.org/issue/23199
	const maxSize = 1 << 16
	if w.buf.Cap() < maxSize {
		w.buf.Reset()
		gzipPool.Put(w)
	}
}

type zlibWriter struct {
	w   *zlib.Writer
	buf *bytes.Buffer
}

var zlibPool = sync.Pool{
	New: func() interface{} {
		buf := new(bytes.Buffer)
		w := zlib.NewWriter(buf)
		return &zlibWriter{
			w:   w,
			buf: buf,
		}
	},
}

func acquireZlibWriter() *zlibWriter {
	ret := zlibPool.Get().(*zlibWriter)
	ret.buf.Reset()
	ret.w.Reset(ret.buf)
	return ret
}

func releaseZlibWriter(w *zlibWriter) {
	// See https://golang.org/issue/23199
	const maxSize = 1 << 16
	if w.buf.Cap() < maxSize {
		w.buf.Reset()
		zlibPool.Put(w)
	}
}

const size128k = 128 * 1024

var b128kPool = sync.Pool{
	New: func() interface{} {
		return make128kSlicePointer()
	},
}

// Get128KBytes 获取一个128k大小 []byte
func Get128KBytes() *[]byte {
	buf := b128kPool.Get().(*[]byte)
	if buf == nil {
		return make128kSlicePointer()
	}
	if cap(*buf) < size128k {
		return make128kSlicePointer()
	}
	*buf = (*buf)[:size128k]
	return buf
}

// Put128KBytes 放回一个128k大小 []byte
func Put128KBytes(b *[]byte) {
	if cap(*b) < size128k || cap(*b) > 2*size128k { // 太大或太小的 []byte 不要放入
		return
	}
	*b = (*b)[:cap(*b)]
	b128kPool.Put(b)
}

func make128kSlicePointer() *[]byte {
	data := make([]byte, size128k)
	return &data
}