serverutils

func ReadMessage ¶ added in v0.13.4

func ReadMessage(ctx context.Context, msg protoreflect.ProtoMessage, r *bufio.Reader) error

ReadMessage reads protocol buffers from the given bufio.Reader.

func Serve ¶

func Serve(probeFunc func(*serverpb.ProbeRequest, *serverpb.ProbeReply))

Serve is similar to ServeContext but uses the background context.

func ServeContext ¶ added in v0.13.4

func ServeContext(probeFunc func(*serverpb.ProbeRequest, *serverpb.ProbeReply))

ServeContext blocks indefinitely, servicing probe requests. Note that this function is provided mainly to help external probe server implementations. Cloudprober doesn't make use of it. Example usage:

	import (
		serverpb "github.com/cloudprober/cloudprober/probes/external/proto"
		"github.com/cloudprober/cloudprober/probes/external/serverutils"
	)
	func runProbe(opts []*cppb.ProbeRequest_Option) {
 	...
	}
	serverutils.ServeContext(ctx, func(req *serverpb.ProbeRequest, reply *serverpb.ProbeReply) {
		payload, errMsg, _ := runProbe(req.GetOptions())
		reply.Payload = proto.String(payload)
		if errMsg != "" {
			reply.ErrorMessage = proto.String(errMsg)
		}
	})

func WriteMessage ¶

func WriteMessage(pb proto.Message, w io.Writer) error

WriteMessage marshals the a proto message and writes it to the writer "w" with appropriate Content-Length header.