changes in sockets and terminal development

This commit is contained in:
2025-04-21 14:59:25 +05:30
parent 305650925e
commit 86dcfa2a4a
11 changed files with 1357 additions and 9 deletions

View File

@@ -0,0 +1,209 @@
package handlers
import (
"encoding/json"
"log"
"net/http"
"sync"
"time"
"github.com/arnab-afk/monaco/internal/executor"
"github.com/arnab-afk/monaco/internal/models"
)
// Handler manages HTTP requests for code submissions
type Handler struct {
executionService *executor.ExecutionService
mu sync.Mutex
submissions map[string]*models.CodeSubmission
}
// NewHandler creates a new handler instance
func NewHandler() *Handler {
return &Handler{
executionService: executor.NewExecutionService(),
submissions: make(map[string]*models.CodeSubmission),
}
}
// SubmitHandler handles code submission requests
func (h *Handler) SubmitHandler(w http.ResponseWriter, r *http.Request) {
// Only allow POST method
if r.Method != http.MethodPost {
http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
return
}
// Parse the request body
var submission models.CodeSubmission
if err := json.NewDecoder(r.Body).Decode(&submission); err != nil {
http.Error(w, "Invalid request body: "+err.Error(), http.StatusBadRequest)
return
}
// Validate the submission
if submission.Code == "" {
http.Error(w, "Code is required", http.StatusBadRequest)
return
}
if submission.Language == "" {
http.Error(w, "Language is required", http.StatusBadRequest)
return
}
// Generate a unique ID for the submission
h.mu.Lock()
submission.ID = executor.GenerateUUID()
submission.Status = "pending"
h.submissions[submission.ID] = &submission
h.mu.Unlock()
// Execute the code in a goroutine
go h.executionService.ExecuteCode(&submission)
// Return the submission ID
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(http.StatusAccepted)
json.NewEncoder(w).Encode(map[string]string{"id": submission.ID})
}
// StatusHandler handles status check requests
func (h *Handler) StatusHandler(w http.ResponseWriter, r *http.Request) {
// Only allow GET method
if r.Method != http.MethodGet {
http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
return
}
// Get the submission ID from the query parameters
id := r.URL.Query().Get("id")
if id == "" {
http.Error(w, "ID is required", http.StatusBadRequest)
return
}
// Get the submission from the map
h.mu.Lock()
submission, exists := h.submissions[id]
h.mu.Unlock()
if !exists {
http.Error(w, "Submission not found", http.StatusNotFound)
return
}
// Return the submission status
response := map[string]interface{}{
"id": submission.ID,
"status": submission.Status,
}
// Add time information based on status
if !submission.QueuedAt.IsZero() {
response["queuedAt"] = submission.QueuedAt.Format(time.RFC3339)
}
if !submission.StartedAt.IsZero() {
response["startedAt"] = submission.StartedAt.Format(time.RFC3339)
}
if !submission.CompletedAt.IsZero() {
response["completedAt"] = submission.CompletedAt.Format(time.RFC3339)
}
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(response)
}
// ResultHandler handles result requests
func (h *Handler) ResultHandler(w http.ResponseWriter, r *http.Request) {
// Only allow GET method
if r.Method != http.MethodGet {
http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
return
}
// Get the submission ID from the query parameters
id := r.URL.Query().Get("id")
if id == "" {
http.Error(w, "ID is required", http.StatusBadRequest)
return
}
// Get the submission from the map
h.mu.Lock()
submission, exists := h.submissions[id]
h.mu.Unlock()
if !exists {
http.Error(w, "Submission not found", http.StatusNotFound)
return
}
// Return the submission result
response := map[string]interface{}{
"id": submission.ID,
"status": submission.Status,
"language": submission.Language,
"output": submission.Output,
}
// Add error information if available
if submission.Error != "" {
response["error"] = submission.Error
}
// Add time information
if !submission.QueuedAt.IsZero() {
response["queuedAt"] = submission.QueuedAt.Format(time.RFC3339)
}
if !submission.StartedAt.IsZero() {
response["startedAt"] = submission.StartedAt.Format(time.RFC3339)
}
if !submission.CompletedAt.IsZero() {
response["completedAt"] = submission.CompletedAt.Format(time.RFC3339)
if !submission.StartedAt.IsZero() {
response["executionTime"] = submission.CompletedAt.Sub(submission.StartedAt).Milliseconds()
}
}
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(response)
}
// QueueStatsHandler provides information about the job queue
func (h *Handler) QueueStatsHandler(w http.ResponseWriter, r *http.Request) {
// Only allow GET method
if r.Method != http.MethodGet {
http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
return
}
// Get the queue statistics
stats := h.executionService.GetQueueStats()
// Return the queue statistics
response := map[string]interface{}{
"queue_stats": stats,
"submissions": len(h.submissions),
}
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(response)
}
// HealthCheckHandler handles health check requests
func (h *Handler) HealthCheckHandler(w http.ResponseWriter, r *http.Request) {
// Only allow GET method
if r.Method != http.MethodGet {
http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
return
}
// Return a simple health check response
response := map[string]interface{}{
"status": "ok",
"timestamp": time.Now().Format(time.RFC3339),
}
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(response)
}

View File

@@ -0,0 +1,70 @@
package handlers
import (
"bytes"
"encoding/json"
"net/http"
"net/http/httptest"
"testing"
"github.com/stretchr/testify/assert"
)
func TestSubmitHandler(t *testing.T) {
h := NewHandler()
// Create a test request
reqBody := map[string]string{
"language": "python",
"code": "print('Hello, World!')",
"input": "",
}
reqJSON, _ := json.Marshal(reqBody)
req, err := http.NewRequest("POST", "/submit", bytes.NewBuffer(reqJSON))
if err != nil {
t.Fatal(err)
}
req.Header.Set("Content-Type", "application/json")
// Create a response recorder
rr := httptest.NewRecorder()
// Call the handler
h.SubmitHandler(rr, req)
// Check the status code
assert.Equal(t, http.StatusAccepted, rr.Code)
// Check the response body
var response map[string]string
err = json.Unmarshal(rr.Body.Bytes(), &response)
assert.NoError(t, err)
assert.Contains(t, response, "id")
assert.NotEmpty(t, response["id"])
}
func TestHealthCheckHandler(t *testing.T) {
h := NewHandler()
// Create a test request
req, err := http.NewRequest("GET", "/health", nil)
if err != nil {
t.Fatal(err)
}
// Create a response recorder
rr := httptest.NewRecorder()
// Call the handler
h.HealthCheckHandler(rr, req)
// Check the status code
assert.Equal(t, http.StatusOK, rr.Code)
// Check the response body
var response map[string]interface{}
err = json.Unmarshal(rr.Body.Bytes(), &response)
assert.NoError(t, err)
assert.Equal(t, "ok", response["status"])
assert.Contains(t, response, "timestamp")
}

View File

@@ -0,0 +1,49 @@
package handlers
import (
"log"
"net/http"
"time"
)
// LoggingMiddleware logs HTTP requests
func LoggingMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
startTime := time.Now()
log.Printf("[HTTP] %s %s %s", r.Method, r.URL.Path, r.RemoteAddr)
next.ServeHTTP(w, r)
log.Printf("[HTTP] %s %s completed in %v", r.Method, r.URL.Path, time.Since(startTime))
})
}
// CORSMiddleware adds CORS headers to responses
func CORSMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
// Set CORS headers
w.Header().Set("Access-Control-Allow-Origin", "*")
w.Header().Set("Access-Control-Allow-Methods", "GET, POST, OPTIONS")
w.Header().Set("Access-Control-Allow-Headers", "Content-Type, Authorization")
// Handle preflight requests
if r.Method == http.MethodOptions {
w.WriteHeader(http.StatusOK)
return
}
// Call the next handler
next.ServeHTTP(w, r)
})
}
// RecoveryMiddleware recovers from panics
func RecoveryMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
defer func() {
if err := recover(); err != nil {
log.Printf("[PANIC] %v", err)
http.Error(w, "Internal server error", http.StatusInternalServerError)
}
}()
next.ServeHTTP(w, r)
})
}

View File

@@ -0,0 +1,31 @@
package api
import (
"net/http"
"github.com/arnab-afk/monaco/internal/api/handlers"
)
// SetupRoutes configures all API routes
func SetupRoutes() http.Handler {
// Create a new handler
h := handlers.NewHandler()
// Create a new router
mux := http.NewServeMux()
// Apply middleware to all routes
var handler http.Handler = mux
handler = handlers.RecoveryMiddleware(handler)
handler = handlers.LoggingMiddleware(handler)
handler = handlers.CORSMiddleware(handler)
// Register routes
mux.HandleFunc("/submit", h.SubmitHandler)
mux.HandleFunc("/status", h.StatusHandler)
mux.HandleFunc("/result", h.ResultHandler)
mux.HandleFunc("/queue-stats", h.QueueStatsHandler)
mux.HandleFunc("/health", h.HealthCheckHandler)
return handler
}

View File

@@ -0,0 +1,637 @@
package executor
import (
"context"
"crypto/rand"
"encoding/hex"
"fmt"
"log"
"os"
"os/exec"
"path/filepath"
"strings"
"time"
"github.com/arnab-afk/monaco/internal/models"
"github.com/arnab-afk/monaco/internal/queue"
)
// ExecutionService manages code execution
type ExecutionService struct {
queue *queue.JobQueue
}
// CodeExecutionJob represents a code execution job
type CodeExecutionJob struct {
service *ExecutionService
submission *models.CodeSubmission
}
// NewExecutionService creates a new execution service
func NewExecutionService() *ExecutionService {
return &ExecutionService{
queue: queue.NewJobQueue(5), // 5 concurrent workers
}
}
// NewCodeExecutionJob creates a new code execution job
func NewCodeExecutionJob(service *ExecutionService, submission *models.CodeSubmission) *CodeExecutionJob {
return &CodeExecutionJob{
service: service,
submission: submission,
}
}
// Execute runs the code execution job
func (j *CodeExecutionJob) Execute() {
submission := j.submission
submission.Status = "running"
submission.StartedAt = time.Now()
log.Printf("[JOB-%s] Starting execution for language: %s", submission.ID, submission.Language)
j.service.executeLanguageSpecific(submission)
submission.CompletedAt = time.Now()
log.Printf("[JOB-%s] Execution completed in %v", submission.ID, submission.CompletedAt.Sub(submission.StartedAt))
}
// ExecuteCode adds the submission to the execution queue
func (s *ExecutionService) ExecuteCode(submission *models.CodeSubmission) {
submission.Status = "queued"
submission.QueuedAt = time.Now()
log.Printf("[SUBMISSION-%s] Code submission queued for language: %s", submission.ID, submission.Language)
// Create and add the job to the queue
job := NewCodeExecutionJob(s, submission)
s.queue.AddJob(job)
}
// executeLanguageSpecific executes code based on the language
func (s *ExecutionService) executeLanguageSpecific(submission *models.CodeSubmission) {
switch strings.ToLower(submission.Language) {
case "python":
s.executePython(submission)
case "javascript", "js":
s.executeJavaScript(submission)
case "go", "golang":
s.executeGo(submission)
case "java":
s.executeJava(submission)
case "c":
s.executeC(submission)
case "cpp", "c++":
s.executeCpp(submission)
default:
submission.Status = "failed"
submission.Error = fmt.Sprintf("Unsupported language: %s", submission.Language)
log.Printf("[EXEC-%s] ERROR: Unsupported language: %s", submission.ID, submission.Language)
}
}
// executePython runs Python code in a container
func (s *ExecutionService) executePython(submission *models.CodeSubmission) {
log.Printf("[PYTHON-%s] Preparing Python execution environment", submission.ID)
startTime := time.Now()
// Create a temporary file for the code
tempDir, err := os.MkdirTemp("", "monaco-python-*")
if err != nil {
submission.Status = "failed"
submission.Error = fmt.Sprintf("Failed to create temp directory: %v", err)
return
}
defer os.RemoveAll(tempDir)
// Write the code to a file
codePath := filepath.Join(tempDir, "code.py")
if err := os.WriteFile(codePath, []byte(submission.Code), 0644); err != nil {
submission.Status = "failed"
submission.Error = fmt.Sprintf("Failed to write code file: %v", err)
return
}
// Create a file for input if provided
inputPath := ""
if submission.Input != "" {
inputPath = filepath.Join(tempDir, "input.txt")
if err := os.WriteFile(inputPath, []byte(submission.Input), 0644); err != nil {
submission.Status = "failed"
submission.Error = fmt.Sprintf("Failed to write input file: %v", err)
return
}
}
// Run the code in a Docker container
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
var cmd *exec.Cmd
if inputPath != "" {
cmd = exec.CommandContext(ctx, "docker", "run", "--rm",
"--network=none", // No network access
"--memory=100m", // Memory limit
"--cpu-period=100000", // CPU quota period
"--cpu-quota=10000", // 10% CPU
"--ulimit", "nofile=64:64", // File descriptor limits
"-v", tempDir+":/code", // Mount code directory
"python:3.9",
"sh", "-c", "cat /code/input.txt | python /code/code.py")
} else {
cmd = exec.CommandContext(ctx, "docker", "run", "--rm",
"--network=none", // No network access
"--memory=100m", // Memory limit
"--cpu-period=100000", // CPU quota period
"--cpu-quota=10000", // 10% CPU
"--ulimit", "nofile=64:64", // File descriptor limits
"-v", tempDir+":/code", // Mount code directory
"python:3.9",
"python", "/code/code.py")
}
output, err := cmd.CombinedOutput()
elapsed := time.Since(startTime)
log.Printf("[PYTHON-%s] Python execution completed in %v", submission.ID, elapsed)
s.updateSubmissionResult(submission, output, err, ctx.Err() != nil)
}
// executeJavaScript runs JavaScript code in a container
func (s *ExecutionService) executeJavaScript(submission *models.CodeSubmission) {
log.Printf("[JS-%s] Preparing JavaScript execution environment", submission.ID)
startTime := time.Now()
// Create a temporary file for the code
tempDir, err := os.MkdirTemp("", "monaco-js-*")
if err != nil {
submission.Status = "failed"
submission.Error = fmt.Sprintf("Failed to create temp directory: %v", err)
return
}
defer os.RemoveAll(tempDir)
// Write the code to a file
codePath := filepath.Join(tempDir, "code.js")
if err := os.WriteFile(codePath, []byte(submission.Code), 0644); err != nil {
submission.Status = "failed"
submission.Error = fmt.Sprintf("Failed to write code file: %v", err)
return
}
// Create a file for input if provided
inputPath := ""
if submission.Input != "" {
inputPath = filepath.Join(tempDir, "input.txt")
if err := os.WriteFile(inputPath, []byte(submission.Input), 0644); err != nil {
submission.Status = "failed"
submission.Error = fmt.Sprintf("Failed to write input file: %v", err)
return
}
}
// Run the code in a Docker container
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
var cmd *exec.Cmd
if inputPath != "" {
// Create a wrapper script to handle input
wrapperPath := filepath.Join(tempDir, "wrapper.js")
wrapperCode := `
const fs = require('fs');
const input = fs.readFileSync('/code/input.txt', 'utf8');
// Redirect input to stdin
process.stdin.push(input);
process.stdin.push(null);
// Load and run the user code
require('./code.js');
`
if err := os.WriteFile(wrapperPath, []byte(wrapperCode), 0644); err != nil {
submission.Status = "failed"
submission.Error = fmt.Sprintf("Failed to write wrapper file: %v", err)
return
}
cmd = exec.CommandContext(ctx, "docker", "run", "--rm",
"--network=none", // No network access
"--memory=100m", // Memory limit
"--cpu-period=100000", // CPU quota period
"--cpu-quota=10000", // 10% CPU
"-v", tempDir+":/code", // Mount code directory
"node:18-alpine",
"node", "/code/wrapper.js")
} else {
cmd = exec.CommandContext(ctx, "docker", "run", "--rm",
"--network=none", // No network access
"--memory=100m", // Memory limit
"--cpu-period=100000", // CPU quota period
"--cpu-quota=10000", // 10% CPU
"-v", tempDir+":/code", // Mount code directory
"node:18-alpine",
"node", "/code/code.js")
}
output, err := cmd.CombinedOutput()
elapsed := time.Since(startTime)
log.Printf("[JS-%s] JavaScript execution completed in %v", submission.ID, elapsed)
s.updateSubmissionResult(submission, output, err, ctx.Err() != nil)
}
// executeGo runs Go code in a container
func (s *ExecutionService) executeGo(submission *models.CodeSubmission) {
log.Printf("[GO-%s] Preparing Go execution environment", submission.ID)
startTime := time.Now()
// Create a temporary file for the code
tempDir, err := os.MkdirTemp("", "monaco-go-*")
if err != nil {
submission.Status = "failed"
submission.Error = fmt.Sprintf("Failed to create temp directory: %v", err)
return
}
defer os.RemoveAll(tempDir)
// Write the code to a file
codePath := filepath.Join(tempDir, "main.go")
if err := os.WriteFile(codePath, []byte(submission.Code), 0644); err != nil {
submission.Status = "failed"
submission.Error = fmt.Sprintf("Failed to write code file: %v", err)
return
}
// Create a file for input if provided
inputPath := ""
if submission.Input != "" {
inputPath = filepath.Join(tempDir, "input.txt")
if err := os.WriteFile(inputPath, []byte(submission.Input), 0644); err != nil {
submission.Status = "failed"
submission.Error = fmt.Sprintf("Failed to write input file: %v", err)
return
}
}
// Run the code in a Docker container
ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
defer cancel()
// First compile the Go code
compileCmd := exec.CommandContext(ctx, "docker", "run", "--rm",
"-v", tempDir+":/code", // Mount code directory
"golang:1.22-alpine",
"go", "build", "-o", "/code/app", "/code/main.go")
compileOutput, compileErr := compileCmd.CombinedOutput()
if compileErr != nil {
log.Printf("[GO-%s] Compilation failed: %v", submission.ID, compileErr)
submission.Status = "failed"
submission.Error = fmt.Sprintf("Compilation error: %s", compileOutput)
return
}
// Then run the compiled binary
var runCmd *exec.Cmd
if inputPath != "" {
runCmd = exec.CommandContext(ctx, "docker", "run", "--rm",
"--network=none", // No network access
"--memory=100m", // Memory limit
"--cpu-period=100000", // CPU quota period
"--cpu-quota=10000", // 10% CPU
"-v", tempDir+":/code", // Mount code directory
"golang:1.22-alpine",
"sh", "-c", "cat /code/input.txt | /code/app")
} else {
runCmd = exec.CommandContext(ctx, "docker", "run", "--rm",
"--network=none", // No network access
"--memory=100m", // Memory limit
"--cpu-period=100000", // CPU quota period
"--cpu-quota=10000", // 10% CPU
"-v", tempDir+":/code", // Mount code directory
"golang:1.22-alpine",
"/code/app")
}
output, err := runCmd.CombinedOutput()
elapsed := time.Since(startTime)
log.Printf("[GO-%s] Go execution completed in %v", submission.ID, elapsed)
s.updateSubmissionResult(submission, output, err, ctx.Err() != nil)
}
// executeJava runs Java code in a container
func (s *ExecutionService) executeJava(submission *models.CodeSubmission) {
log.Printf("[JAVA-%s] Preparing Java execution environment", submission.ID)
startTime := time.Now()
// Create a temporary file for the code
tempDir, err := os.MkdirTemp("", "monaco-java-*")
if err != nil {
submission.Status = "failed"
submission.Error = fmt.Sprintf("Failed to create temp directory: %v", err)
return
}
defer os.RemoveAll(tempDir)
// Extract class name from the code
className := extractJavaClassName(submission.Code)
if className == "" {
className = "Main" // Default class name
}
// Write the code to a file
codePath := filepath.Join(tempDir, className+".java")
if err := os.WriteFile(codePath, []byte(submission.Code), 0644); err != nil {
submission.Status = "failed"
submission.Error = fmt.Sprintf("Failed to write code file: %v", err)
return
}
// Create a file for input if provided
inputPath := ""
if submission.Input != "" {
inputPath = filepath.Join(tempDir, "input.txt")
if err := os.WriteFile(inputPath, []byte(submission.Input), 0644); err != nil {
submission.Status = "failed"
submission.Error = fmt.Sprintf("Failed to write input file: %v", err)
return
}
}
// Run the code in a Docker container
ctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)
defer cancel()
// First compile the Java code
compileCmd := exec.CommandContext(ctx, "docker", "run", "--rm",
"-v", tempDir+":/code", // Mount code directory
"eclipse-temurin:11-jdk-alpine",
"javac", "/code/"+className+".java")
compileOutput, compileErr := compileCmd.CombinedOutput()
if compileErr != nil {
log.Printf("[JAVA-%s] Compilation failed: %v", submission.ID, compileErr)
submission.Status = "failed"
submission.Error = fmt.Sprintf("Compilation error: %s", compileOutput)
return
}
// Then run the compiled class
var runCmd *exec.Cmd
if inputPath != "" {
runCmd = exec.CommandContext(ctx, "docker", "run", "--rm",
"--network=none", // No network access
"--memory=400m", // Memory limit
"--cpu-period=100000", // CPU quota period
"--cpu-quota=50000", // 50% CPU
"-v", tempDir+":/code", // Mount code directory
"eclipse-temurin:11-jdk-alpine",
"sh", "-c", "cd /code && cat input.txt | java -XX:+TieredCompilation -XX:TieredStopAtLevel=1 -Xverify:none -Xms64m -Xmx256m "+className)
} else {
runCmd = exec.CommandContext(ctx, "docker", "run", "--rm",
"--network=none", // No network access
"--memory=400m", // Memory limit
"--cpu-period=100000", // CPU quota period
"--cpu-quota=50000", // 50% CPU
"-v", tempDir+":/code", // Mount code directory
"eclipse-temurin:11-jdk-alpine",
"java", "-XX:+TieredCompilation", "-XX:TieredStopAtLevel=1", "-Xverify:none", "-Xms64m", "-Xmx256m", "-cp", "/code", className)
}
output, err := runCmd.CombinedOutput()
elapsed := time.Since(startTime)
log.Printf("[JAVA-%s] Java execution completed in %v", submission.ID, elapsed)
s.updateSubmissionResult(submission, output, err, ctx.Err() != nil)
}
// executeC runs C code in a container
func (s *ExecutionService) executeC(submission *models.CodeSubmission) {
log.Printf("[C-%s] Preparing C execution environment", submission.ID)
startTime := time.Now()
// Create a temporary file for the code
tempDir, err := os.MkdirTemp("", "monaco-c-*")
if err != nil {
submission.Status = "failed"
submission.Error = fmt.Sprintf("Failed to create temp directory: %v", err)
return
}
defer os.RemoveAll(tempDir)
// Write the code to a file
codePath := filepath.Join(tempDir, "code.c")
if err := os.WriteFile(codePath, []byte(submission.Code), 0644); err != nil {
submission.Status = "failed"
submission.Error = fmt.Sprintf("Failed to write code file: %v", err)
return
}
// Create a file for input if provided
inputPath := ""
if submission.Input != "" {
inputPath = filepath.Join(tempDir, "input.txt")
if err := os.WriteFile(inputPath, []byte(submission.Input), 0644); err != nil {
submission.Status = "failed"
submission.Error = fmt.Sprintf("Failed to write input file: %v", err)
return
}
}
// Run the code in a Docker container
ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
defer cancel()
// First compile the C code
compileCmd := exec.CommandContext(ctx, "docker", "run", "--rm",
"-v", tempDir+":/code", // Mount code directory
"gcc:latest",
"gcc", "-o", "/code/app", "/code/code.c")
compileOutput, compileErr := compileCmd.CombinedOutput()
if compileErr != nil {
log.Printf("[C-%s] Compilation failed: %v", submission.ID, compileErr)
submission.Status = "failed"
submission.Error = fmt.Sprintf("Compilation error: %s", compileOutput)
return
}
// Then run the compiled binary
var runCmd *exec.Cmd
if inputPath != "" {
runCmd = exec.CommandContext(ctx, "docker", "run", "--rm",
"--network=none", // No network access
"--memory=100m", // Memory limit
"--cpu-period=100000", // CPU quota period
"--cpu-quota=10000", // 10% CPU
"-v", tempDir+":/code", // Mount code directory
"gcc:latest",
"sh", "-c", "cat /code/input.txt | /code/app")
} else {
runCmd = exec.CommandContext(ctx, "docker", "run", "--rm",
"--network=none", // No network access
"--memory=100m", // Memory limit
"--cpu-period=100000", // CPU quota period
"--cpu-quota=10000", // 10% CPU
"-v", tempDir+":/code", // Mount code directory
"gcc:latest",
"/code/app")
}
output, err := runCmd.CombinedOutput()
elapsed := time.Since(startTime)
log.Printf("[C-%s] C execution completed in %v", submission.ID, elapsed)
s.updateSubmissionResult(submission, output, err, ctx.Err() != nil)
}
// executeCpp runs C++ code in a container
func (s *ExecutionService) executeCpp(submission *models.CodeSubmission) {
log.Printf("[CPP-%s] Preparing C++ execution environment", submission.ID)
startTime := time.Now()
// Create a temporary file for the code
tempDir, err := os.MkdirTemp("", "monaco-cpp-*")
if err != nil {
submission.Status = "failed"
submission.Error = fmt.Sprintf("Failed to create temp directory: %v", err)
return
}
defer os.RemoveAll(tempDir)
// Write the code to a file
codePath := filepath.Join(tempDir, "code.cpp")
if err := os.WriteFile(codePath, []byte(submission.Code), 0644); err != nil {
submission.Status = "failed"
submission.Error = fmt.Sprintf("Failed to write code file: %v", err)
return
}
// Create a file for input if provided
inputPath := ""
if submission.Input != "" {
inputPath = filepath.Join(tempDir, "input.txt")
if err := os.WriteFile(inputPath, []byte(submission.Input), 0644); err != nil {
submission.Status = "failed"
submission.Error = fmt.Sprintf("Failed to write input file: %v", err)
return
}
}
// Run the code in a Docker container
ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
defer cancel()
// First compile the C++ code
compileCmd := exec.CommandContext(ctx, "docker", "run", "--rm",
"-v", tempDir+":/code", // Mount code directory
"gcc:latest",
"g++", "-o", "/code/app", "/code/code.cpp")
compileOutput, compileErr := compileCmd.CombinedOutput()
if compileErr != nil {
log.Printf("[CPP-%s] Compilation failed: %v", submission.ID, compileErr)
submission.Status = "failed"
submission.Error = fmt.Sprintf("Compilation error: %s", compileOutput)
return
}
// Then run the compiled binary
var runCmd *exec.Cmd
if inputPath != "" {
runCmd = exec.CommandContext(ctx, "docker", "run", "--rm",
"--network=none", // No network access
"--memory=100m", // Memory limit
"--cpu-period=100000", // CPU quota period
"--cpu-quota=10000", // 10% CPU
"-v", tempDir+":/code", // Mount code directory
"gcc:latest",
"sh", "-c", "cat /code/input.txt | /code/app")
} else {
runCmd = exec.CommandContext(ctx, "docker", "run", "--rm",
"--network=none", // No network access
"--memory=100m", // Memory limit
"--cpu-period=100000", // CPU quota period
"--cpu-quota=10000", // 10% CPU
"-v", tempDir+":/code", // Mount code directory
"gcc:latest",
"/code/app")
}
output, err := runCmd.CombinedOutput()
elapsed := time.Since(startTime)
log.Printf("[CPP-%s] C++ execution completed in %v", submission.ID, elapsed)
s.updateSubmissionResult(submission, output, err, ctx.Err() != nil)
}
// updateSubmissionResult updates the submission with the execution result
func (s *ExecutionService) updateSubmissionResult(submission *models.CodeSubmission, output []byte, err error, timedOut bool) {
// Format the output to include the input if provided
formattedOutput := ""
if submission.Input != "" {
// Only add input lines that were actually used
inputLines := strings.Split(submission.Input, "\n")
for _, line := range inputLines {
if line != "" {
// Don't add the input marker for empty lines
formattedOutput += "[Input] " + line + "\n"
}
}
}
// Add the actual output
rawOutput := string(output)
if timedOut {
submission.Status = "failed"
submission.Error = "Execution timed out"
submission.Output = formattedOutput + rawOutput
return
}
if err != nil {
submission.Status = "failed"
submission.Error = err.Error()
submission.Output = formattedOutput + rawOutput
return
}
submission.Status = "completed"
submission.Output = formattedOutput + rawOutput
}
// GetQueueStats returns statistics about the job queue
func (s *ExecutionService) GetQueueStats() models.QueueStats {
return s.queue.GetStats()
}
// GenerateUUID generates a unique ID for submissions
func GenerateUUID() string {
b := make([]byte, 16)
_, err := rand.Read(b)
if err != nil {
return fmt.Sprintf("%d", time.Now().UnixNano())
}
return hex.EncodeToString(b)
}
// extractJavaClassName extracts the class name from Java code
func extractJavaClassName(code string) string {
// Simple regex-like extraction
lines := strings.Split(code, "\n")
for _, line := range lines {
line = strings.TrimSpace(line)
if strings.HasPrefix(line, "public class ") {
parts := strings.Split(line, " ")
if len(parts) > 2 {
className := parts[2]
// Remove any { or implements/extends
className = strings.Split(className, "{")[0]
className = strings.Split(className, " ")[0]
return strings.TrimSpace(className)
}
}
}
return ""
}

View File

@@ -0,0 +1,34 @@
package models
import "time"
// CodeSubmission represents a code submission for execution
type CodeSubmission struct {
ID string `json:"id"`
Code string `json:"code"`
Language string `json:"language"`
Input string `json:"input"`
Status string `json:"status"` // "pending", "queued", "running", "completed", "failed"
QueuedAt time.Time `json:"queuedAt,omitempty"`
StartedAt time.Time `json:"startedAt,omitempty"`
CompletedAt time.Time `json:"completedAt,omitempty"`
Output string `json:"output,omitempty"`
Error string `json:"error,omitempty"`
}
// ExecutionResult represents the result of code execution
type ExecutionResult struct {
Output string `json:"output"`
Error string `json:"error"`
ExitCode int `json:"exitCode"`
ExecutionMS int64 `json:"executionMs"`
}
// QueueStats represents statistics about the job queue
type QueueStats struct {
QueueLength int `json:"queueLength"`
RunningJobs int `json:"runningJobs"`
CompletedJobs int `json:"completedJobs"`
FailedJobs int `json:"failedJobs"`
TotalProcessed int `json:"totalProcessed"`
}

View File

@@ -0,0 +1,112 @@
package queue
import (
"log"
"sync"
"time"
"github.com/arnab-afk/monaco/internal/models"
)
// Job represents a job to be executed
type Job interface {
Execute()
}
// JobQueue manages the execution of jobs
type JobQueue struct {
queue chan Job
wg sync.WaitGroup
mu sync.Mutex
runningJobs int
completedJobs int
failedJobs int
totalProcessed int
workerCount int
}
// NewJobQueue creates a new job queue with the specified number of workers
func NewJobQueue(workerCount int) *JobQueue {
q := &JobQueue{
queue: make(chan Job, 100), // Buffer size of 100 jobs
workerCount: workerCount,
}
// Start workers
for i := 0; i < workerCount; i++ {
q.wg.Add(1)
go q.worker(i)
}
return q
}
// worker processes jobs from the queue
func (q *JobQueue) worker(id int) {
defer q.wg.Done()
log.Printf("[WORKER-%d] Started", id)
for job := range q.queue {
// Update stats
q.mu.Lock()
q.runningJobs++
q.mu.Unlock()
// Execute the job
startTime := time.Now()
log.Printf("[WORKER-%d] Processing job", id)
// Execute the job and handle panics
func() {
defer func() {
if r := recover(); r != nil {
log.Printf("[WORKER-%d] Panic in job execution: %v", id, r)
q.mu.Lock()
q.failedJobs++
q.runningJobs--
q.totalProcessed++
q.mu.Unlock()
}
}()
job.Execute()
}()
// Update stats if no panic occurred
q.mu.Lock()
q.completedJobs++
q.runningJobs--
q.totalProcessed++
q.mu.Unlock()
log.Printf("[WORKER-%d] Job completed in %v", id, time.Since(startTime))
}
log.Printf("[WORKER-%d] Stopped", id)
}
// AddJob adds a job to the queue
func (q *JobQueue) AddJob(job Job) {
q.queue <- job
}
// GetStats returns statistics about the job queue
func (q *JobQueue) GetStats() models.QueueStats {
q.mu.Lock()
defer q.mu.Unlock()
return models.QueueStats{
QueueLength: len(q.queue),
RunningJobs: q.runningJobs,
CompletedJobs: q.completedJobs,
FailedJobs: q.failedJobs,
TotalProcessed: q.totalProcessed,
}
}
// Shutdown stops the job queue
func (q *JobQueue) Shutdown() {
close(q.queue)
q.wg.Wait()
}