sing/common/rw/discard.go

package rw

import (
	"io"

	"github.com/sagernet/sing/common/buf"
)

var Discard io.Writer = discard{}

type discard struct{}

var _ io.ReaderFrom = discard{}

func (discard) Write(p []byte) (int, error) {
	return len(p), nil
}

func (discard) WriteString(s string) (int, error) {
	return len(s), nil
}

func (discard) ReadFrom(r io.Reader) (n int64, err error) {
	buffer := buf.Get(buf.BufferSize)
	readSize := 0
	for {
		readSize, err = r.Read(buffer)
		n += int64(readSize)
		if err != nil {
			buf.Put(buffer)
			if err == io.EOF {
				return n, nil
			}
			return
		}
	}
}