add swagger demo

pkg/common/config/config.go

@@ -1,6 +1,7 @@
 package config
 
 import (
+	"fmt"
 	"io/ioutil"
 	"os"
 	"path/filepath"
@@ -486,6 +487,8 @@ type PDefaultTips struct {
 
 func init() {
 	cfgName := os.Getenv("CONFIG_NAME")
+	fmt.Println(Root, cfgName)
+
 	if len(cfgName) == 0 {
 		cfgName = Root + "/config/config.yaml"
 	}

pkg/common/db/model_struct.go

@@ -7,6 +7,7 @@ type Register struct {
 	Password string `gorm:"column:password;type:varchar(255)" json:"password"`
 	Ex       string `gorm:"column:ex;size:1024" json:"ex"`
 	UserID   string `gorm:"column:user_id;type:varchar(255)" json:"userID"`
+	AreaCode string `gorm:"column:area_code;type:varchar(255)"`
 }
 
 //

pkg/common/db/mysql_model/im_mysql_model/demo_model.go

@@ -5,22 +5,23 @@ import (
 	_ "github.com/jinzhu/gorm"
 )
 
-func GetRegister(account string) (*db.Register, error) {
+func GetRegister(account, areaCode string) (*db.Register, error) {
 	dbConn, err := db.DB.MysqlDB.DefaultGormDB()
 	if err != nil {
 		return nil, err
 	}
 	var r db.Register
-	return &r, dbConn.Table("registers").Where("account = ?",
-		account).Take(&r).Error
+	return &r, dbConn.Table("registers").Where("account = ? or account =? and area_code=?",
+		account, account, areaCode).Take(&r).Error
 }
 
-func SetPassword(account, password, ex, userID string) error {
+func SetPassword(account, password, ex, userID, areaCode string) error {
 	r := db.Register{
 		Account:  account,
 		Password: password,
 		Ex:       ex,
 		UserID:   userID,
+		AreaCode: areaCode,
 	}
 	dbConn, err := db.DB.MysqlDB.DefaultGormDB()
 	if err != nil {