readme for database + one-to-one relationships

Author: misha567889

db/db.go

@@ -13,6 +13,14 @@ func NewDb(config Config) (*gorm.DB, error) {
 		slog.Error("Can't open database", "dsn", config.Dsn, "err", err)
 		return nil, err
 	}
+	if err = db.AutoMigrate(&models.TestingResult{}); err != nil {
+		slog.Error("Can't migrate TestingResult", "err", err)
+		return nil, err
+	}
+	if err = db.AutoMigrate(&models.ProblemConfig{}); err != nil {
+		slog.Error("Can't migrate ProblemConfig", "err", err)
+		return nil, err
+	}
 	if err = db.AutoMigrate(&models.Problem{}); err != nil {
 		slog.Error("Can't migrate Problem", "err", err)
 		return nil, err

db/models/problem.go

@@ -4,5 +4,6 @@ import "gorm.io/gorm"
 
 type Problem struct {
 	gorm.Model
-	ProblemConfig
+	ProblemConfigId int
+	ProblemConfig   ProblemConfig
 }

db/models/problem_config.go

@@ -1,13 +1,16 @@
 package models
 
+import "gorm.io/gorm"
+
 type ProblemType int
 
 const (
-	ProblemType_ICPC ProblemType = iota
+	ProblemType_ICPC ProblemType = iota + 1
 	ProblemType_IOI
 )
 
 type ProblemConfig struct {
+	gorm.Model
 	ProblemType ProblemType
 	// everything we need here
 }

db/models/submission.go

@@ -4,8 +4,9 @@ import "gorm.io/gorm"
 
 type Submission struct {
 	gorm.Model
-	TestingResult
-	ContentId     uint64 // id of submission in the storage
-	TaskId        uint64
-	TaskVersionId uint64
+	ContentId       uint64 // id of submission in the storage
+	TaskId          uint64
+	TaskVersionId   uint64
+	TestingResultId int
+	TestingResult   TestingResult
 }

db/models/testing_result.go

@@ -1,11 +1,14 @@
 package models
 
-import "github.com/lib/pq"
+import (
+	"github.com/lib/pq"
+	"gorm.io/gorm"
+)
 
 type Verdict int
 
 const (
-	Verdict_OK int = iota
+	Verdict_OK Verdict = iota + 1
 	Verdict_WA
 	Verdict_TL
 	Verdict_ML
@@ -14,5 +17,6 @@ const (
 )
 
 type TestingResult struct {
+	gorm.Model
 	Verdicts pq.Int64Array `gorm:"type:int[]"`
 }

db/readme.md

@@ -0,0 +1,60 @@
+# Как пользоваться базой
+
+### Подключаемся
+
+```go
+conn, err := db.NewDb(db.Config{Dsn: "DSN_STRING"})
+if err != nil {...}
+```
+
+### Создание объекта
+
+```go
+config := &models.ProblemConfig{ProblemType: models.ProblemType_ICPC}
+result := conn.Create(&config)
+config.ID  // номер нового объекта
+result.Error  // если была ошибка
+result.RowsAffected  // сколько строк вставилосб
+```
+
+https://gorm.io/docs/create.html
+
+### Получение объекта (обычно одного)
+
+```go
+var problem Problem
+var problems []Problem
+err := conn.First(&problem)
+// теперь либо err.Error != nil, либо в problem лежит запись с минимальным id
+conn.Take(&problem)  // рандомная запись
+conn.Last(&problem)  // последняя запись
+errors.Is(result.Error, gorm.ErrRecordNotFound) // пример ошибки
+
+result := map[string]interface{}{}
+conn.Model(&models.Problem{}).First(&result)  // после этого можно обращаться как к мапе
+
+conn.Find(&problem, 1)  // выбрать с id == 1
+conn.Find(&problem, "id = ?", 1)
+conn.Find(&problems, []int{1, 2, 3})  // выбрать с id \in {1, 2, 3}
+conn.Find(&problems)  // все объекты
+```
+
+https://gorm.io/docs/query.html
+
+### Продвинутые штуки
+
+Если просто вытащить задачу, то не вытащится ее конфиг. Для этого надо сделать join:
+
+``conn.Joins("ProblemConfig").Find(&problems)``
+
+Вообще gorm это какой-то полный пиздец. Например, чтобы вытащить все ICPC задачи, надо сделать так:
+
+```go
+conn.
+    Model(&models.Problem{}).
+    InnerJoins("ProblemConfig", conn.Where(&models.ProblemConfig{ProblemType: models.ProblemType_ICPC})).
+    Find(&problems)
+```
+
+Выглядит легко, но это одна из самых простых вещей...)
+Все равно придется писать запросы (наполовину) руками и молиться, чтобы оно работало