| 534 | return nil |
| 535 | } |
| 536 | func (nvq *NodeVersionQuery) loadStorageFile(ctx context.Context, query *StorageFileQuery, nodes []*NodeVersion, init func(*NodeVersion), assign func(*NodeVersion, *StorageFile)) error { |
| 537 | ids := make([]uuid.UUID, 0, len(nodes)) |
| 538 | nodeids := make(map[uuid.UUID][]*NodeVersion) |
| 539 | for i := range nodes { |
| 540 | if nodes[i].node_version_storage_file == nil { |
| 541 | continue |
| 542 | } |
| 543 | fk := *nodes[i].node_version_storage_file |
| 544 | if _, ok := nodeids[fk]; !ok { |
| 545 | ids = append(ids, fk) |
| 546 | } |
| 547 | nodeids[fk] = append(nodeids[fk], nodes[i]) |
| 548 | } |
| 549 | if len(ids) == 0 { |
| 550 | return nil |
| 551 | } |
| 552 | query.Where(storagefile.IDIn(ids...)) |
| 553 | neighbors, err := query.All(ctx) |
| 554 | if err != nil { |
| 555 | return err |
| 556 | } |
| 557 | for _, n := range neighbors { |
| 558 | nodes, ok := nodeids[n.ID] |
| 559 | if !ok { |
| 560 | return fmt.Errorf(`unexpected foreign-key "node_version_storage_file" returned %v`, n.ID) |
| 561 | } |
| 562 | for i := range nodes { |
| 563 | assign(nodes[i], n) |
| 564 | } |
| 565 | } |
| 566 | return nil |
| 567 | } |
| 568 | func (nvq *NodeVersionQuery) loadComfyNodes(ctx context.Context, query *ComfyNodeQuery, nodes []*NodeVersion, init func(*NodeVersion), assign func(*NodeVersion, *ComfyNode)) error { |
| 569 | fks := make([]driver.Value, 0, len(nodes)) |
| 570 | nodeids := make(map[uuid.UUID]*NodeVersion) |