Add benchmark

Add Count() for GradeResultCache

Author: markhuang1212

backend/internal/core/cache.go

@@ -32,6 +32,12 @@ func (c *GradeResultCache) Add(key string, val types.GradeResult) {
 	}()
 }
 
+func (c *GradeResultCache) Count() int {
+	c.Lock.RLock()
+	defer c.Lock.RUnlock()
+	return len(c.Data)
+}
+
 func (c *GradeResultCache) Del(key string) {
 	c.Lock.Lock()
 	defer c.Lock.Unlock()

backend/internal/core/core_controller_test.go

@@ -1,6 +1,7 @@
 package core_test
 
 import (
+	"fmt"
 	"runtime"
 	"testing"
 
@@ -28,3 +29,21 @@ func TestCoreController(t *testing.T) {
 		runtime.Gosched()
 	}
 }
+
+func BenchmarkCoreController(b *testing.B) {
+	threads := 20
+	cc := core.NewCoreController(threads)
+	for i := 0; i < b.N; i++ {
+		cc.GradeQueue <- types.GradeRequest{
+			Id:           "test" + fmt.Sprint(i),
+			TestCaseName: "example-1",
+			UserCode:     "int main() { cout << \"Hello\" << endl; }",
+		}
+	}
+	for {
+		if cc.Cache.Count() == b.N {
+			return
+		}
+		runtime.Gosched()
+	}
+}

backend/internal/grader/compile_user_code.go

@@ -66,6 +66,7 @@ func CompileUserCode(ctx context.Context, gr types.GradeRequest, tmpDir string)
 		Resources: container.Resources{
 			Memory:     CompilationMemoryLimit,
 			MemorySwap: CompilationMemoryLimit,
+			CPUQuota:   10000,
 		},
 	}, nil, nil, "")