a447f55 · crush · git.iko.soy

git.iko.soy/crush

perf(config): simplify loadFromConfigPaths (#1821)

commita447f55c5c3091d0d1dd406ea9e96e97aaaa12ff authorCarlos Alexandro Becker <caarlos0@users.noreply.github.com> dateMon 12 Jan 2026, 10:24 (9 months ago) parent2350f7e

6 files changed+126−80

compare with main

unifiedsplit

internal/config/attribution_migration_test.go+1−3 internal/config/load.go+17−28 internal/config/load_bench_test.go+103 internal/config/load_test.go+5−6 internal/config/merge.go−16 internal/config/merge_test.go−27

internal/config/attribution_migration_test.go+1−3

@@ -1,8 +1,6 @@

11package config

33import (

4− "io"

5− "strings"

64 "testing"

86 "github.com/stretchr/testify/require"

@@ -83,7 +81,7 @@ func TestAttributionMigration(t *testing.T) {

8381 t.Run(tt.name, func(t *testing.T) {

8482 t.Parallel()

8583

86− cfg, err := loadFromReaders([]io.Reader{strings.NewReader(tt.configJSON)})

84+ cfg, err := loadFromBytes([][]byte{[]byte(tt.configJSON)})

8785 require.NoError(t, err)

8886

8987 cfg.setDefaults(t.TempDir(), "")

internal/config/load.go+17−28

@@ -5,7 +5,6 @@ import (

55 "context"

66 "encoding/json"

77 "fmt"

8− "io"

98 "log/slog"

109 "maps"

1110 "os"

@@ -25,25 +24,11 @@ import (

2524 "github.com/charmbracelet/crush/internal/home"

2625 "github.com/charmbracelet/crush/internal/log"

2726 powernapConfig "github.com/charmbracelet/x/powernap/pkg/config"

27+ "github.com/qjebbs/go-jsons"

2828)

2929

3030const defaultCatwalkURL = "https://catwalk.charm.sh"

3131

32−// LoadReader config via io.Reader.

33−func LoadReader(fd io.Reader) (*Config, error) {

34− data, err := io.ReadAll(fd)

35− if err != nil {

36− return nil, err

37− }

38−

39− var config Config

40− err = json.Unmarshal(data, &config)

41− if err != nil {

42− return nil, err

43− }

44− return &config, err

45−}

46−

4732// Load loads the configuration from the default paths.

4833func Load(workingDir, dataDir string, debug bool) (*Config, error) {

4934 configPaths := lookupConfigs(workingDir)

@@ -632,35 +617,39 @@ func lookupConfigs(cwd string) []string {

632617}

633618

634619func loadFromConfigPaths(configPaths []string) (*Config, error) {

635− var configs []io.Reader

620+ var configs [][]byte

636621

637622 for _, path := range configPaths {

638− fd, err := os.Open(path)

623+ data, err := os.ReadFile(path)

639624 if err != nil {

640625 if os.IsNotExist(err) {

641626 continue

642627 }

643628 return nil, fmt.Errorf("failed to open config file %s: %w", path, err)

644629 }

645− defer fd.Close()

646−

647− configs = append(configs, fd)

630+ if len(data) == 0 {

631+ continue

632+ }

633+ configs = append(configs, data)

648634 }

649635

650− return loadFromReaders(configs)

636+ return loadFromBytes(configs)

651637}

652638

653−func loadFromReaders(readers []io.Reader) (*Config, error) {

654− if len(readers) == 0 {

639+func loadFromBytes(configs [][]byte) (*Config, error) {

640+ if len(configs) == 0 {

655641 return &Config{}, nil

656642 }

657643

658− merged, err := Merge(readers)

644+ data, err := jsons.Merge(configs)

659645 if err != nil {

660− return nil, fmt.Errorf("failed to merge configuration readers: %w", err)

646+ return nil, err

661647 }

662−

663− return LoadReader(merged)

648+ var config Config

649+ if err := json.Unmarshal(data, &config); err != nil {

650+ return nil, err

651+ }

652+ return &config, nil

664653}

665654

666655func hasVertexCredentials(env env.Env) bool {

internal/config/load_bench_test.gonew file+103

@@ -0,0 +1,103 @@

1+package config

3+import (

4+ "os"

5+ "path/filepath"

6+ "testing"

7+)

9+func BenchmarkLoadFromConfigPaths(b *testing.B) {

10+ // Create temp config files with realistic content.

11+ tmpDir := b.TempDir()

12+

13+ globalConfig := filepath.Join(tmpDir, "global.json")

14+ localConfig := filepath.Join(tmpDir, "local.json")

15+

16+ globalContent := []byte(`{

17+ "providers": {

18+ "openai": {

19+ "api_key": "$OPENAI_API_KEY",

20+ "base_url": "https://api.openai.com/v1"

21+ },

22+ "anthropic": {

23+ "api_key": "$ANTHROPIC_API_KEY",

24+ "base_url": "https://api.anthropic.com"

25+ }

26+ },

27+ "options": {

28+ "tui": {

29+ "theme": "dark"

30+ }

31+ }

32+ }`)

33+

34+ localContent := []byte(`{

35+ "providers": {

36+ "openai": {

37+ "api_key": "sk-override-key"

38+ }

39+ },

40+ "options": {

41+ "context_paths": ["README.md", "AGENTS.md"]

42+ }

43+ }`)

44+

45+ if err := os.WriteFile(globalConfig, globalContent, 0o644); err != nil {

46+ b.Fatal(err)

47+ }

48+ if err := os.WriteFile(localConfig, localContent, 0o644); err != nil {

49+ b.Fatal(err)

50+ }

51+

52+ configPaths := []string{globalConfig, localConfig}

53+

54+ b.ReportAllocs()

55+ for b.Loop() {

56+ _, err := loadFromConfigPaths(configPaths)

57+ if err != nil {

58+ b.Fatal(err)

59+ }

60+ }

61+}

62+

63+func BenchmarkLoadFromConfigPaths_MissingFiles(b *testing.B) {

64+ // Test with mix of existing and non-existing paths.

65+ tmpDir := b.TempDir()

66+

67+ existingConfig := filepath.Join(tmpDir, "exists.json")

68+ content := []byte(`{"options": {"tui": {"theme": "dark"}}}`)

69+ if err := os.WriteFile(existingConfig, content, 0o644); err != nil {

70+ b.Fatal(err)

71+ }

72+

73+ configPaths := []string{

74+ filepath.Join(tmpDir, "nonexistent1.json"),

75+ existingConfig,

76+ filepath.Join(tmpDir, "nonexistent2.json"),

77+ }

78+

79+ b.ReportAllocs()

80+ for b.Loop() {

81+ _, err := loadFromConfigPaths(configPaths)

82+ if err != nil {

83+ b.Fatal(err)

84+ }

85+ }

86+}

87+

88+func BenchmarkLoadFromConfigPaths_Empty(b *testing.B) {

89+ // Test with no config files.

90+ tmpDir := b.TempDir()

91+ configPaths := []string{

92+ filepath.Join(tmpDir, "nonexistent1.json"),

93+ filepath.Join(tmpDir, "nonexistent2.json"),

94+ }

95+

96+ b.ReportAllocs()

97+ for b.Loop() {

98+ _, err := loadFromConfigPaths(configPaths)

99+ if err != nil {

100+ b.Fatal(err)

101+ }

102+ }

103+}

internal/config/load_test.go+5−6

@@ -5,7 +5,6 @@ import (

55 "log/slog"

66 "os"

77 "path/filepath"

8− "strings"

98 "testing"

109

1110 "github.com/charmbracelet/catwalk/pkg/catwalk"

@@ -22,12 +21,12 @@ func TestMain(m *testing.M) {

2221 os.Exit(exitVal)

2322}

2423

25−func TestConfig_LoadFromReaders(t *testing.T) {

26− data1 := strings.NewReader(`{"providers": {"openai": {"api_key": "key1", "base_url": "https://api.openai.com/v1"}}}`)

27− data2 := strings.NewReader(`{"providers": {"openai": {"api_key": "key2", "base_url": "https://api.openai.com/v2"}}}`)

28− data3 := strings.NewReader(`{"providers": {"openai": {}}}`)

24+func TestConfig_LoadFromBytes(t *testing.T) {

25+ data1 := []byte(`{"providers": {"openai": {"api_key": "key1", "base_url": "https://api.openai.com/v1"}}}`)

26+ data2 := []byte(`{"providers": {"openai": {"api_key": "key2", "base_url": "https://api.openai.com/v2"}}}`)

27+ data3 := []byte(`{"providers": {"openai": {}}}`)

2928

30− loadedConfig, err := loadFromReaders([]io.Reader{data1, data2, data3})

29+ loadedConfig, err := loadFromBytes([][]byte{data1, data2, data3})

3130

3231 require.NoError(t, err)

3332 require.NotNil(t, loadedConfig)

internal/config/merge.godeleted−16

@@ -1,16 +0,0 @@

1−package config

2−

3−import (

4− "bytes"

5− "io"

6−

7− "github.com/qjebbs/go-jsons"

8−)

9−

10−func Merge(data []io.Reader) (io.Reader, error) {

11− got, err := jsons.Merge(data)

12− if err != nil {

13− return nil, err

14− }

15− return bytes.NewReader(got), nil

16−}

internal/config/merge_test.godeleted−27

@@ -1,27 +0,0 @@

1−package config

2−

3−import (

4− "io"

5− "strings"

6− "testing"

7−)

8−

9−func TestMerge(t *testing.T) {

10− data1 := strings.NewReader(`{"foo": "bar"}`)

11− data2 := strings.NewReader(`{"baz": "qux"}`)

12−

13− merged, err := Merge([]io.Reader{data1, data2})

14− if err != nil {

15− t.Fatalf("expected no error, got %v", err)

16− }

17−

18− expected := `{"foo":"bar","baz":"qux"}`

19− got, err := io.ReadAll(merged)

20− if err != nil {

21− t.Fatalf("expected no error reading merged data, got %v", err)

22− }

23−

24− if string(got) != expected {

25− t.Errorf("expected %s, got %s", expected, string(got))

26− }

27−}