74 lines
1.7 KiB
Go
74 lines
1.7 KiB
Go
|
// Copyright (C) 2021 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"flag"
|
||
|
|
"log"
|
||
|
|
|
||
|
|
"go.uber.org/zap"
|
||
|
|
"go.uber.org/zap/zapcore"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
defaultSamplePercent = 1.0
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
cpuprofile = flag.String("cpuprofile", "", "write cpu profile to file")
|
||
|
|
|
||
|
|
samplePercent = flag.Float64("samplePercent", defaultSamplePercent, "sample size to verify in percents")
|
||
|
|
|
||
|
|
pointerdb = flag.String("pointerdb", "", "connection URL for PointerDB")
|
||
|
|
metabasedb = flag.String("metabasedb", "", "connection URL for MetabaseDB")
|
||
|
|
)
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
flag.Parse()
|
||
|
|
|
||
|
|
if *pointerdb == "" {
|
||
|
|
log.Fatalln("Flag '--pointerdb' is not set")
|
||
|
|
}
|
||
|
|
if *metabasedb == "" {
|
||
|
|
log.Fatalln("Flag '--metabasedb' is not set")
|
||
|
|
}
|
||
|
|
if *samplePercent < 0 || *samplePercent > 100 {
|
||
|
|
log.Fatalln("Flag '--samplePercent' can take values between 0 and 100")
|
||
|
|
}
|
||
|
|
|
||
|
|
ctx := context.Background()
|
||
|
|
log, err := zap.Config{
|
||
|
|
Encoding: "console",
|
||
|
|
Level: zap.NewAtomicLevelAt(zapcore.DebugLevel),
|
||
|
|
OutputPaths: []string{"stdout"},
|
||
|
|
ErrorOutputPaths: []string{"stdout"},
|
||
|
|
EncoderConfig: zapcore.EncoderConfig{
|
||
|
|
LevelKey: "L",
|
||
|
|
NameKey: "N",
|
||
|
|
CallerKey: "C",
|
||
|
|
MessageKey: "M",
|
||
|
|
StacktraceKey: "S",
|
||
|
|
LineEnding: zapcore.DefaultLineEnding,
|
||
|
|
EncodeLevel: zapcore.CapitalLevelEncoder,
|
||
|
|
EncodeTime: zapcore.ISO8601TimeEncoder,
|
||
|
|
EncodeDuration: zapcore.StringDurationEncoder,
|
||
|
|
EncodeCaller: zapcore.ShortCallerEncoder,
|
||
|
|
},
|
||
|
|
}.Build()
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
defer func() { _ = log.Sync() }()
|
||
|
|
|
||
|
|
config := Config{
|
||
|
|
SamplePercent: *samplePercent,
|
||
|
|
}
|
||
|
|
verifier := NewVerifier(log, *pointerdb, *metabasedb, config)
|
||
|
|
err = verifier.VerifyPointers(ctx)
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
}
|