/
analysis.go
40 lines (32 loc) · 982 Bytes
/
analysis.go
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
40
package table
import (
"fmt"
"hash/crc32"
"strconv"
"gorm.io/gorm"
)
// Crc32Hash 用于32张分表
func Crc32Hash(src string) string {
sum := crc32.ChecksumIEEE([]byte(src)) % 32
return strconv.Itoa(int(sum))
}
// Crc16Hash 用于16张分表
func Crc16Hash(src string) string {
sum := crc32.ChecksumIEEE([]byte(src)) % 16
return strconv.Itoa(int(sum))
}
// Crc8Hash 用于8张分表
func Crc8Hash(src string) string {
sum := crc32.ChecksumIEEE([]byte(src)) % 8
return strconv.Itoa(int(sum))
}
// DynamicTable 动态表名
// e.g. DB.Scopes(DynamicTable(Crc32Hash, "test", "小圈圈")).Find(&tests)
// 设置动态表名scope params: f 分表计算函数 baseTable 基础表名 fieldValue 参与分表字段
func DynamicTable(f func(string) string, baseTable, fieldValue string) func(db *gorm.DB) *gorm.DB {
return func(db *gorm.DB) *gorm.DB {
return db.Table(fmt.Sprintf("%s_%s", baseTable, f(fieldValue)))
}
}
func CreateSubTable(f func(string) string) {
}