Skip to content

Commit

Permalink
perf(linux): 🔥 remove unnecessary custom logger from mem worker
Browse files Browse the repository at this point in the history
  • Loading branch information
joshuar committed Sep 6, 2024
1 parent ca3ba6e commit 2588436
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 26 deletions.
12 changes: 3 additions & 9 deletions internal/linux/mem/memUsage.go
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@ package mem
import (
"context"
"fmt"
"log/slog"
"math"
"time"

Expand All @@ -18,7 +17,6 @@ import (
"github.com/joshuar/go-hass-agent/internal/hass/sensor"
"github.com/joshuar/go-hass-agent/internal/hass/sensor/types"
"github.com/joshuar/go-hass-agent/internal/linux"
"github.com/joshuar/go-hass-agent/internal/logging"
)

const (
Expand Down Expand Up @@ -114,9 +112,7 @@ func newSwapUsedPc(stats memoryStats) *linux.Sensor {
return newMemSensorPc("Swap Usage", swapUsed, swapTotal)
}

type usageWorker struct {
logger *slog.Logger
}
type usageWorker struct{}

func (w *usageWorker) Interval() time.Duration { return updateInterval }

Expand Down Expand Up @@ -147,11 +143,9 @@ func (w *usageWorker) Sensors(_ context.Context, _ time.Duration) ([]sensor.Deta
return sensors, nil
}

func NewUsageWorker(ctx context.Context) (*linux.SensorWorker, error) {
func NewUsageWorker(_ context.Context) (*linux.SensorWorker, error) {
return &linux.SensorWorker{
Value: &usageWorker{
logger: logging.FromContext(ctx).With(slog.String("worker", workerID)),
},
Value: &usageWorker{},
WorkerID: workerID,
},
nil
Expand Down
24 changes: 7 additions & 17 deletions internal/linux/mem/memUsage_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,6 @@ package mem

import (
"context"
"log/slog"
"math"
"testing"
"time"
Expand Down Expand Up @@ -154,45 +153,36 @@ func Test_usageWorker_Sensors(t *testing.T) {
}
withoutSwapSensors = append(withoutSwapSensors, newMemUsedPc(withoutSwap))

type fields struct {
logger *slog.Logger
}
type args struct {
in0 context.Context
file string
in1 time.Duration
}
tests := []struct {
name string
fields fields
args args
want []sensor.Details
wantErr bool
}{
{
name: "with swap",
fields: fields{logger: slog.Default()},
args: args{in0: linux.NewContext(context.TODO()), file: "testing/data/meminfowithswap"},
want: withSwapSensors,
name: "with swap",
args: args{in0: linux.NewContext(context.TODO()), file: "testing/data/meminfowithswap"},
want: withSwapSensors,
},
{
name: "without swap",
fields: fields{logger: slog.Default()},
args: args{in0: linux.NewContext(context.TODO()), file: "testing/data/meminfowithoutswap"},
want: withoutSwapSensors,
name: "without swap",
args: args{in0: linux.NewContext(context.TODO()), file: "testing/data/meminfowithoutswap"},
want: withoutSwapSensors,
},
{
name: "no stats file",
fields: fields{logger: slog.Default()},
args: args{in0: linux.NewContext(context.TODO()), file: "/nonexistent"},
wantErr: true,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
w := &usageWorker{
logger: tt.fields.logger,
}
w := &usageWorker{}
memStatFile = tt.args.file
got, err := w.Sensors(tt.args.in0, tt.args.in1)
if (err != nil) != tt.wantErr {
Expand Down

0 comments on commit 2588436

Please sign in to comment.