summaryrefslogtreecommitdiff
path: root/teleirc/matterbridge/vendor/github.com/shazow/rateio/writer.go
diff options
context:
space:
mode:
authorMistivia <i@mistivia.com>2025-11-02 15:27:18 +0800
committerMistivia <i@mistivia.com>2025-11-02 15:27:18 +0800
commite9c24f4af7ed56760f6db7941827d09f6db9020b (patch)
tree62128c43b883ce5e3148113350978755779bb5de /teleirc/matterbridge/vendor/github.com/shazow/rateio/writer.go
parent58d5e7cfda4781d8a57ec52aefd02983835c301a (diff)
add matterbridge
Diffstat (limited to 'teleirc/matterbridge/vendor/github.com/shazow/rateio/writer.go')
-rw-r--r--teleirc/matterbridge/vendor/github.com/shazow/rateio/writer.go25
1 files changed, 25 insertions, 0 deletions
diff --git a/teleirc/matterbridge/vendor/github.com/shazow/rateio/writer.go b/teleirc/matterbridge/vendor/github.com/shazow/rateio/writer.go
new file mode 100644
index 0000000..d4feea4
--- /dev/null
+++ b/teleirc/matterbridge/vendor/github.com/shazow/rateio/writer.go
@@ -0,0 +1,25 @@
+package rateio
+
+import "io"
+
+type writer struct {
+ io.Writer
+ Limiter
+}
+
+// Write writes the contents of p into the buffer.
+// Returns ErrRateExceeded error if our specified read is exceeded.
+func (w *writer) Write(p []byte) (n int, err error) {
+ n, err = w.Writer.Write(p)
+ if err != nil {
+ return
+ }
+
+ err = w.Limiter.Count(n)
+ return
+}
+
+// NewWriter proxies an io.Writer but keeps track of bytes read based on our Limiter.
+func NewWriter(w io.Writer, limiter Limiter) io.Writer {
+ return &writer{w, limiter}
+}