同时操作部署在不同服务器的多种数据库

1.本项目骨架在单元测试已经提供了同时操作多服务器、多种数据库的示例代码,为了将此功能更清晰地展现出来,本篇将单独进行介绍.
2.面对复杂场景,需要多个客户端连接到部署在多个不同服务器的 mysqlsqlserverpostgresql 等数据库时, 由于配置文件(config/gorm_v2.yml)只提供了一份数据库连接,无法满足需求,这时您可以通过自定义参数直接连接任意数据库,获取一个数据库句柄,供业务使用.

相关代码

在线代码预览, 251 行处。

  1. func TestCustomeParamsConnMysql(t *testing.T) {
  2. // 定义一个查询结果接受结构体
  3. type DataList struct {
  4. Id int
  5. Username string
  6. LastLoginIp string
  7. Status int
  8. }
  9. // 设置动态参数连接任意多个数据库,以mysql为例进行单元测试
  10. // 参数结构体 Write 和 Read 只有设置了具体值,才会生效,否则程序自动使用配置目录(config/gorm_v2.yml)中的参数
  11. confPrams := gorm_v2.ConfigParams{
  12. Write: struct {
  13. Host string
  14. DataBase string
  15. Port int
  16. Prefix string
  17. User string
  18. Pass string
  19. Charset string
  20. }{Host: "127.0.0.1", DataBase: "db_test", Port: 3306, Prefix: "tb_", User: "root", Pass: "DRsXT5ZJ6Oi55LPQ", Charset: "utf8"},
  21. Read: struct {
  22. Host string
  23. DataBase string
  24. Port int
  25. Prefix string
  26. User string
  27. Pass string
  28. Charset string
  29. }{Host: "127.0.0.1", DataBase: "db_stocks", Port: 3306, Prefix: "tb_", User: "root", Pass: "DRsXT5ZJ6Oi55LPQ", Charset: "utf8"}}
  30. var vDataList []DataList
  31. //gorm_v2.GetSqlDriver 参数介绍
  32. // sqlType : mysql 、sqlserver、postgresql 等数据库库类型
  33. // readDbIsOpen : 是否开启读写分离,1表示开启读数据库的配置,那么 confPrams.Read 参数部分才会生效; 0 则表示 confPrams.Read 部分参数直接忽略(即 读、写同库)
  34. // confPrams 动态配置的数据库参数
  35. // 此外,其他参数,例如数据库连接池参数等,则直接调用配置项数据库连接池参数,基本不需要配置,这部分对实际操作影响不大
  36. if gormDbMysql, err := gorm_v2.GetSqlDriver("mysql", 0, confPrams); err == nil {
  37. gormDbMysql.Raw("select id,username,status,last_login_ip from tb_users").Find(&vDataList)
  38. fmt.Printf("Read 数据库查询结果:%v\n", vDataList)
  39. res := gormDbMysql.Exec("update tb_users set real_name='Write数据库更新' where id<=2 ")
  40. if res.Error==nil{
  41. fmt.Println("write 数据库更新成功")
  42. }else{
  43. t.Errorf("单元测试失败,相关错误:%s\n",res.Error.Error())
  44. }
  45. }
  46. }