diff options
Diffstat (limited to 'backend/game/hub.go')
| -rw-r--r-- | backend/game/hub.go | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/backend/game/hub.go b/backend/game/hub.go index d19c12d..e08ed4a 100644 --- a/backend/game/hub.go +++ b/backend/game/hub.go @@ -6,6 +6,7 @@ import ( "errors" "fmt" "log" + "regexp" "strings" "time" @@ -86,7 +87,7 @@ func (hub *gameHub) run() { // TODO: assert game state is gaming log.Printf("submit: %v", message.message) code := msg.Data.Code - codeSize := len(code) // TODO: exclude whitespaces. + codeSize := calcCodeSize(code) codeHash := calcHash(code) if err := hub.taskQueue.EnqueueTaskCreateSubmissionRecord( hub.game.gameID, @@ -663,3 +664,8 @@ func isTestcaseResultCorrect(expectedStdout, actualStdout string) bool { func calcHash(code string) string { return fmt.Sprintf("%x", md5.Sum([]byte(code))) } + +func calcCodeSize(code string) int { + re := regexp.MustCompile(`\s+`) + return len(re.ReplaceAllString(code, "")) +} |