前言

最近在开发的时候，需要进行数据库计算，主要是根据表中的某些字段进行汇总计算，但由于数据库表中已有同名字段名，ent 不会使用计算后的指标，默认使用schema中定义的字段，导致无法返回正确的结果。

针对这种情况，我能想到的方法有2种: 1) 不使用同名的字段名 2) 查找ent是否有相关的解决方案。

这里我采用了第二种方法，查找相关的issues, 通过查找相关issue,找到了相关的解决方案: ent 的 sql/modifier 特性。

场景重现

定义一个新的数据库表结构，结构如下:

1
2
3
4
5
6
7
8

func (Ad) Fields() []ent.Field {
    return []ent.Field{
        field.Float("estimated_earnings"),
        field.Int("page_views"),
        field.Time("date"),
        field.Float("page_views_rpm").Optional(),
    }
}

page_views_rpm 字段是由 estimated_earnings 和 page_views 计算而来。

编写相应的查询代码:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

package main

import (
	"context"
	"entgo.io/ent/dialect/sql"
	"fmt"
	_ "github.com/lib/pq"
	"log"
	"modifier-demo/ent"
	"modifier-demo/ent/ad"
	"time"
)

type Ads struct {
	EstimatedEarnings float64   `json:"estimated_earnings"`
	PageViews         int64     `json:"page_views"`
	Date              time.Time `json:"date"`
	PageViewsRpm      float64   `json:"page_views_rpm"`
}

func main() {
	client, err := ent.Open("postgres", "host=127.0.0.1 port=5432 sslmode=disable user=postgres dbname=data_test password=mysecretpassword")
	if err != nil {
		log.Fatalf("failed opening connection to postgres: %v", err)
	}
	defer client.Close()
	// Run the auto migration tool.
	if err := client.Schema.Create(context.Background()); err != nil {
		log.Fatalf("failed creating schema resources: %v", err)
	}
	var a []Ads
	err = client.Debug().Ad.Query().Order(ent.Desc(ad.FieldPageViewsRpm)).GroupBy(ad.FieldDate).Aggregate(func(selector *sql.Selector) string {
		return sql.As(" CAST(COALESCE(SUM(estimated_earnings) / NULLIF(SUM(page_views)*1.0, 0.0)*1000, 0)AS numeric(10,2))", "page_views_rpm")
	}).Scan(context.TODO(), &a)
	if err != nil {
		return
	}
	fmt.Println(a)
}

主要是根据date来汇总并重新计算 page_views_rpm 字段,运行代码后发现没有成功输出，打印后发现ent使用了旧的PageViewsRpm字段进行排序，导致sql无法顺利运行。

1
2
3
4
5
6

SELECT "ads"."date",
       CAST(COALESCE(SUM(estimated_earnings) / NULLIF(SUM(page_views) * 1.0, 0.0) * 1000,
                     0) AS numeric(10,2)) AS "page_views_rpm"
FROM "ads"
GROUP BY "ads"."date"
ORDER BY "ads"."page_views_rpm"

解决

这里可以使用 ent 的 feature sql/modifier 来解决，我们先在 generate.go 开启特性 –feature sql/modifier，然后重新生成代码，使用Modify方法添加自定义修饰符即可，相应的代码如下:

1
2
3
4
5
6
7
8

	err = client.Debug().Ad.Query().Modify(func(s *sql.Selector) {
		s.Select(sql.As(" CAST(COALESCE(SUM(estimated_earnings) / NULLIF(SUM(page_views)*1.0, 0.0)*1000, 0)AS numeric(10,2))", "page_views_rpm")).
			GroupBy("date").
			OrderBy("page_views_rpm")
	}).Scan(context.TODO(), &a)
	if err != nil {
		return
	}

运行修改后的代码，结果顺利输出，相应的sql如下:

1
2
3
4
5

SELECT CAST(COALESCE(SUM(estimated_earnings) / NULLIF(SUM(page_views) * 1.0, 0.0) * 1000,
                     0) AS numeric(10, 2)) AS "page_views_rpm"
FROM "ads"
GROUP BY "date"
ORDER BY "page_views_rpm"

参考