70 lines
2.0 KiB
Go
70 lines
2.0 KiB
Go
|
// Copyright 2017 The Prometheus Authors
|
||
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
// you may not use this file except in compliance with the License.
|
||
|
|
// You may obtain a copy of the License at
|
||
|
|
//
|
||
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
//
|
||
|
|
// Unless required by applicable law or agreed to in writing, software
|
||
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
// See the License for the specific language governing permissions and
|
||
|
|
// limitations under the License.
|
||
|
|
|
||
|
|
//go:build !windows
|
||
|
|
|
||
|
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"io"
|
||
|
|
"math"
|
||
|
|
"os"
|
||
|
|
"path"
|
||
|
|
"syscall"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
|
||
|
|
"github.com/prometheus/prometheus/tsdb"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestTSDBDumpOpenMetricsRoundTripPipe(t *testing.T) {
|
||
|
|
initialMetrics, err := os.ReadFile("testdata/dump-openmetrics-roundtrip-test.prom")
|
||
|
|
require.NoError(t, err)
|
||
|
|
initialMetrics = normalizeNewLine(initialMetrics)
|
||
|
|
|
||
|
|
pipeDir := t.TempDir()
|
||
|
|
dbDir := t.TempDir()
|
||
|
|
|
||
|
|
// create pipe
|
||
|
|
pipe := path.Join(pipeDir, "pipe")
|
||
|
|
err = syscall.Mkfifo(pipe, 0o666)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
go func() {
|
||
|
|
// open pipe to write
|
||
|
|
in, err := os.OpenFile(pipe, os.O_WRONLY, os.ModeNamedPipe)
|
||
|
|
require.NoError(t, err)
|
||
|
|
defer func() { require.NoError(t, in.Close()) }()
|
||
|
|
_, err = io.Copy(in, bytes.NewReader(initialMetrics))
|
||
|
|
require.NoError(t, err)
|
||
|
|
}()
|
||
|
|
|
||
|
|
// Import samples from OM format
|
||
|
|
code := backfillOpenMetrics(pipe, dbDir, false, false, 2*time.Hour, map[string]string{})
|
||
|
|
require.Equal(t, 0, code)
|
||
|
|
db, err := tsdb.Open(dbDir, nil, nil, tsdb.DefaultOptions(), nil)
|
||
|
|
require.NoError(t, err)
|
||
|
|
t.Cleanup(func() {
|
||
|
|
require.NoError(t, db.Close())
|
||
|
|
})
|
||
|
|
|
||
|
|
// Dump the blocks into OM format
|
||
|
|
dumpedMetrics := getDumpedSamples(t, dbDir, "", math.MinInt64, math.MaxInt64, []string{"{__name__=~'(?s:.*)'}"}, formatSeriesSetOpenMetrics)
|
||
|
|
|
||
|
|
// Should get back the initial metrics.
|
||
|
|
require.Equal(t, string(initialMetrics), dumpedMetrics)
|
||
|
|
}
|