Skip to content

Commit

Permalink
Remove DB interface
Browse files Browse the repository at this point in the history
Remove the DB interface, as there will only be one implementation for it
  • Loading branch information
mjwolf committed Jan 18, 2024
1 parent f35b4c7 commit bf38e89
Show file tree
Hide file tree
Showing 7 changed files with 774 additions and 808 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ func init() {
type addSessionMetadata struct {
config Config
logger *logp.Logger
db processdb.DB
db *processdb.DB
provider provider.Provider
}

Expand All @@ -50,14 +50,14 @@ func New(cfg *config.C) (beat.Processor, error) {

ctx := context.TODO()
reader := procfs.NewProcfsReader(*logger)
db := processdb.NewSimpleDB(reader, *logger)
db := processdb.NewDB(reader, *logger)

backfilledPIDs := db.ScrapeProcfs()
logger.Debugf("backfilled %d processes", len(backfilledPIDs))

switch c.Backend {
case "ebpf":
p, err := ebpf_provider.NewProvider(ctx, *logger, db)
p, err := ebpf_provider.NewProvider(ctx, logger, db)
if err != nil {
return nil, fmt.Errorf("failed to create ebpf provider: %w", err)
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -156,7 +156,7 @@ var (
func TestEnrich(t *testing.T) {
for _, tt := range enrichTests {
reader := procfs.NewMockReader()
db := processdb.NewSimpleDB(reader, *logger)
db := processdb.NewDB(reader, *logger)

for _, ev := range tt.mockProcesses {
err := db.InsertExec(ev)
Expand Down
Loading

0 comments on commit bf38e89

Please sign in to comment.