summaryrefslogtreecommitdiff
path: root/benchmarks/dcat_direct_benchmark_test.go
blob: 0d23749636dd44e4874cb08bebec1884787e3ac7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
package benchmarks

import (
	"bytes"
	"fmt"
	"os"
	"os/exec"
	"testing"
	"path/filepath"
)

// BenchmarkDCatDirect benchmarks dcat in direct/serverless mode
func BenchmarkDCatDirect(b *testing.B) {
	// Create test file if it doesn't exist
	testFile := filepath.Join(b.TempDir(), "benchmark_test.log")
	
	// Create a 100K line file for benchmarking
	var buf bytes.Buffer
	for i := 0; i < 100000; i++ {
		fmt.Fprintf(&buf, "2025-01-01 10:00:00 INFO [app] Processing request ID=%d status=OK latency=42ms user=user%d path=/api/v1/endpoint%d\n", 
			i, i%100, i%10)
	}
	
	if err := os.WriteFile(testFile, buf.Bytes(), 0644); err != nil {
		b.Fatal(err)
	}
	
	// Ensure dcat binary exists
	dcatPath := "../dcat"
	if _, err := os.Stat(dcatPath); err != nil {
		b.Skip("dcat binary not found, run 'make build' first")
	}
	
	b.Run("NonTurbo", func(b *testing.B) {
		os.Unsetenv("DTAIL_TURBOBOOST_ENABLE")
		b.ResetTimer()
		
		for i := 0; i < b.N; i++ {
			cmd := exec.Command(dcatPath, testFile)
			cmd.Stdout = os.NewFile(0, os.DevNull)
			if err := cmd.Run(); err != nil {
				b.Fatal(err)
			}
		}
	})
	
	b.Run("Turbo", func(b *testing.B) {
		os.Setenv("DTAIL_TURBOBOOST_ENABLE", "yes")
		defer os.Unsetenv("DTAIL_TURBOBOOST_ENABLE")
		b.ResetTimer()
		
		for i := 0; i < b.N; i++ {
			cmd := exec.Command(dcatPath, testFile)
			cmd.Stdout = os.NewFile(0, os.DevNull)
			if err := cmd.Run(); err != nil {
				b.Fatal(err)
			}
		}
	})
}

// BenchmarkDCatDirectWithSizes tests different file sizes
func BenchmarkDCatDirectWithSizes(b *testing.B) {
	sizes := []struct {
		name  string
		lines int
	}{
		{"1K", 1000},
		{"10K", 10000},
		{"100K", 100000},
	}
	
	dcatPath := "../dcat"
	if _, err := os.Stat(dcatPath); err != nil {
		b.Skip("dcat binary not found, run 'make build' first")
	}
	
	for _, size := range sizes {
		b.Run(size.name, func(b *testing.B) {
			// Create test file
			testFile := filepath.Join(b.TempDir(), fmt.Sprintf("benchmark_%s.log", size.name))
			var buf bytes.Buffer
			for i := 0; i < size.lines; i++ {
				fmt.Fprintf(&buf, "2025-01-01 10:00:00 INFO [app] Line %d data\n", i)
			}
			if err := os.WriteFile(testFile, buf.Bytes(), 0644); err != nil {
				b.Fatal(err)
			}
			
			b.Run("NonTurbo", func(b *testing.B) {
				os.Unsetenv("DTAIL_TURBOBOOST_ENABLE")
				b.ResetTimer()
				
				for i := 0; i < b.N; i++ {
					cmd := exec.Command(dcatPath, testFile)
					cmd.Stdout = os.NewFile(0, os.DevNull)
					if err := cmd.Run(); err != nil {
						b.Fatal(err)
					}
				}
			})
			
			b.Run("Turbo", func(b *testing.B) {
				os.Setenv("DTAIL_TURBOBOOST_ENABLE", "yes")
				defer os.Unsetenv("DTAIL_TURBOBOOST_ENABLE")
				b.ResetTimer()
				
				for i := 0; i < b.N; i++ {
					cmd := exec.Command(dcatPath, testFile)
					cmd.Stdout = os.NewFile(0, os.DevNull)
					if err := cmd.Run(); err != nil {
						b.Fatal(err)
					}
				}
			})
		})
	}
}