注册数据库
()
| 66 | |
| 67 | //注册数据库 |
| 68 | func RegisterDB() { |
| 69 | orm.RegisterDriver("mysql", orm.DRMySQL) |
| 70 | models := []interface{}{ |
| 71 | NewUser(), |
| 72 | NewUserInfo(), |
| 73 | NewAdmin(), |
| 74 | NewCategory(), |
| 75 | NewDocument(), |
| 76 | NewDocumentInfo(), |
| 77 | NewDocumentStore(), |
| 78 | NewDocumentRecycle(), |
| 79 | NewDocumentIllegal(), |
| 80 | NewDocumentComment(), |
| 81 | NewBanner(), |
| 82 | NewRelate(), |
| 83 | NewCollectFolder(), |
| 84 | NewCollect(), |
| 85 | NewFriend(), |
| 86 | NewSys(), |
| 87 | NewWord(), |
| 88 | NewSeo(), |
| 89 | NewPages(), |
| 90 | NewSign(), |
| 91 | NewCoinLog(), |
| 92 | NewReport(), |
| 93 | NewSuggest(), |
| 94 | NewDocumentRemark(), |
| 95 | NewFreeDown(), |
| 96 | NewSearchLog(), |
| 97 | NewDocText(), |
| 98 | NewConfig(), |
| 99 | //NewAdPosition(), |
| 100 | //NewAd(), |
| 101 | } |
| 102 | orm.RegisterModelWithPrefix(beego.AppConfig.DefaultString("db::prefix", "hc_"), models...) |
| 103 | dbUser := beego.AppConfig.String("db::user") |
| 104 | dbPassword := beego.AppConfig.String("db::password") |
| 105 | if envpass := os.Getenv("MYSQL_PASSWORD"); envpass != "" { |
| 106 | dbPassword = envpass |
| 107 | } |
| 108 | dbDatabase := beego.AppConfig.String("db::database") |
| 109 | if envdatabase := os.Getenv("MYSQL_DATABASE"); envdatabase != "" { |
| 110 | dbDatabase = envdatabase |
| 111 | } |
| 112 | dbCharset := beego.AppConfig.String("db::charset") |
| 113 | dbHost := beego.AppConfig.String("db::host") |
| 114 | if envhost := os.Getenv("MYSQL_HOST"); envhost != "" { |
| 115 | dbHost = envhost |
| 116 | } |
| 117 | dbPort := beego.AppConfig.String("db::port") |
| 118 | if envport := os.Getenv("MYSQL_PORT"); envport != "" { |
| 119 | dbPort = envport |
| 120 | } |
| 121 | loc := "Local" |
| 122 | if timezone := beego.AppConfig.String("db::timezone"); timezone != "" { |
| 123 | loc = url.QueryEscape(timezone) |
| 124 | } |
| 125 | dbLink := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=%s&loc=%v", dbUser, dbPassword, dbHost, dbPort, dbDatabase, dbCharset, loc) |
no test coverage detected