feat: support golang

internal/service/runner/problem_config.go

@@ -15,33 +15,38 @@ var (
 	}{
 		"c":       {""},
 		"cpp":     {""},
-		"rust":    {""},
+		"go":      {""},
 		"python3": {"/usr/bin/python3"},
 		"pypy3":   {"/usr/bin/pypy3"},
+		"rust":    {""},
 	}
 )
 
 type ConfigRuntime struct {
-	TimeLimit   int `json:"TimeLimit"`   // in ms
-	MemoryLimit int `json:"MemoryLimit"` // in mb
-	NProcLimit  int `json:"NProcLimit"`
+	TimeLimit      int `json:"TimeLimit"`   // in ms
+	MemoryLimit    int `json:"MemoryLimit"` // in mb
+	NProcLimit     int `json:"NProcLimit"`
+	WriteFileLimit int `json:"WriteFileLimit"` // in mb
 }
 
 var (
 	DefaultPrebuildRuntime = ConfigRuntime{
-		TimeLimit:   300000,
-		MemoryLimit: 256,
-		NProcLimit:  64,
+		TimeLimit:      300000,
+		MemoryLimit:    256,
+		NProcLimit:     64,
+		WriteFileLimit: 1,
 	}
 	DefaultCompileRuntime = ConfigRuntime{
-		TimeLimit:   60000,
-		MemoryLimit: 256,
-		NProcLimit:  64,
+		TimeLimit:      60000,
+		MemoryLimit:    256,
+		NProcLimit:     64,
+		WriteFileLimit: 64,
 	}
 	DefaultCheckRuntime = ConfigRuntime{
-		TimeLimit:   60000,
-		MemoryLimit: 128,
-		NProcLimit:  64,
+		TimeLimit:      60000,
+		MemoryLimit:    128,
+		NProcLimit:     64,
+		WriteFileLimit: 16,
 	}
 )
 
@@ -55,6 +60,9 @@ func (c *ConfigRuntime) Validate() error {
 	if c.NProcLimit <= 0 {
 		return errors.New("nproc limit <= 0")
 	}
+	if c.WriteFileLimit <= 0 {
+		return errors.New("write file limit <= 0")
+	}
 	return nil
 }
 

internal/service/runner/run_judge.go

@@ -25,6 +25,7 @@ func (s *service) SandboxArgsBuilder(meta *JudgeMeta, id int) string {
 	args = append(args, fmt.Sprintf("--memory_limit=%d", meta.Cfg.Lang.Runtime.Run.MemoryLimit))
 	args = append(args, fmt.Sprintf("--nproc_limit=%d", meta.Cfg.Lang.Runtime.Run.NProcLimit))
 	args = append(args, fmt.Sprintf("--time_limit=%d", meta.Cfg.Lang.Runtime.Run.TimeLimit))
+	args = append(args, fmt.Sprintf("--fsize_limit=%d", meta.Cfg.Lang.Runtime.Run.WriteFileLimit))
 	args = append(args, fmt.Sprintf("--sandbox_template=%s", meta.Cfg.Lang.Lang))
 	args = append(args, fmt.Sprintf("--sandbox_action=%s", "ret"))
 	args = append(args, fmt.Sprintf("--uid=%d", 1000))