Custom table name

iyear · Jul 6, 2021 · f85de1b · f85de1b
1 parent 0ce6822
commit f85de1b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/config/config.go b/config/config.go
@@ -40,6 +40,7 @@ type MysqlConfig struct {
 	User     string
 	Password string
 	DB       string
+	Table    string
 }
 
 func InitConfig() {
@@ -71,6 +72,7 @@ func InitConfig() {
 		User:     viper.GetString("mysql.user"),
 		Password: viper.GetString("mysql.password"),
 		DB:       viper.GetString("mysql.database"),
+		Table:    viper.GetString("mysql.table"),
 	}
 	viper.WatchConfig()
 	viper.OnConfigChange(func(e fsnotify.Event) {

diff --git a/model/client.go b/model/client.go
@@ -2,6 +2,7 @@ package model
 
 import (
 	"github.com/guonaihong/gout"
+	"github.com/iyear/E5SubBot/config"
 	"github.com/pkg/errors"
 	"github.com/tidwall/gjson"
 	"net/url"
@@ -30,6 +31,9 @@ const (
 	scope       string = "openid offline_access mail.read user.read"
 )
 
+func (c *Client) TableName() string {
+	return config.Mysql.Table
+}
 func NewClient(clientId string, clientSecret string) *Client {
 	return &Client{
 		ClientId:     clientId,