db.Use(sharding.Register(sharding.Config{ ShardingKey: "user_id", NumberOfShards: 64, PrimaryKeyGenerator: sharding.PKSnowflake, }, "orders").Register(sharding.Config{ ShardingKey: "user_id", NumberOfShards: 256, PrimaryKeyGenerator: sharding.PKSnowflake, // This case for show up give notifications, audit_logs table use same sharding rule. }, Notification{}, AuditLog{}))
像往常一样使用数据库会话。请注意,在操作分片表时,查询应具有“分片键”。
// Gorm create example, this will insert to orders_02 db.Create(&Order{UserID: 2}) // sql: INSERT INTO orders_2 ...
// Show have use Raw SQL to insert, this will insert into orders_03 db.Exec("INSERT INTO orders(user_id) VALUES(?)", int64(3))
// This will throw ErrMissingShardingKey error, because there not have sharding key presented. db.Create(&Order{Amount: 10, ProductID: 100}) fmt.Println(err)
// Find, this will redirect query to orders_02 var orders []Order db.Model(&Order{}).Where("user_id", int64(2)).Find(&orders) fmt.Printf("%#v\n", orders)
// Raw SQL also supported db.Raw("SELECT * FROM orders WHERE user_id = ?", int64(3)).Scan(&orders) fmt.Printf("%#v\n", orders)
// This will throw ErrMissingShardingKey error, because WHERE conditions not included sharding key err = db.Model(&Order{}).Where("product_id", "1").Find(&orders).Error fmt.Println(err)
// Update and Delete are similar to create and query db.Exec("UPDATE orders SET product_id = ? WHERE user_id = ?", 2, int64(3)) err = db.Exec("DELETE FROM orders WHERE product_id = 3").Error fmt.Println(err) // ErrMissingShardingKey