mirror of
https://github.com/bolucat/Archive.git
synced 2025-12-24 13:28:37 +08:00
63 lines
1.6 KiB
Go
63 lines
1.6 KiB
Go
// Copyright (C) 2022 mieru authors
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
package common
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"time"
|
|
)
|
|
|
|
type SetReadDeadlineInterface interface {
|
|
SetReadDeadline(t time.Time) error
|
|
}
|
|
|
|
// SetReadTimeout set read deadline.
|
|
// It cancels the deadline if the timeout is 0 or negative.
|
|
func SetReadTimeout(conn SetReadDeadlineInterface, timeout time.Duration) {
|
|
if timeout > 0 {
|
|
conn.SetReadDeadline(time.Now().Add(timeout))
|
|
} else {
|
|
conn.SetReadDeadline(time.Time{})
|
|
}
|
|
}
|
|
|
|
// ReadAllAndDiscard reads from r until an error or EOF.
|
|
// All the data are discarded.
|
|
func ReadAllAndDiscard(r io.Reader) {
|
|
b := make([]byte, 1024)
|
|
for {
|
|
_, err := r.Read(b)
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
// RoundTrip sends a request to the connection and returns the response.
|
|
func RoundTrip(ctx context.Context, rw io.ReadWriter, req []byte, maxRespSize int) (resp []byte, err error) {
|
|
_, err = rw.Write(req)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
resp = make([]byte, maxRespSize)
|
|
var n int
|
|
n, err = rw.Read(resp)
|
|
resp = resp[:n]
|
|
return
|
|
}
|