Множество улучшенийv0.0.2

1 files changed, 5 insertions, 5 deletions

diff --git a/repository/node.go b/repository/node.go
index d5794b6..27282af 100644
--- a/repository/node.go
+++ b/repository/node.go
@@ -42,17 +42,17 @@ func (t *Node) Get(ctx context.Context, topicID int) (*models.Node, error) {
 		Model(node).
 		Where(`n.id = ?`, topicID).
 		Relation("Author").
+		Relation("Parent").
 		Scan(ctx)
 }
 
-func (t *Node) List(ctx context.Context, topicID int) ([]*models.Node, int, error) {
+func (t *Node) List(ctx context.Context, topicID int) ([]*models.Node, error) {
 	posts := make([]*models.Node, 0)
 
-	count, err := t.db.NewSelect().
+	return posts, t.db.NewSelect().
 		Model(&posts).
 		Where(`parent_id = ?`, topicID).
 		Relation("Author").
-		ScanAndCount(ctx)
-
-	return posts, count, err
+		Relation("Children").
+		Scan(ctx)
 }