Skip to content

Commit

Permalink
code refactor
Browse files Browse the repository at this point in the history
  • Loading branch information
UlricQin committed Nov 21, 2023
1 parent 905b3a3 commit 5d66642
Show file tree
Hide file tree
Showing 7 changed files with 14 additions and 36 deletions.
6 changes: 3 additions & 3 deletions httpd/httpd.go
Original file line number Diff line number Diff line change
Expand Up @@ -128,14 +128,14 @@ func (r *HTTPRouter) Start() func() error {
if flags.HTTPTLSEnable {
tc, err := httptls.GetServerTLSConfig(flags.HTTPTLSCertFile, flags.HTTPTLSKeyFile, flags.HTTPTLSMinVersion, flags.HTTPTLSCipherSuitesArray)
if err != nil {
logger.Fatalf("failed to get TLS config for http server: %s", err)
logger.Fatalf("cannot get TLS config for http server: %s", err)
}
tlsConfig = tc
}

listner, err := net.Listen("tcp", flags.HTTPListen)
if err != nil {
logger.Fatalf("failed to listen %q: %s", flags.HTTPListen, err)
logger.Fatalf("cannot listen %q: %s", flags.HTTPListen, err)
}

if tlsConfig != nil {
Expand All @@ -147,7 +147,7 @@ func (r *HTTPRouter) Start() func() error {
// The server gracefully closed.
return
}
logger.Fatalf("failed to serve http at %s: %s", flags.HTTPListen, err)
logger.Fatalf("cannot serve http at %s: %s", flags.HTTPListen, err)
}
}()

Expand Down
4 changes: 2 additions & 2 deletions lib/encoding/zstd/zstd_pure.go
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ func init() {
var err error
decoder, err = zstd.NewReader(nil)
if err != nil {
logger.Panicf("BUG: failed to create ZSTD reader: %s", err)
logger.Panicf("BUG: cannot create ZSTD reader: %s", err)
}
}

Expand Down Expand Up @@ -74,7 +74,7 @@ func newEncoder(compressionLevel int) *zstd.Encoder {
zstd.WithEncoderCRC(false), // Disable CRC for performance reasons.
zstd.WithEncoderLevel(level))
if err != nil {
logger.Panicf("BUG: failed to create ZSTD writer: %s", err)
logger.Panicf("BUG: cannot create ZSTD writer: %s", err)
}
return e
}
4 changes: 2 additions & 2 deletions lib/promauth/config.go
Original file line number Diff line number Diff line change
Expand Up @@ -334,7 +334,7 @@ func (ac *Config) SetHeaders(req *http.Request, setAuthHeader bool) error {
if setAuthHeader {
ah, err := ac.GetAuthHeader()
if err != nil {
return fmt.Errorf("failed to obtain Authorization request header: %w", err)
return fmt.Errorf("cannot obtain Authorization request header: %w", err)
}
if ah != "" {
reqHeaders.Set("Authorization", ah)
Expand All @@ -352,7 +352,7 @@ func (ac *Config) SetFasthttpHeaders(req *fasthttp.Request, setAuthHeader bool)
if setAuthHeader {
ah, err := ac.GetAuthHeader()
if err != nil {
return fmt.Errorf("failed to obtain Authorization request header: %w", err)
return fmt.Errorf("cannot obtain Authorization request header: %w", err)
}
if ah != "" {
reqHeaders.Set("Authorization", ah)
Expand Down
2 changes: 1 addition & 1 deletion lib/runner/runner.go
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ func PrintRuntime() {

hostname, err := os.Hostname()
if err != nil {
hostname = fmt.Sprintf("failed to get hostname: %s", err)
hostname = fmt.Sprintf("cannot get hostname: %s", err)
}

logger.Infof("hostname: %s", hostname)
Expand Down
26 changes: 2 additions & 24 deletions main.go
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@ import (
"github.com/cprobe/cprobe/httpd"
"github.com/cprobe/cprobe/lib/buildinfo"
"github.com/cprobe/cprobe/lib/envflag"
"github.com/cprobe/cprobe/lib/fasttime"
"github.com/cprobe/cprobe/lib/flagutil"
"github.com/cprobe/cprobe/lib/logger"
"github.com/cprobe/cprobe/lib/procutil"
Expand All @@ -32,29 +31,8 @@ func main() {

writer.Init()

var vs []*writer.Vector
vs = append(vs, &writer.Vector{
Labels: map[string]string{
"__name__": "cprobe_test02",
"job": "cprobe01",
},
Clock: int64(fasttime.UnixTimestamp()) * 1000,
Value: 1.1,
})

vs = append(vs, &writer.Vector{
Labels: map[string]string{
"__name__": "cprobe_test02",
"job": "cprobe02",
},
Clock: int64(fasttime.UnixTimestamp()) * 1000,
Value: 1.1,
})

writer.WriteVectors(vs)

if err := probe.Start(ctx); err != nil {
logger.Fatalf("failed to start probe: %v", err)
logger.Fatalf("cannot start probe: %v", err)
}

// http server
Expand All @@ -65,7 +43,7 @@ func main() {
sig := procutil.WaitForSigterm()
logger.Infof("service received signal %s", sig)
if err := closeHTTP(); err != nil {
logger.Fatalf("failed to stop the webservice: %s", err)
logger.Fatalf("cannot stop the webservice: %s", err)
}

cancel()
Expand Down
4 changes: 2 additions & 2 deletions writer/input.go
Original file line number Diff line number Diff line change
Expand Up @@ -73,13 +73,13 @@ func (w *Writer) writeVectors(vs []*Vector) {

bs, err := req.Marshal()
if err != nil {
logger.Warnf("failed to marshal WriteRequest: %s", err)
logger.Warnf("cannot marshal WriteRequest: %s", err)
return
}

httpReq, err := w.NewRequest(snappy.Encode(nil, bs))
if err != nil {
logger.Warnf("failed to create http request: %s", err)
logger.Warnf("cannot create http request: %s", err)
return
}

Expand Down
4 changes: 2 additions & 2 deletions writer/writer.go
Original file line number Diff line number Diff line change
Expand Up @@ -185,11 +185,11 @@ func Init() error {

err := fileutil.ReadYaml(*writerFile, WriterConfig)
if err != nil {
return errors.Wrap(err, "failed to read writer config")
return errors.Wrap(err, "cannot read writer config")
}

if err = WriterConfig.Parse(); err != nil {
return errors.Wrap(err, "failed to set writer fields")
return errors.Wrap(err, "cannot set writer fields")
}

return nil
Expand Down

0 comments on commit 5d66642

Please sign in to comment.