// Copyright (C) 2014-2018 Goodrain Co., Ltd. // RAINBOND, Application Management Platform // 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, either version 3 of the License, or // (at your option) any later version. For any non-GPL usage of Rainbond, // one or multiple Commercial Licenses authorized by Goodrain Co., Ltd. // must be obtained first. // 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, see . package util import ( "bytes" "io" "sync" ) type Buffer struct { b *bytes.Buffer m sync.Mutex } func NewBuffer(source []byte) *Buffer { return &Buffer{ b: bytes.NewBuffer(source), m: sync.Mutex{}, } } func (b *Buffer) Read(p []byte) (n int, err error) { b.m.Lock() defer b.m.Unlock() return b.b.Read(p) } func (b *Buffer) Write(p []byte) (n int, err error) { b.m.Lock() defer b.m.Unlock() return b.b.Write(p) } func (b *Buffer) String() string { b.m.Lock() defer b.m.Unlock() return b.b.String() } func (b *Buffer) Bytes() []byte { b.m.Lock() defer b.m.Unlock() return b.b.Bytes() } func (b *Buffer) Cap() int { b.m.Lock() defer b.m.Unlock() return b.b.Cap() } func (b *Buffer) Grow(n int) { b.m.Lock() defer b.m.Unlock() b.b.Grow(n) } func (b *Buffer) Len() int { b.m.Lock() defer b.m.Unlock() return b.b.Len() } func (b *Buffer) Next(n int) []byte { b.m.Lock() defer b.m.Unlock() return b.b.Next(n) } func (b *Buffer) ReadByte() (c byte, err error) { b.m.Lock() defer b.m.Unlock() return b.b.ReadByte() } func (b *Buffer) ReadBytes(delim byte) (line []byte, err error) { b.m.Lock() defer b.m.Unlock() return b.b.ReadBytes(delim) } func (b *Buffer) ReadFrom(r io.Reader) (n int64, err error) { b.m.Lock() defer b.m.Unlock() return b.b.ReadFrom(r) } func (b *Buffer) ReadRune() (r rune, size int, err error) { b.m.Lock() defer b.m.Unlock() return b.b.ReadRune() } func (b *Buffer) ReadString(delim byte) (line string, err error) { b.m.Lock() defer b.m.Unlock() return b.b.ReadString(delim) } func (b *Buffer) Reset() { b.m.Lock() defer b.m.Unlock() b.b.Reset() } func (b *Buffer) Truncate(n int) { b.m.Lock() defer b.m.Unlock() b.b.Truncate(n) } func (b *Buffer) UnreadByte() error { b.m.Lock() defer b.m.Unlock() return b.b.UnreadByte() } func (b *Buffer) UnreadRune() error { b.m.Lock() defer b.m.Unlock() return b.b.UnreadRune() } func (b *Buffer) WriteByte(c byte) error { b.m.Lock() defer b.m.Unlock() return b.b.WriteByte(c) } func (b *Buffer) WriteRune(r rune) (n int, err error) { b.m.Lock() defer b.m.Unlock() return b.b.WriteRune(r) } func (b *Buffer) WriteString(s string) (n int, err error) { b.m.Lock() defer b.m.Unlock() return b.b.WriteString(s) } func (b *Buffer) WriteTo(w io.Writer) (n int64, err error) { b.m.Lock() defer b.m.Unlock() return b.b.WriteTo(w) }