init

Signed-off-by: storyicon <yuanchao@bilibili.com>

pkg/util/concurrent/buffer.go

@@ -0,0 +1,38 @@
+// Copyright 2021 storyicon@foxmail.com
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// 	http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package concurrent
+
+import (
+	"io"
+	"sync"
+)
+
+// Buffer is goroutine safety buffer
+type Buffer struct {
+	writer io.Writer
+	mutex  sync.Mutex
+}
+
+// Write implements the io.Writer
+func (s *Buffer) Write(p []byte) (n int, err error) {
+	s.mutex.Lock()
+	defer s.mutex.Unlock()
+	return s.writer.Write(p)
+}
+
+// NewBuffer is used to create a new buffer
+func NewBuffer(writer io.Writer) io.Writer {
+	return &Buffer{writer: writer}
+}