You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
962 B
Go
49 lines
962 B
Go
package io
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
)
|
|
|
|
// WriteWrapper warp io.Writer support more operate methods.
|
|
type WriteWrapper struct {
|
|
Out io.Writer
|
|
}
|
|
|
|
// NewWriteWrapper instance
|
|
func NewWriteWrapper(w io.Writer) *WriteWrapper {
|
|
return &WriteWrapper{Out: w}
|
|
}
|
|
|
|
// Write bytes data
|
|
func (w *WriteWrapper) Write(p []byte) (n int, err error) {
|
|
return w.Out.Write(p)
|
|
}
|
|
|
|
// Writef data to output
|
|
func (w *WriteWrapper) Writef(tpl string, vs ...any) (n int, err error) {
|
|
return fmt.Fprintf(w.Out, tpl, vs...)
|
|
}
|
|
|
|
// WriteByte data
|
|
func (w *WriteWrapper) WriteByte(c byte) error {
|
|
_, err := w.Out.Write([]byte{c})
|
|
return err
|
|
}
|
|
|
|
// WriteString data
|
|
func (w *WriteWrapper) WriteString(s string) (n int, err error) {
|
|
if sw, ok := w.Out.(io.StringWriter); ok {
|
|
return sw.WriteString(s)
|
|
}
|
|
return w.Out.Write([]byte(s))
|
|
}
|
|
|
|
// String get write data string
|
|
func (w *WriteWrapper) String() string {
|
|
if sw, ok := w.Out.(fmt.Stringer); ok {
|
|
return sw.String()
|
|
}
|
|
return ""
|
|
}
|