支持remote write和exemplar的prometheus服务

最近项目组在做Prometheus指标采集和告警，其中用到了Prometheus的exemplar特性，由于该特性比较新，当前支持该特性的存储有比较少。因此需要自行实现exemplar功能。

我在github上创建了一个代码库，内容如下：

功能支持列表

• 支持从kafka消费Prometheus指标数据，数据使用protobuf编码
• 支持Prometheus exemplar功能
• 支持exemplar的wal
• 支持remote write指标到存储

kafka消费端

本项目使用的是腾讯的cKafka

golang的kafka消费端需要用到github.com/confluentinc/confluent-kafka-go/kafka，使用该库之前需要安装librdkafka库，但不支持在Windows系统上安装librdkafka。安装步骤如下：

git clone https://github.com/edenhill/librdkafka.gitcd librdkafka./configuremakesudo make install

环境上运行时可以考虑将librdkafka库编译到镜像中。如使用Alpine镜像时执行apk add librdkafka-dev pkgconf安装即可。官方文档中有提到，如果使用Alpine Linux ，编译方式为：go build -tags musl ./...

Metrics的写入

只需将metrics使用proto.Marshal编码到promWR即可:

func (c *client) WriteRaw(
    ctx context.Context,
    promWR []byte,
    opts WriteOptions,
) (WriteResult, WriteError) {
    var result WriteResult
    encoded := snappy.Encode(nil, promWR)

    body := bytes.NewReader(encoded)
    req, err := http.NewRequest("POST", c.writeURL, body)
    if err != nil {
        return result, writeError{err: err}
    }

    req.Header.Set("Content-Type", "application/x-protobuf")
    req.Header.Set("Content-Encoding", "snappy")
    req.Header.Set("User-Agent", c.userAgent)
    req.Header.Set("X-Prometheus-Remote-Write-Version", "0.1.0")
    if opts.Headers != nil {
        for k, v := range opts.Headers {
            req.Header.Set(k, v)
        }
    }

    resp, err := c.httpClient.Do(req.WithContext(ctx))
    if err != nil {
        return result, writeError{err: err}
    }

    result.StatusCode = resp.StatusCode

    defer resp.Body.Close()

    if result.StatusCode/100 != 2 {
        writeErr := writeError{
            err:  fmt.Errorf("expected HTTP 200 status code: actual=%d", resp.StatusCode),
            code: result.StatusCode,
        }

        body, err := ioutil.ReadAll(resp.Body)
        if err != nil {
            writeErr.err = fmt.Errorf("%v, body_read_error=%s", writeErr.err, err)
            return result, writeErr
        }

        writeErr.err = fmt.Errorf("%v, body=%s", writeErr.err, body)
        return result, writeErr
    }

    return result, nil
}

metric的查询

使用victoriametrics时，强烈建议同时部署grafana，使用grafana中的Explore功能来查找metrics。victoriametrics的vmselect组件自带的UI很不方便。

镜像编译

如上所述，如果需要在需要Alpine Linux中进行编译，则需要在在Dockerfile中添加如下内容：

RUN apk add git && apk add librdkafka-dev pkgconf && apk add build-base && apk add alpine-sdk

由于上述lib的安装比较慢，为了加快安装，可以将安装了这些lib的镜像作为基础镜像。

FROM golang:1.16.8-alpine3.14 as build
WORKDIR /app
RUN apk add git && apk add librdkafka-dev pkgconf && apk add build-base && apk add alpine-sdk
ENV http_proxy= GO111MODULE=on GOPROXY=https://goproxy.cn,direct GOPRIVATE=*.weimob.com

COPY go.mod .
COPY go.sum .
COPY . .

RUN cd cmd/ && GOOS=linux go build -tags musl -o ../prometheusWriter main.go

CMD ["/app/prometheusWriter"]

FROM alpine:latest
WORKDIR /app
RUN sed -i s/dl-cdn.alpinelinux.org/mirrors.aliyun.com/g /etc/apk/repositories
RUN apk add tzdata
RUN ln -sf /usr/share/zoneinfo/Asia/Shanghai /etc/localtime && echo "Asia/Shanghai" > /etc/timezone
COPY --from=build /app/prometheusWriter /app/
RUN chmod +x /app/prometheusWriter
COPY config.json /app/config.json
CMD ["/app/prometheusWriter"]

支持Exemplar

Exemplar的数据结构比较简单，就是个ring buffer。

下面是使用curl命令进行查找的例子：

# curl  '127.0.0.1:8000' --header 'Content-Type: application/json' -d '{"start":"1632980302","end":"1632980402","query":"{testlabel11="test"}"}'