55 lines
1.1 KiB
Go
55 lines
1.1 KiB
Go
|
package stat
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"encoding/json"
|
||
|
|
"errors"
|
||
|
|
"net/http"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"git.wishpal.cn/wishpal_ironfan/xframe/component/logger"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
httpTimeout = time.Second * 5
|
||
|
|
jsonContentType = "application/json; charset=utf-8"
|
||
|
|
)
|
||
|
|
|
||
|
|
// 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, jsonContentType, bytes.NewReader(bs))
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
defer resp.Body.Close()
|
||
|
|
|
||
|
|
if resp.StatusCode != http.StatusOK {
|
||
|
|
logger.Errorf("write report failed, code: %d, reason: %s", resp.StatusCode, resp.Status)
|
||
|
|
return ErrWriteFailed
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|