vendor/github.com/gin-contrib/sse/writer.go

   1 package sse
   2
   3 import "io"
   4
   5 type stringWriter interface {
   6         io.Writer
   7         WriteString(string) (int, error)
   8 }
   9
  10 type stringWrapper struct {
  11         io.Writer
  12 }
  13
  14 func (w stringWrapper) WriteString(str string) (int, error) {
  15         return w.Writer.Write([]byte(str))
  16 }
  17
  18 func checkWriter(writer io.Writer) stringWriter {
  19         if w, ok := writer.(stringWriter); ok {
  20                 return w
  21         } else {
  22                 return stringWrapper{writer}
  23         }
  24 }