close #88: now honoring stack memory option

mlnoga · Dec 31, 2023 · 181eff6 · 181eff6
1 parent 12a9c16
commit 181eff6
Show file tree

Hide file tree

Showing 4 changed files with 337 additions and 308 deletions.
diff --git a/cmd/nightlight/main.go b/cmd/nightlight/main.go
@@ -273,7 +273,7 @@ Flags:
 	}
 
 	// create context
-	c := ops.NewContext(logWriter, stats.LSEstimatorMode(*lsEst))
+	c := ops.NewContext(logWriter, int(*stMemory), stats.LSEstimatorMode(*lsEst))
 
 	// glob filename arguments into an opLoadMany operator
 	var err error
@@ -300,7 +300,7 @@ Flags:
 	switch args[0] {
 	case "serve":
 		rest.MakeSandbox(*chroot, int(*setuid))
-		rest.Serve(int(*port))
+		rest.Serve(int(*port), int(*stMemory))
 
 	case "stats":
 		opSeq := ops.NewOpSequence(opLoadMany, opPreProc)

diff --git a/internal/ops/operator.go b/internal/ops/operator.go
@@ -55,13 +55,13 @@ type Context struct {
 	StatsBufWriter *bufio.Writer `json:"-"` // buffered writer for the output file. use this
 }
 
-func NewContext(log io.Writer, lsEstimatorMode stats.LSEstimatorMode) *Context {
+func NewContext(log io.Writer, stMemory int, lsEstimatorMode stats.LSEstimatorMode) *Context {
 	memoryMB := int(memory.TotalMemory() / 1024 / 1024)
 	return &Context{
 		Log:             log,
 		LSEstimatorMode: lsEstimatorMode,
 		MemoryMB:        memoryMB,
-		StackMemoryMB:   memoryMB * 7 / 10,
+		StackMemoryMB:   stMemory,
 		MaxThreads:      runtime.GOMAXPROCS(0),
 	}
 }