From d1f72b758e9d97878b6dd21ca68762951c83b948 Mon Sep 17 00:00:00 2001
From: Paul Pan <i@0x7f.app>
Date: Sat, 6 Jan 2024 20:34:16 +0800
Subject: [PATCH] feat: also log user dir in status table

---
 internal/api/consumer/submitUpdate.go | 1 +
 internal/api/runner/judge.go          | 1 +
 internal/model/Status.go              | 1 +
 internal/model/Task.go                | 1 +
 internal/service/status/create.go     | 2 ++
 5 files changed, 6 insertions(+)

diff --git a/internal/api/consumer/submitUpdate.go b/internal/api/consumer/submitUpdate.go
index d016438..c54491a 100644
--- a/internal/api/consumer/submitUpdate.go
+++ b/internal/api/consumer/submitUpdate.go
@@ -24,6 +24,7 @@ func (h *handler) SubmitUpdate(_ context.Context, t *asynq.Task) error {
 	createData := &status.CreateData{
 		SubmissionID:     p.SubmissionID,
 		ProblemVersionID: p.ProblemVersionID,
+		UserDir:          p.UserDir,
 		Context:          p.Context,
 		Point:            p.Point,
 	}
diff --git a/internal/api/runner/judge.go b/internal/api/runner/judge.go
index 8ea1bb5..511d8a4 100644
--- a/internal/api/runner/judge.go
+++ b/internal/api/runner/judge.go
@@ -68,6 +68,7 @@ func (h *handler) Judge(_ context.Context, t *asynq.Task) error {
 		Status:           status,
 		SubmissionID:     p.Submission.ID,
 		ProblemVersionID: p.ProblemVersionID,
+		UserDir:          user,
 		Point:            point,
 	}, ctx)
 
diff --git a/internal/model/Status.go b/internal/model/Status.go
index 2e14cf1..ad1db38 100644
--- a/internal/model/Status.go
+++ b/internal/model/Status.go
@@ -10,6 +10,7 @@ type Status struct {
 	SubmissionID     uint        `json:"-"                  gorm:"not null;index"`
 	Submission       Submission  `json:"submission"         gorm:"foreignKey:SubmissionID"`
 	ProblemVersionID uint        `json:"problem_version_id" gorm:"not null;index"`
+	UserDir          string      `json:"user_dir"           gorm:"not null"`
 	Context          pgtype.JSON `json:"context"            gorm:"type:json;not null"`
 	Point            int32       `json:"point"              gorm:"not null"`
 	IsEnabled        bool        `json:"is_enabled"         gorm:"not null;index"`
diff --git a/internal/model/Task.go b/internal/model/Task.go
index 9b2cc05..9aea117 100644
--- a/internal/model/Task.go
+++ b/internal/model/Task.go
@@ -38,5 +38,6 @@ type SubmitUpdatePayload struct {
 	SubmissionID     uint
 	ProblemVersionID uint
 	Point            int32
+	UserDir          string
 	Context          string
 }
diff --git a/internal/service/status/create.go b/internal/service/status/create.go
index e933dba..6121df6 100644
--- a/internal/service/status/create.go
+++ b/internal/service/status/create.go
@@ -10,6 +10,7 @@ import (
 type CreateData struct {
 	SubmissionID     uint
 	ProblemVersionID uint
+	UserDir          string
 	Context          string
 	Point            int32
 }
@@ -18,6 +19,7 @@ func (s *service) Create(data *CreateData) (*model.Status, e.Status) {
 	status := &model.Status{
 		SubmissionID:     data.SubmissionID,
 		ProblemVersionID: data.ProblemVersionID,
+		UserDir:          data.UserDir,
 		Context: pgtype.JSON{
 			Bytes:  []byte(data.Context),
 			Status: pgtype.Present,