Merge branch 'master' into wallet-store-interface-account

[bytom/vapor.git] / vendor / github.com / gin-contrib / sse / writer.go

diff --git a/vendor/github.com/gin-contrib/sse/writer.go b/vendor/github.com/gin-contrib/sse/writer.go
new file mode 100644 (file)
index 0000000..6f9806c
--- /dev/null
+++ b/vendor/github.com/gin-contrib/sse/writer.go
@@ -0,0 +1,24 @@
+package sse
+
+import "io"
+
+type stringWriter interface {
+       io.Writer
+       WriteString(string) (int, error)
+}
+
+type stringWrapper struct {
+       io.Writer
+}
+
+func (w stringWrapper) WriteString(str string) (int, error) {
+       return w.Writer.Write([]byte(str))
+}
+
+func checkWriter(writer io.Writer) stringWriter {
+       if w, ok := writer.(stringWriter); ok {
+               return w
+       } else {
+               return stringWrapper{writer}
+       }
+}