Entgo多表链接查询

在Entgo框架中进行多表链接查询可以使用Entgo提供的关系方法。具体步骤如下：

1. 在Entgo中定义相关的Schema，例如：

package schema

import (
    "entgo.io/ent"
    "entgo.io/ent/schema/field"
    "entgo.io/ent/schema/index"
    "entgo.io/ent/schema/edge"
)

// User holds the schema definition for the User entity.
type User struct {
    ent.Schema
}

// Fields of the User.
func (User) Fields() []ent.Field {
    return []ent.Field{
        field.String("name"),
    }
}

// Edges of the User.
func (User) Edges() []ent.Edge {
    return []ent.Edge{
        edge.To("pets", Pet.Type),
    }
}

// Indexes of the User.
func (User) Indexes() []ent.Index {
    return []ent.Index{
        index.Fields("name"),
    }
}

// Pet holds the schema definition for the Pet entity.
type Pet struct {
    ent.Schema
}

// Fields of the Pet.
func (Pet) Fields() []ent.Field {
    return []ent.Field{
        field.String("name"),
    }
}

// Edges of the Pet.
func (Pet) Edges() []ent.Edge {
    return []ent.Edge{
        edge.From("owner", User.Type).
            Ref("pets").
            Unique(),
    }
}

// Indexes of the Pet.
func (Pet) Indexes() []ent.Index {
    return []ent.Index{
        index.Fields("name"),
    }
}

在上面的示例中，我们定义了两个Schema，分别为`User`和`Pet`。在`User`中，我们定义了一个`pets`边，表示一个用户可以拥有多个宠物。在`Pet`中，我们定义了一个`owner`边，表示一个宠物只属于一个用户。

2. 在代码中使用关系方法进行多表链接查询，例如：

package main

import (
    "context"
    "fmt"
    "log"

    "entgo.io/ent/dialect"
    "entgo.io/ent/dialect/sql"
    "entgo.io/ent/entc"
    "entgo.io/ent/entc/gen"
    "entgo.io/ent/examples/entc/integration/ent"
)

func main() {
    client, err := ent.Open(dialect.SQLite, "file:ent?mode=memory&cache=shared&_fk=1")
    if err != nil {
        log.Fatalf("failed opening connection to sqlite: %v", err)
    }
    defer client.Close()

    if err := client.Schema.Create(context.Background()); err != nil {
        log.Fatalf("failed creating schema resources: %v", err)
    }

    // 创建一些测试数据
    if err := createTestData(client); err != nil {
        log.Fatalf("failed creating test data: %v", err)
    }

    // 查询所有宠物及其所属的用户姓名
    pets, err := client.Pet.Query().
        WithOwner().
        Order(sql.OrderBy("name")).
        All(context.Background())
    if err != nil {
        log.Fatalf("failed querying pets: %v", err)
    }

    // 打印宠物及其所属的用户姓名
    for _, pet := range pets {
        fmt.Printf("%s - %s\n", pet.Name, pet.Edges.Owner.Name)
    }
}

func createTestData(client *ent.Client) error {
    tx, err := client.Tx(context.Background())
    if err != nil {
        return fmt.Errorf("failed starting transaction: %v", err)
    }
    defer tx.Rollback()

    user1, err := tx.User.Create().SetName("张三").Save(context.Background())
    if err != nil {
        return fmt.Errorf("failed creating user: %v", err)
    }

    user2, err := tx.User.Create().SetName("李四").Save(context.Background())
    if err != nil {
        return fmt.Errorf("failed creating user: %v", err)
    }

    _, err = tx.Pet.Create().SetName("旺财").SetOwner(user1).Save(context.Background())
    if err != nil {
        return fmt.Errorf("failed creating pet: %v", err)
    }

    _, err = tx.Pet.Create().SetName("小黄").SetOwner(user1).Save(context.Background())
    if err != nil {
        return fmt.Errorf("failed creating pet: %v", err)
    }

    _, err = tx.Pet.Create().SetName("小黑").SetOwner(user2).Save(context.Background())
    if err != nil {
        return fmt.Errorf("failed creating pet: %v", err)
    }

    _, err = tx.Pet.Create().SetName("小白").SetOwner(user2).Save(context.Background())
    if err != nil {
        return fmt.Errorf("failed creating pet: %v", err)
    }

    return tx.Commit()
}

在上面的示例中，我们首先使用`WithOwner()`方法在查询语句中加载宠物的所有者信息，然后使用`Order()`方法按照宠物的名称进行排序。最后，我们通过`All()`方法查询所有宠物及其所属的用户姓名，并打印结果。

需要注意的是，在使用关系方法时，我们可以使用`WithXxx()`方法加载相关的实体信息，例如在上面的示例中，我们使用了`WithOwner()`方法加载宠物的所有者信息。