feat: added optional profiling

Makefile

@@ -5,7 +5,7 @@ it:
 	@ chmod +x ${BINARY_NAME}
 
 simple: it
-	@ ./lambda.exe - < ./samples/simple.txt > program.out
+	@ ./lambda.exe -p profile/simple.prof - < ./samples/simple.txt > program.out
 
 thunk: it
 	@ ./lambda.exe - < ./samples/thunk.txt > program.out

cmd/lambda/lambda.go

@@ -3,11 +3,10 @@ package main
 import (
 	"fmt"
 	"os"
-	"runtime/pprof"
-	"time"
 
 	"git.maximhutz.com/max/lambda/internal/cli"
 	"git.maximhutz.com/max/lambda/internal/config"
+	"git.maximhutz.com/max/lambda/internal/executer"
 	"git.maximhutz.com/max/lambda/pkg/convert"
 	"git.maximhutz.com/max/lambda/pkg/lambda"
 	"git.maximhutz.com/max/lambda/pkg/saccharine"
@@ -15,14 +14,6 @@ import (
 
 // Run main application.
 func main() {
-	f, err := os.Create("profile/cpu.prof")
-	cli.HandleError(err)
-	defer f.Close()
-
-	err = pprof.StartCPUProfile(f)
-	cli.HandleError(err)
-	defer pprof.StopCPUProfile()
-
 	options, err := config.FromArgs()
 	cli.HandleError(err)
 
@@ -50,29 +41,10 @@ func main() {
 		logger.Info("compiled lambda expression", "tree", lambda.Stringify(compiled))
 	}
 
-	// Reduce expression.
-	start := time.Now()
-
-	if options.Explanation {
-		fmt.Println(lambda.Stringify(compiled))
-	}
-
-	steps := 0
-
-	for lambda.ReduceOnce(&compiled) {
-		if options.Verbose {
-			logger.Info("reduction", "tree", lambda.Stringify(compiled))
-		}
-		if options.Explanation {
-			fmt.Println("    =", lambda.Stringify(compiled))
-		}
-		steps++
-	}
-
-	elapsed := time.Since(start).Milliseconds()
+	executor := executer.New(options)
+	results, err := executor.Run(&compiled)
+	cli.HandleError(err)
 
 	fmt.Println(lambda.Stringify(compiled))
-	fmt.Fprintln(os.Stderr, "Time Spent:", elapsed, "ms")
-	fmt.Fprintln(os.Stderr, "Steps:", steps)
-	fmt.Fprintln(os.Stderr, "Speed:", float32(steps)/(float32(elapsed)/1000), "ops")
+	fmt.Fprint(os.Stderr, results.String())
 }

internal/cli/arguments.go

@@ -1,37 +0,0 @@
-package cli
-
-import (
-	"flag"
-	"fmt"
-)
-
-// Arguments given to program.
-type Options struct {
-	// The source code given to the program.
-	Input string
-	// Whether or not to print debug logs.
-	Verbose bool
-	// Whether or not to print an explanation of the reduction.
-	Explanation bool
-}
-
-// Extract the program configuration from the command-line arguments.
-func ParseOptions() (*Options, error) {
-	// Parse flags.
-	verbose := flag.Bool("v", false, "Verbosity. If set, the program will print logs.")
-	explanation := flag.Bool("x", false, "Explanation. Whether or not to show all reduction steps.")
-	flag.Parse()
-
-	// Parse non-flag arguments.
-	if flag.NArg() == 0 {
-		return nil, fmt.Errorf("no input given")
-	} else if flag.NArg() > 1 {
-		return nil, fmt.Errorf("more than 1 command-line argument")
-	}
-
-	return &Options{
-		Input:       flag.Arg(0),
-		Verbose:     *verbose,
-		Explanation: *explanation,
-	}, nil
-}

internal/config/config.go

@@ -2,10 +2,8 @@ package config
 
 // Arguments given to program.
 type Config struct {
-	// The source code given to the program.
-	Source Source
-	// Whether or not to print debug logs.
-	Verbose bool
-	// Whether or not to print an explanation of the reduction.
-	Explanation bool
+	Source      Source // The source code given to the program.
+	Verbose     bool   // Whether or not to print debug logs.
+	Explanation bool   // Whether or not to print an explanation of the reduction.
+	Profile     string // If not nil, print a CPU profile during execution.
 }

internal/config/parse_from_args.go

@@ -10,6 +10,7 @@ func FromArgs() (*Config, error) {
 	// Parse flags.
 	verbose := flag.Bool("v", false, "Verbosity. If set, the program will print logs.")
 	explanation := flag.Bool("x", false, "Explanation. Whether or not to show all reduction steps.")
+	profile := flag.String("p", "", "CPU profiling. If set, the program will run a performance profile during execution.")
 	flag.Parse()
 
 	// Parse non-flag arguments.
@@ -31,5 +32,6 @@ func FromArgs() (*Config, error) {
 		Source:      source,
 		Verbose:     *verbose,
 		Explanation: *explanation,
+		Profile:     *profile,
 	}, nil
 }

internal/executer/executer.go

@@ -0,0 +1,49 @@
+package executer
+
+import (
+	"fmt"
+	"log/slog"
+	"time"
+
+	"git.maximhutz.com/max/lambda/internal/config"
+	"git.maximhutz.com/max/lambda/pkg/lambda"
+)
+
+type Executor struct {
+	Config *config.Config
+}
+
+func New(config *config.Config) *Executor {
+	return &Executor{Config: config}
+}
+
+func (e Executor) Run(expr *lambda.Expression) (*Results, error) {
+	results := &Results{}
+
+	if e.Config.Profile != "" {
+		profiler := Profiler{File: e.Config.Profile}
+		if err := profiler.Start(); err != nil {
+			return nil, err
+		}
+		defer profiler.End()
+	}
+
+	start := time.Now()
+
+	if e.Config.Explanation {
+		fmt.Println(lambda.Stringify(*expr))
+	}
+
+	for lambda.ReduceOnce(expr) {
+		if e.Config.Verbose {
+			slog.Info("reduction", "tree", lambda.Stringify(*expr))
+		}
+		if e.Config.Explanation {
+			fmt.Println("    =", lambda.Stringify(*expr))
+		}
+		results.StepsTaken++
+	}
+
+	results.TimeElapsed = uint64(time.Since(start).Milliseconds())
+	return results, nil
+}

internal/executer/profiler.go

@@ -0,0 +1,40 @@
+package executer
+
+import (
+	"os"
+	"path/filepath"
+	"runtime/pprof"
+)
+
+type Profiler struct {
+	File        string
+	filePointer *os.File
+}
+
+func (p *Profiler) Start() error {
+	absPath, err := filepath.Abs(p.File)
+	if err != nil {
+		return err
+	}
+
+	err = os.MkdirAll(filepath.Dir(absPath), 0777)
+	if err != nil {
+		return err
+	}
+
+	p.filePointer, err = os.Create(absPath)
+	if err != nil {
+		return err
+	}
+
+	if err = pprof.StartCPUProfile(p.filePointer); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (p *Profiler) End() {
+	pprof.StopCPUProfile()
+	p.filePointer.Close()
+}

internal/executer/results.go

@@ -0,0 +1,23 @@
+package executer
+
+import (
+	"fmt"
+	"strings"
+)
+
+type Results struct {
+	StepsTaken  uint64 // Number of steps taken during execution.
+	TimeElapsed uint64 // The time (ms) taken for execution to complete.
+}
+
+func (r Results) OpsPerSecond() float32 {
+	return float32(r.StepsTaken) / (float32(r.TimeElapsed) / 1000)
+}
+
+func (r Results) String() string {
+	builder := strings.Builder{}
+	fmt.Fprintln(&builder, "Time Spent:", r.TimeElapsed, "ms")
+	fmt.Fprintln(&builder, "Steps:", r.StepsTaken)
+	fmt.Fprintln(&builder, "Speed:", r.OpsPerSecond(), "ops")
+	return builder.String()
+}