Commit 88b12c87 authored by shajiaiming's avatar shajiaiming

fix

parent 02f90c5a
......@@ -2,10 +2,11 @@ package models
import (
"github.com/jinzhu/gorm"
"bwallet/pkg/setting"
)
type Auth struct {
ID int `gorm:"primary_key" json:"id"`
Uid int `gorm:"primary_key" json:"uid"`
Username string `json:"username"`
Password string `json:"password"`
Group string `json:"group"`
......@@ -13,17 +14,17 @@ type Auth struct {
}
func (a Auth) TableName() string {
return "coin_user"
return setting.DatabaseSetting.Name_Manage + ".gli_admin"
}
func CheckAuth(username, password string) (bool, error) {
var auth Auth
err := db.Select("id").Where(Auth{Username: username, Password: password}).First(&auth).Error
err := db.Select("uid").Where(Auth{Username: username, Password: password}).First(&auth).Error
if err != nil && err != gorm.ErrRecordNotFound {
return false, err
}
if auth.ID > 0 {
if auth.Uid > 0 {
return true, nil
}
......@@ -32,7 +33,7 @@ func CheckAuth(username, password string) (bool, error) {
func CheckToken(token string) (bool, error) {
var auth Auth
err := db.Select("id").Where("access_token = ?", token).First(&auth).Error
err := db.Select("uid").Where("access_token = ?", token).First(&auth).Error
if err != nil {
return false, err
}
......
......@@ -2,6 +2,7 @@ package models
import (
"github.com/jinzhu/gorm"
"bwallet/pkg/setting"
)
type Chain struct {
......@@ -16,7 +17,7 @@ type Chain struct {
}
func (c Chain) TableName() string {
return "wallet_chain"
return setting.DatabaseSetting.Name_Sources + ".wallet_chain"
}
/**
......
......@@ -2,6 +2,7 @@ package models
import (
"github.com/jinzhu/gorm"
"bwallet/pkg/setting"
)
type Coin struct {
......@@ -32,10 +33,9 @@ type Coin struct {
}
func (c Coin) TableName() string {
return "wallet_coin"
return setting.DatabaseSetting.Name_Sources + ".wallet_coin"
}
/**
* 通过id检查币种是否存在
* @param id
......
......@@ -27,7 +27,7 @@ func Setup() {
setting.DatabaseSetting.User,
setting.DatabaseSetting.Password,
setting.DatabaseSetting.Host,
setting.DatabaseSetting.Name))
setting.DatabaseSetting.Name_Sources))
if err != nil {
log.Fatalf("models.Setup err: %v", err)
......
......@@ -2,6 +2,7 @@ package models
import (
"github.com/jinzhu/gorm"
"bwallet/pkg/setting"
)
type Wallet struct {
......@@ -16,7 +17,7 @@ type Wallet struct {
}
func (w Wallet) TableName() string {
return "coin_platform"
return setting.DatabaseSetting.Name_Sources + ".coin_platform"
}
func ExistWalletById(id int) (bool, error) {
......
......@@ -45,7 +45,9 @@ type Database struct {
User string
Password string
Host string
Name string
Name_Sources string
Name_Manage string
Name_Coin string
TablePrefix string
}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment