mirror of
https://github.com/featurebasedb/featurebase.git
synced 2026-10-06 19:07:50 +00:00
2bdf1f5081
* unifying idk and featurebase: first pass * resolved conflict with master for gitignore & dockerignore * deleted binaries that were accidentally pushed to git * combined gitlab jobs for idk & featurebase * run go fmt for idk * updated ssh env variable, and made docker password variable in gitlab env variables * fixed typo assigning variable name * trying to fix docker login error * trying a different solution for docker password * pass registry * fixed docker login * updated paths for idk * exclude idk tests from featurebase test run * fix vendor error * update certificates * grpc needs to be in version 1.38 genproto, which is imported by big query updates the grpc version to 1.47.0 grpc 1.47.0 causes etcd to deadlock when calling etcd.Close() the fix is to have a replace in go.mod to specify a specific grpc version * run go mod tidy * go mod * run go mod tidy * exclude bigquery since it is causing issues and undo grpc replace in go.mod * fix grpc version * fix formatting error * update formatting * attempt to fix formatting * update path for code coverage * update to use current branch binaries, not master * fix for building idk - path updates * udpate path for binaries * update job dependecies * update docker idk tests to use the current branch registry * update stages for jobs * updated job dependencies * not allow idk s3 dump to fail since it is a dependency for integration tests * update dependecy for idk tests * update paths for idk build and code coverage * download featurebase binary from s3 * pass branch name to all setup scripts * change to current branch instead of master * updated sonarcloud * sonarcloud fix and branch name fix * trying to speed up pipeline run time * update stage * branch name fix + sonar cloud * sonarcloud
55 lines
2 KiB
Go
55 lines
2 KiB
Go
package kafka_static
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/molecula/featurebase/v3/idk"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type Main struct {
|
|
idk.Main `flag:"!embed"`
|
|
KafkaHosts []string `help:"Comma separated list of host:port pairs for Kafka."`
|
|
Group string `help:"Kafka group."`
|
|
Topics []string `help:"Kafka topics to read from."`
|
|
Timeout time.Duration `help:"Time to wait for more records from Kafka before flushing a batch. 0 to disable."`
|
|
SkipOld bool `short:"" help:"Skip to the most recent Kafka message rather than starting at the beginning."`
|
|
Header string `help:"Path to the static schema, in JSON header format. May be a path on the local filesystem, or an S3 URI (requires setting --s3-region or environment variable AWS_REGION)."`
|
|
S3Region string `help:"S3 Region, optionally used when header is specified as an S3 URI. Alternatively, use environment variable AWS_REGION."`
|
|
AllowMissingFields bool `help:"Will proceed with ingest even if a field is missing from a record but specified in the JSON config file. Default false"`
|
|
|
|
KafkaTLS idk.TLSConfig
|
|
}
|
|
|
|
func NewMain() *Main {
|
|
m := &Main{
|
|
Main: *idk.NewMain(),
|
|
KafkaHosts: []string{"localhost:9092"},
|
|
Group: "defaultgroup",
|
|
Topics: []string{"defaulttopic"},
|
|
Timeout: time.Second,
|
|
}
|
|
m.OffsetMode = true
|
|
m.Main.Namespace = "ingester_kafka_static"
|
|
m.Main.Pprof = "" // don't initialize pprof until we actually use it in tests
|
|
m.NewSource = func() (idk.Source, error) {
|
|
source := NewSource()
|
|
source.Hosts = m.KafkaHosts
|
|
source.Group = m.Group
|
|
source.Topics = m.Topics
|
|
source.Log = m.Main.Log()
|
|
source.TLS = m.KafkaTLS
|
|
source.Timeout = m.Timeout
|
|
source.SkipOld = m.SkipOld
|
|
source.Header = m.Header
|
|
source.S3Region = m.S3Region
|
|
source.AllowMissingFields = m.AllowMissingFields
|
|
|
|
err := source.Open()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "opening source")
|
|
}
|
|
return source, nil
|
|
}
|
|
return m
|
|
}
|