@@ -204,20 +204,26 @@ var localAppConf = `name = "hello-local"
204
204
var previewAppConf = `include "app.conf"
205
205
#当前环境的配置可以配置在这里
206
206
207
+ name = "preivew"
208
+
207
209
#请保持在最后一行
208
210
include "local_app.conf"
209
211
`
210
212
211
213
var prodAppConf = `include "app.conf"
212
214
#当前环境的配置可以配置在这里
213
215
216
+ name = "prod"
217
+
214
218
#请保持在最后一行
215
219
include "local_app.conf"
216
220
`
217
221
218
222
var testAppConf = `include "app.conf"
219
223
#当前环境的配置可以配置在这里
220
224
225
+ name = "test"
226
+
221
227
#请保持在最后一行
222
228
include "local_app.conf"
223
229
`
@@ -359,20 +365,20 @@ import (
359
365
"flag"
360
366
"github.com/jinzhu/gorm"
361
367
"fmt"
362
- "github.com/astaxie/beego "
368
+ "{{baseDir}}/pkg/utils "
363
369
)
364
370
365
371
var (
366
372
Conn *gorm.DB
367
373
)
368
374
369
375
func GetDbConnect() (*gorm.DB, error) {
370
- dbConStr := fmt.Sprintf("%s:%s@tcp(%s)/%s%s?charset=utf8&parseTime=True&loc=Local", beego.AppConfig .String("db::user"), beego.AppConfig .String("db::password"), beego.AppConfig .String("db::host"), beego.AppConfig .String("db::database"), getConnectDbName())
376
+ dbConStr := fmt.Sprintf("%s:%s@tcp(%s)/%s%s?charset=utf8&parseTime=True&loc=Local", utils.BeeConfig .String("db::user"), utils.BeeConfig .String("db::password"), utils.BeeConfig .String("db::host"), utils.BeeConfig .String("db::database"), getConnectDbName())
371
377
fmt.Println(dbConStr, "dbConStr")
372
378
db, err := gorm.Open("mysql", dbConStr)
373
379
Conn = db
374
380
375
- runMode := beego.AppConfig .String("runmode")
381
+ runMode := utils.BeeConfig .String("runmode")
376
382
if runMode == "dev" || runMode == "test" {
377
383
db.LogMode(true)
378
384
}
@@ -395,11 +401,11 @@ var pkgDbMongodbTemp = `package db
395
401
import (
396
402
"context"
397
403
"fmt"
398
- "github.com/astaxie/beego"
404
+ "time"
405
+ "{{baseDir}}/pkg/utils"
399
406
"go.mongodb.org/mongo-driver/mongo"
400
407
"go.mongodb.org/mongo-driver/mongo/options"
401
408
"go.mongodb.org/mongo-driver/mongo/readpref"
402
- "time"
403
409
)
404
410
405
411
var (
@@ -409,7 +415,7 @@ var (
409
415
//连接mongodb
410
416
func GetMongodbClient() (*mongo.Client, error) {
411
417
ctx, cancel := context.WithTimeout(context.Background(), 10 * time.Second)
412
- host := fmt.Sprintf("mongodb://%s", beego.AppConfig .String("mongodb::host"))
418
+ host := fmt.Sprintf("mongodb://%s", utils.BeeConfig .String("mongodb::host"))
413
419
client, err := mongo.Connect(ctx, options.Client().ApplyURI(host))
414
420
cancel()
415
421
if err != nil {
@@ -435,7 +441,7 @@ var pkgDbRedisTemp = `package db
435
441
/*
436
442
import (
437
443
"github.com/go-redis/redis"
438
- "github.com/astaxie/beego "
444
+ "{{baseDir}}/pkg/utils "
439
445
)
440
446
441
447
var (
@@ -444,15 +450,15 @@ var (
444
450
445
451
//连接redis
446
452
func GetRedisClient() (*redis.Client, error){
447
- db ,_ := beego.AppConfig .Int("redis::database")
453
+ db ,_ := utils.BeeConfig .Int("redis::database")
448
454
redisOption := &redis.Options{
449
- Addr: beego.AppConfig .String("redis::host"),
455
+ Addr: utils.BeeConfig .String("redis::host"),
450
456
DB: db,
451
457
}
452
458
453
- runMode := beego.AppConfig .String("runmode")
459
+ runMode := utils.BeeConfig .String("runmode")
454
460
if runMode != "prod" {
455
- redisOption.Password = beego.AppConfig .String("redis::password")
461
+ redisOption.Password = utils.BeeConfig .String("redis::password")
456
462
}
457
463
client := redis.NewClient(redisOption)
458
464
@@ -505,8 +511,11 @@ type UserForm struct {
505
511
506
512
var pkgUtilsConfigTemp = `package utils
507
513
514
+ import "github.com/astaxie/beego/config"
515
+
508
516
var (
509
517
MyConfig *WebConfig
518
+ BeeConfig config.Configer
510
519
)
511
520
512
521
type WebConfig struct {
@@ -782,7 +791,7 @@ import (
782
791
"log"
783
792
"os"
784
793
"{{baseDir}}/pkg/utils"
785
- "github.com/astaxie/beego"
794
+ "github.com/astaxie/beego/config "
786
795
)
787
796
788
797
var envMode = "WebRunMode"
@@ -797,10 +806,11 @@ func InitConfig() {
797
806
}
798
807
799
808
configName := fmt.Sprintf("conf/%s_app.conf", env)
800
- err := beego.LoadAppConfig ("ini", configName)
809
+ beeConfig, err := config.NewConfig ("ini", configName)
801
810
if err != nil {
802
811
log.Fatal(err)
803
812
}
813
+ utils.BeeConfig = beeConfig
804
814
805
815
configList, err := LoadConfig()
806
816
if err != nil {
@@ -816,10 +826,10 @@ func InitConfig() {
816
826
*/
817
827
func LoadConfig() (*utils.WebConfig, error) {
818
828
config := &utils.WebConfig{}
819
- config.DbHost = beego.AppConfig .String("db::host")
820
- config.DbName = beego.AppConfig .String("db::user")
821
- config.DbPassword = beego.AppConfig .String("db::password")
822
- version, err := beego.AppConfig .Float("version")
829
+ config.DbHost = utils.BeeConfig .String("db::host")
830
+ config.DbName = utils.BeeConfig .String("db::user")
831
+ config.DbPassword = utils.BeeConfig .String("db::password")
832
+ version, err := utils.BeeConfig .Float("version")
823
833
if err != nil {
824
834
config.Version = version
825
835
}
0 commit comments