summaryrefslogtreecommitdiff
path: root/vendor/github.com/minio/minio-go/v7/pkg/peeker/peek-reader-closer.go
diff options
context:
space:
mode:
authorLibravatar kim <grufwub@gmail.com>2025-07-22 18:00:27 +0200
committerLibravatar kim <gruf@noreply.codeberg.org>2025-07-22 18:00:27 +0200
commitc00cad2cebcb8136a998f6f7ba2c27672f785d10 (patch)
tree863516d8459713cc4b91c83d8aeeef3cac486b39 /vendor/github.com/minio/minio-go/v7/pkg/peeker/peek-reader-closer.go
parent[chore/deps] Upgrade to go-sqlite 0.27.1 (#4334) (diff)
downloadgotosocial-c00cad2cebcb8136a998f6f7ba2c27672f785d10.tar.xz
[chore] bump dependencies (#4339)
- github.com/KimMachineGun/automemlimit v0.7.4 - github.com/miekg/dns v1.1.67 - github.com/minio/minio-go/v7 v7.0.95 - github.com/spf13/pflag v1.0.7 - github.com/tdewolff/minify/v2 v2.23.9 - github.com/uptrace/bun v1.2.15 - github.com/uptrace/bun/dialect/pgdialect v1.2.15 - github.com/uptrace/bun/dialect/sqlitedialect v1.2.15 - github.com/uptrace/bun/extra/bunotel v1.2.15 - golang.org/x/image v0.29.0 - golang.org/x/net v0.42.0 Reviewed-on: https://codeberg.org/superseriousbusiness/gotosocial/pulls/4339 Co-authored-by: kim <grufwub@gmail.com> Co-committed-by: kim <grufwub@gmail.com>
Diffstat (limited to 'vendor/github.com/minio/minio-go/v7/pkg/peeker/peek-reader-closer.go')
-rw-r--r--vendor/github.com/minio/minio-go/v7/pkg/peeker/peek-reader-closer.go73
1 files changed, 73 insertions, 0 deletions
diff --git a/vendor/github.com/minio/minio-go/v7/pkg/peeker/peek-reader-closer.go b/vendor/github.com/minio/minio-go/v7/pkg/peeker/peek-reader-closer.go
new file mode 100644
index 000000000..26c9cf637
--- /dev/null
+++ b/vendor/github.com/minio/minio-go/v7/pkg/peeker/peek-reader-closer.go
@@ -0,0 +1,73 @@
+/*
+ * MinIO Go Library for Amazon S3 Compatible Cloud Storage
+ * Copyright 2015-2025 MinIO, Inc.
+ *
+ * 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 peeker
+
+import (
+ "bytes"
+ "errors"
+ "io"
+)
+
+// PeekReadCloser offers a way to peek a ReadCloser stream and then
+// return the exact stream of the underlying ReadCloser
+type PeekReadCloser struct {
+ io.ReadCloser
+
+ recordMode bool
+ recordMaxBuf int
+ recordBuf *bytes.Buffer
+}
+
+// ReplayFromStart ensures next Read() will restart to stream the
+// underlying ReadCloser stream from the beginning
+func (prc *PeekReadCloser) ReplayFromStart() {
+ prc.recordMode = false
+}
+
+func (prc *PeekReadCloser) Read(p []byte) (int, error) {
+ if prc.recordMode {
+ if prc.recordBuf.Len() > prc.recordMaxBuf {
+ return 0, errors.New("maximum peek buffer exceeded")
+ }
+ n, err := prc.ReadCloser.Read(p)
+ prc.recordBuf.Write(p[:n])
+ return n, err
+ }
+ // Replay mode
+ if prc.recordBuf.Len() > 0 {
+ pn, _ := prc.recordBuf.Read(p)
+ return pn, nil
+ }
+ return prc.ReadCloser.Read(p)
+}
+
+// Close releases the record buffer memory and close the underlying ReadCloser
+func (prc *PeekReadCloser) Close() error {
+ prc.recordBuf.Reset()
+ return prc.ReadCloser.Close()
+}
+
+// NewPeekReadCloser returns a new peek reader
+func NewPeekReadCloser(rc io.ReadCloser, maxBufSize int) *PeekReadCloser {
+ return &PeekReadCloser{
+ ReadCloser: rc,
+ recordMode: true, // recording mode by default
+ recordBuf: bytes.NewBuffer(make([]byte, 0, 1024)),
+ recordMaxBuf: maxBufSize,
+ }
+}