Доступ ко многим данным многих таблиц - PullRequest
Новая
       25

Доступ ко многим данным многих таблиц

4 голосов
/ 04 октября 2019

Я хочу работать с таблицей many2many, автоматически сгенерированной GORM db.AutoMigrate(&Document{}, &Folder{}).

. В этом сценарии у папки есть документы, а у документов есть папки, пока это хорошо, но у ассоциации естьстатус (действительный, недействительный ...), поэтому мне нужно посмотреть на таблицу documents_folders, чтобы узнать состояние.

База данных: 

+-----------------------+
| Table                 |
+-----------------------+
| documents             |
| documents_folders     |
| folders               |
+-----------------------+

Модели:

Материал

(идентификатор primary_key == UUID) 

type Stuff struct {
    ID          uuid.UUID  `gorm:"type:char(36);primary_key;" json:"id"`
    CreatedAt   time.Time  `json:"created_at"`
    UpdatedAt   time.Time  `json:"update_at"`
    DeletedAt   *time.Time `sql:"index" json:"deleted_at"`
}

Папка 

type Folder struct {
    Stuff
    Name      string      `json:"name" form:"name" gorm:"name;"`
    Documents []*Document `json:"documents" gorm:"many2many:documents_folders;"`
}

Документ 

type Document struct {
    Stuff
    Name        string    `json:"name" form:"name" gorm:"name;"`
    Folders     []*Folder `json:"folders" gorm:"many2many:documents_folders;"`
}

DocumentsFolders

Я создал эту модель для доступа к таблице, она генерируется автоматически, но мне нужно добавить столбец State 

type DocumentsFolders struct {
    DocumentID string    `json:"document_id" sql:"type:char(36) REFERENCES documents(id)"`
    Document   *Document `json:"document"`
    FolderID   string    `json:"folder_id" sql:"type:char(36) REFERENCES folders(id)"`
    Folder     *Folder   `json:"folder"`
    State      string    `json:"state" form:"state" gorm:"state;"`
}

Я не могу получить к ней доступ при предварительной загрузкеdocument и folder вроде: 

folder := Folder{}
db.Preload("Documents").Where("id = ?", "myUUID").First(&folder)

Хотелось бы увидеть состояние ассоциации документов.

Спасибо за вашу помощь

1 Ответ

0 голосов
/ 11 октября 2019

У вас есть следующие проблемы:

  • Вы пропустили добавление AutoMigrate для DocumentsFolders.
  • Не используйте слайсы с указателями.

Проверьте приведенный ниже пример для получения дополнительной справки: 

package main

import (
    "fmt"
    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/sqlite"
    "math/rand"
    "time"
)

type UUID = string

type Model struct {
    ID        UUID       `gorm:"type:char(36);primary_key;" json:"id"`
    CreatedAt time.Time  `json:"created_at"`
    UpdatedAt time.Time  `json:"update_at"`
    DeletedAt *time.Time `sql:"index" json:"deleted_at"`
}

type Folder struct {
    Model
    Name      string     `json:"name" form:"name" gorm:"name;"`
    Documents []Document `json:"documents" gorm:"many2many:documents_folders;"`
}

type Document struct {
    Model
    Name    string   `json:"name" form:"name" gorm:"name;"`
    Folders []Folder `json:"folders" gorm:"many2many:documents_folders;"`
}

type DocumentsFolders struct {
    DocumentID string    `json:"document_id" sql:"type:char(36) REFERENCES documents(id)"`
    Document   *Document `json:"document"`
    FolderID   string    `json:"folder_id" sql:"type:char(36) REFERENCES folders(id)"`
    Folder     *Folder   `json:"folder"`
    State      string    `json:"state" form:"state" gorm:"state;"`
}

var i = rand.Int()

func genId() string {
    id := i
    i++
    return fmt.Sprintf("%036d", id)
}

func main() {
    db, _ := gorm.Open("sqlite3", "./db.sqlite")
    db.AutoMigrate(&Document{}, &Folder{}, &DocumentsFolders{})

    folder1 := &Folder{
        Model:     Model{ID: genId()},
        Name:      "myFolder",
        Documents: nil,
    }
    db.FirstOrCreate(&folder1, folder1.ID)

    doc1 := &Document{
        Model:   Model{ID: genId()},
        Name:    "doc1",
    }
    db.FirstOrCreate(doc1, doc1.ID)

    doc2 := &Document{
        Model:   Model{ID: genId()},
        Name:    "doc2",
    }
    db.FirstOrCreate(doc2, doc2.ID)

    // join them
    db.Create(&DocumentsFolders{
        DocumentID: doc1.ID,
        FolderID:   folder1.ID,
        State:      "happy!",
    })
    db.Create(&DocumentsFolders{
        DocumentID: doc2.ID,
        FolderID:   folder1.ID,
        State:      "happy!",
    })

    // get the new folder to check
    db.Preload("Documents").First(folder1, folder1.ID)

    // two documents are printed
    fmt.Printf("Documents count: %d\n", len(folder1.Documents)) // prints 2
}
...