You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

90 lines
2.6 KiB

4 years ago
4 years ago
4 years ago
  1. package mysql2
  2. import (
  3. "context"
  4. "database/sql"
  5. "fmt"
  6. "time"
  7. "github.com/chrislusf/seaweedfs/weed/filer"
  8. "github.com/chrislusf/seaweedfs/weed/filer/abstract_sql"
  9. "github.com/chrislusf/seaweedfs/weed/filer/mysql"
  10. "github.com/chrislusf/seaweedfs/weed/util"
  11. _ "github.com/go-sql-driver/mysql"
  12. )
  13. const (
  14. CONNECTION_URL_PATTERN = "%s:%s@tcp(%s:%d)/%s?charset=utf8"
  15. )
  16. func init() {
  17. filer.Stores = append(filer.Stores, &MysqlStore2{})
  18. }
  19. type MysqlStore2 struct {
  20. abstract_sql.AbstractSqlStore
  21. }
  22. func (store *MysqlStore2) GetName() string {
  23. return "mysql2"
  24. }
  25. func (store *MysqlStore2) Initialize(configuration util.Configuration, prefix string) (err error) {
  26. return store.initialize(
  27. configuration.GetString(prefix+"createTable"),
  28. configuration.GetString(prefix+"upsertQuery"),
  29. configuration.GetBool(prefix+"enableUpsert"),
  30. configuration.GetString(prefix+"username"),
  31. configuration.GetString(prefix+"password"),
  32. configuration.GetString(prefix+"hostname"),
  33. configuration.GetInt(prefix+"port"),
  34. configuration.GetString(prefix+"database"),
  35. configuration.GetInt(prefix+"connection_max_idle"),
  36. configuration.GetInt(prefix+"connection_max_open"),
  37. configuration.GetInt(prefix+"connection_max_lifetime_seconds"),
  38. configuration.GetBool(prefix+"interpolateParams"),
  39. )
  40. }
  41. func (store *MysqlStore2) initialize(createTable, upsertQuery string, enableUpsert bool, user, password, hostname string, port int, database string, maxIdle, maxOpen,
  42. maxLifetimeSeconds int, interpolateParams bool) (err error) {
  43. store.SupportBucketTable = true
  44. if !enableUpsert {
  45. upsertQuery = ""
  46. }
  47. store.SqlGenerator = &mysql.SqlGenMysql{
  48. CreateTableSqlTemplate: createTable,
  49. DropTableSqlTemplate: "drop table `%s`",
  50. UpsertQueryTemplate: upsertQuery,
  51. }
  52. sqlUrl := fmt.Sprintf(CONNECTION_URL_PATTERN, user, password, hostname, port, database)
  53. adaptedSqlUrl := fmt.Sprintf(CONNECTION_URL_PATTERN, user, "<ADAPTED>", hostname, port, database)
  54. if interpolateParams {
  55. sqlUrl += "&interpolateParams=true"
  56. adaptedSqlUrl += "&interpolateParams=true"
  57. }
  58. var dbErr error
  59. store.DB, dbErr = sql.Open("mysql", sqlUrl)
  60. if dbErr != nil {
  61. store.DB.Close()
  62. store.DB = nil
  63. return fmt.Errorf("can not connect to %s error:%v", adaptedSqlUrl, err)
  64. }
  65. store.DB.SetMaxIdleConns(maxIdle)
  66. store.DB.SetMaxOpenConns(maxOpen)
  67. store.DB.SetConnMaxLifetime(time.Duration(maxLifetimeSeconds) * time.Second)
  68. if err = store.DB.Ping(); err != nil {
  69. return fmt.Errorf("connect to %s error:%v", sqlUrl, err)
  70. }
  71. if err = store.CreateTable(context.Background(), abstract_sql.DEFAULT_TABLE); err != nil {
  72. return fmt.Errorf("init table %s: %v", abstract_sql.DEFAULT_TABLE, err)
  73. }
  74. return nil
  75. }