d

Author: markhuang1212

backend/.vscode/settings.json

@@ -0,0 +1,7 @@
+{
+    "go.testTimeout": "60s",
+    "cSpell.words": [
+        "cout",
+        "endl"
+    ]
+}

backend/Dockerfile

@@ -13,6 +13,7 @@ COPY . .
 WORKDIR /code-grader/backend
 RUN go build
 
+VOLUME [ "/var/run/docker.sock" ]
 EXPOSE 8080
 ENV GIN_MODE=release
 ENV APP_ROOT=/code-grader

backend/internal/api/setup_router.go

@@ -4,9 +4,12 @@ import (
 	"net/http"
 
 	"github.com/gin-gonic/gin"
+	"github.com/markhuang1212/code-grader/backend/internal/core"
+	"github.com/markhuang1212/code-grader/backend/internal/types"
+	"github.com/markhuang1212/code-grader/backend/internal/util"
 )
 
-func SetupRouter() *gin.Engine {
+func SetupRouter(cc *core.CoreController) *gin.Engine {
 
 	r := gin.Default()
 
@@ -18,6 +21,20 @@ func SetupRouter() *gin.Engine {
 
 	authorized.POST("/grade", func(c *gin.Context) {
 
+		gr := types.GradeRequest{}
+
+		err := c.BindJSON(&gr)
+		if err != nil {
+			return
+		}
+
+		gr.Id = util.RandomHex(10)
+
+		cc.GradeQueue <- gr
+
+		c.Header("Location", "/api/result/"+gr.Id)
+		c.Status(202)
+
 	})
 
 	return r

backend/internal/api/setup_router_test.go

@@ -6,11 +6,12 @@ import (
 	"testing"
 
 	"github.com/markhuang1212/code-grader/backend/internal/api"
+	"github.com/markhuang1212/code-grader/backend/internal/core"
 	"github.com/stretchr/testify/assert"
 )
 
 func TestPingRoute(t *testing.T) {
-	router := api.SetupRouter()
+	router := api.SetupRouter(core.NewCoreController(1))
 
 	w := httptest.NewRecorder()
 	req, _ := http.NewRequest("GET", "/ping", nil)

backend/internal/core/cache.go

@@ -0,0 +1,39 @@
+package core
+
+import (
+	"sync"
+
+	"github.com/markhuang1212/code-grader/backend/internal/types"
+)
+
+type GradeResultCache struct {
+	Data map[string]types.GradeResult
+	Lock sync.RWMutex
+}
+
+func NewGradeResultCache() *GradeResultCache {
+	ret := GradeResultCache{
+		Data: make(map[string]types.GradeResult),
+	}
+
+	return &ret
+}
+
+func (c *GradeResultCache) Add(key string, val types.GradeResult) {
+	c.Lock.Lock()
+	defer c.Lock.Unlock()
+	c.Data[key] = val
+}
+
+func (c *GradeResultCache) Del(key string) {
+	c.Lock.Lock()
+	defer c.Lock.Unlock()
+	delete(c.Data, key)
+}
+
+func (c *GradeResultCache) Get(key string) (types.GradeResult, bool) {
+	c.Lock.RLock()
+	defer c.Lock.RUnlock()
+	val, ok := c.Data[key]
+	return val, ok
+}

backend/internal/core/cache_test.go

@@ -0,0 +1,20 @@
+package core_test
+
+import (
+	"testing"
+
+	"github.com/markhuang1212/code-grader/backend/internal/core"
+	"github.com/markhuang1212/code-grader/backend/internal/types"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestCache(t *testing.T) {
+	c := core.NewGradeResultCache()
+	c.Add("id1", types.GradeResult{
+		Status: types.GradeResultCompilationError,
+		Msg:    "Hello",
+	})
+	ret, ok := c.Get("id1")
+	assert.True(t, ok)
+	assert.Equal(t, "Hello", ret.Msg)
+}

backend/internal/core/core_controller.go

@@ -0,0 +1,49 @@
+package core
+
+import (
+	"context"
+
+	"github.com/markhuang1212/code-grader/backend/internal/grader"
+	"github.com/markhuang1212/code-grader/backend/internal/types"
+)
+
+type CoreController struct {
+	Cache      *GradeResultCache
+	GradeQueue chan types.GradeRequest
+}
+
+func (cc *CoreController) handleRequest() {
+	for {
+		request, ok := <-cc.GradeQueue
+
+		if !ok {
+			break
+		}
+
+		ctx := context.Background()
+		result, err := grader.GradeUserCode(ctx, request)
+
+		if err != nil {
+			cc.Cache.Add(request.Id, types.GradeResult{
+				Status: types.GradeResultInternalError,
+				Msg:    "internal error",
+			})
+		} else {
+			cc.Cache.Add(request.Id, *result)
+		}
+	}
+}
+
+func NewCoreController(concurrent int) *CoreController {
+
+	ret := CoreController{
+		Cache:      NewGradeResultCache(),
+		GradeQueue: make(chan types.GradeRequest, 1024),
+	}
+
+	for i := 0; i < concurrent; i++ {
+		go ret.handleRequest()
+	}
+
+	return &ret
+}

backend/internal/core/core_controller_test.go

@@ -0,0 +1,30 @@
+package core_test
+
+import (
+	"runtime"
+	"testing"
+
+	"github.com/markhuang1212/code-grader/backend/internal/core"
+	"github.com/markhuang1212/code-grader/backend/internal/types"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestCoreController(t *testing.T) {
+
+	cc := core.NewCoreController(1)
+
+	cc.GradeQueue <- types.GradeRequest{
+		Id:           "test1",
+		TestCaseName: "example-1",
+		UserCode:     "int main() { cout << \"Hello\" << endl; }",
+	}
+
+	for {
+		val, ok := cc.Cache.Get("test1")
+		if ok == true {
+			assert.Equal(t, types.GradeResultSuccess, val.Status)
+			return
+		}
+		runtime.Gosched()
+	}
+}

backend/internal/grader/exec_user_code_test.go

@@ -11,6 +11,7 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
+// success
 func TestExecUserCode1(t *testing.T) {
 
 	ctx := context.Background()
@@ -37,6 +38,7 @@ func TestExecUserCode1(t *testing.T) {
 
 }
 
+// execution error
 func TestExecUserCode2(t *testing.T) {
 
 	ctx := context.Background()
@@ -64,6 +66,7 @@ func TestExecUserCode2(t *testing.T) {
 
 }
 
+// memory limit exceed
 func TestExecUserCode3(t *testing.T) {
 
 	ctx := context.Background()

backend/internal/types/GradeRequest.go

@@ -1,6 +1,7 @@
 package types
 
 type GradeRequest struct {
+	Id           string
 	TestCaseName string
 	UserCode     string
 }