From 944c2c74c0756e6b768757475546da459d06e95e Mon Sep 17 00:00:00 2001
From: Kevin Chabowski <kevin@kch42.de>
Date: Tue, 6 Aug 2013 14:14:10 +0200
Subject: Initial commit.

---
 LimitedWriter.go | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 LimitedWriter.go

(limited to 'LimitedWriter.go')

diff --git a/LimitedWriter.go b/LimitedWriter.go
new file mode 100644
index 0000000..4d23322
--- /dev/null
+++ b/LimitedWriter.go
@@ -0,0 +1,39 @@
+package kagus
+
+import (
+	"errors"
+	"io"
+)
+
+var (
+	WriteLimitReached = errors.New("Write limit reached")
+)
+
+// LimitWriter wraps around a io.Writer and limits the amount of bytes that can be written to it.
+type LimitedWriter struct {
+	W io.Writer
+	N int64
+}
+
+// Write writes at most lw.N bytes to LimitWriter. It updates lw.N on sucessful write operations.
+// If the limit was reached, the error WriteLimitReached will be returned.
+func (lw *LimitedWriter) Write(p []byte) (int, error) {
+	towrite := int64(len(p))
+	if lw.N < towrite {
+		towrite = lw.N
+	}
+
+	n, err := lw.W.Write(p[:towrite])
+	if err != nil {
+		lw.N = 0
+		return n, err
+	}
+
+	if n == len(p) {
+		lw.N -= int64(n)
+		return n, nil
+	}
+
+	lw.N = 0
+	return n, WriteLimitReached
+}
-- 
cgit v1.2.3-54-g00ecf