feat: added optional profiling

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()
+}