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.
52 lines
1013 B
Go
52 lines
1013 B
Go
package stat
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"errors"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
const httpTimeout = time.Second * 5
|
|
|
|
// ErrWriteFailed is an error that indicates failed to submit a StatReport.
|
|
var ErrWriteFailed = errors.New("submit failed")
|
|
|
|
// A RemoteWriter is a writer to write StatReport.
|
|
type RemoteWriter struct {
|
|
endpoint string
|
|
}
|
|
|
|
// NewRemoteWriter returns a RemoteWriter.
|
|
func NewRemoteWriter(endpoint string) Writer {
|
|
return &RemoteWriter{
|
|
endpoint: endpoint,
|
|
}
|
|
}
|
|
|
|
func (rw *RemoteWriter) Write(report *StatReport) error {
|
|
bs, err := json.Marshal(report)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
client := &http.Client{
|
|
Timeout: httpTimeout,
|
|
}
|
|
resp, err := client.Post(rw.endpoint, "application/json", bytes.NewBuffer(bs))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
logx.Errorf("write report failed, code: %d, reason: %s", resp.StatusCode, resp.Status)
|
|
return ErrWriteFailed
|
|
}
|
|
|
|
return nil
|
|
}
|