/models/ToolsModel.go

https://github.com/3xxx/engineercms · Go · 94 lines · 1 code · 11 blank · 82 comment · 0 complexity · c2831ad3d76644f40212db88ca3e2a84 MD5 · raw file 

    

  1. // //定义一个工具包，用来管理gorm数据库连接池的初始化工作。
    2. 

  2. package models
    3. 

  3. 

  4. // import (
    5. 

  5. // 	// "github.com/astaxie/beego/orm"
    6. 

  6. // 	"fmt"
    7. 

  7. // 	"github.com/astaxie/beego"
    8. 

  8. // 	"github.com/jinzhu/gorm"
    9. 

  9. // 	// _ "github.com/jinzhu/gorm/dialects/sqlite"
    10. 

  10. // )
    11. 

  11. 

  12. // //定义全局的db对象，我们执行数据库操作主要通过他实现。
    13. 

  13. // var _db *gorm.DB
    14. 

  14. 

  15. // //包初始化函数，golang特性，每个包初始化的时候会自动执行init函数，这里用来初始化gorm。
    16. 

  16. // func init() {
    17. 

  17. // 	// ...忽略dsn配置，请参考上面例子...
    18. 

  18. // 	//配置MySQL连接参数
    19. 

  19. // 	// username := "root"  //账号
    20. 

  20. // 	// password := "123456" //密码
    21. 

  21. // 	// host := "127.0.0.1" //数据库地址，可以是Ip或者域名
    22. 

  22. // 	// port := 3306 //数据库端口
    23. 

  23. // 	// Dbname := "tizi365" //数据库名
    24. 

  24. // 	// timeout := "10s" //连接超时，10秒
    25. 

  25. 

  26. // 	//拼接下dsn参数, dsn格式可以参考上面的语法，这里使用Sprintf动态拼接dsn参数，因为一般数据库连接参数，我们都是保存在配置文件里面，需要从配置文件加载参数，然后拼接dsn。
    27. 

  27. // 	// dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8&parseTime=True&loc=Local&timeout=%s", username, password, host, port, Dbname, timeout)
    28. 

  28. 

  29. // 	//连接MYSQL, 获得DB类型实例，用于后面的数据库读写操作。
    30. 

  30. // 	// db, err := gorm.Open("mysql", dsn)
    31. 

  31. // 	// if err != nil {
    32. 

  32. // 	// 	panic("连接数据库失败, error=" + err.Error())
    33. 

  33. // 	// }
    34. 

  34. // 	var err error
    35. 

  35. // 	var dns string
    36. 

  36. // 	db_type := beego.AppConfig.String("db_type")
    37. 

  37. // 	db_name := beego.AppConfig.String("db_name")
    38. 

  38. // 	db_path := beego.AppConfig.String("db_path")
    39. 

  39. // 	if db_path == "" {
    40. 

  40. // 		db_path = "./"
    41. 

  41. // 	}
    42. 

  42. 

  43. // 	dns = fmt.Sprintf("%s%s.db", db_path, db_name)
    44. 

  44. // 	_db, err = gorm.Open(db_type, dns)
    45. 

  45. // 	// _db.LogMode(true)
    46. 

  46. // 	if err != nil {
    47. 

  47. // 		panic("连接数据库失败, error=" + err.Error())
    48. 

  48. // 	}
    49. 

  49. // 	// defer gdb.Close()
    50. 

  50. // 	//禁止表名复数形式
    51. 

  51. // 	_db.SingularTable(true)
    52. 

  52. 

  53. // 	//延时关闭数据库连接
    54. 

  54. // 	// defer db.Close()
    55. 

  55. 

  56. // 	//连接MYSQL, 获得DB类型实例，用于后面的数据库读写操作。
    57. 

  57. // 	//    _db, err := gorm.Open("mysql", dsn)
    58. 

  58. // 	//    if err != nil {
    59. 

  59. // 	// 	panic("连接数据库失败, error=" + err.Error())
    60. 

  60. // 	// }
    61. 

  61. // 	// 开发的时候需要打开调试日志
    62. 

  62. // 	_db.LogMode(true)
    63. 

  63. // 	//设置数据库连接池参数
    64. 

  64. // 	_db.DB().SetMaxOpenConns(100) //设置数据库连接池最大连接数
    65. 

  65. // 	_db.DB().SetMaxIdleConns(20)  //连接池最大允许的空闲连接数，如果没有sql任务需要执行的连接数大于20，超过的连接会被连接池关闭。
    66. 

  66. // }
    67. 

  67. 

  68. // //获取gorm db对象，其他包需要执行数据库查询的时候，只要通过tools.getDB()获取db对象即可。
    69. 

  69. // //不用担心协程并发使用同样的db对象会共用同一个连接，
    70. 

  70. // // db对象在调用他的方法的时候会从数据库连接池中获取新的连接
    71. 

  71. // // 注意：使用连接池技术后，千万不要使用完db后调用db.Close关闭数据库连接，
    72. 

  72. // // 这样会导致整个数据库连接池关闭，导致连接池没有可用的连接
    73. 

  73. // func GetDB() *gorm.DB {
    74. 

  74. // 	return _db
    75. 

  75. // }
    76. 

  76. 

  77. // // func GetDB() *gorm.DB {
    78. 

  78. // // 	var err error
    79. 

  79. // // 	var dns string
    80. 

  80. // // 	db_type := beego.AppConfig.String("db_type")
    81. 

  81. // // 	db_name := beego.AppConfig.String("db_name")
    82. 

  82. // // 	db_path := beego.AppConfig.String("db_path")
    83. 

  83. // // 	if db_path == "" {
    84. 

  84. // // 		db_path = "./"
    85. 

  85. // // 	}
    86. 

  86. 

  87. // // 	dns = fmt.Sprintf("%s%s.db", db_path, db_name)
    88. 

  88. // // 	gdb, err = gorm.Open(db_type, dns)
    89. 

  89. // // 	gdb.LogMode(true)
    90. 

  90. // // 	if err != nil {
    91. 

  91. // // 		panic("failed to connect database")
    92. 

  92. // // 	}
    93. 

  93. // // 	defer gdb.Close()
    94. 

  94. // // }
    95.