fix: meta_sql文件中windows换行符不同问题

This commit is contained in:
meilin.huang
2023-06-17 16:04:21 +08:00
parent 0eca951465
commit 64f8f9a200
6 changed files with 18 additions and 11 deletions

View File

@@ -3,6 +3,7 @@ package application
import (
"embed"
"mayfly-go/pkg/biz"
"mayfly-go/pkg/utils"
"strings"
)
@@ -84,9 +85,10 @@ func GetLocalSql(file, key string) string {
biz.ErrIsNilAppendErr(err, "获取sql meta文件内容失败: %s")
allSql := string(bytes)
sqls := strings.Split(allSql, "\n\n")
sqls := strings.Split(allSql, "---------------------------------------")
var resSql string
for _, sql := range sqls {
sql = utils.StrTrimSpaceAndBr(sql)
// 获取sql第一行的sql备注信息如--MYSQL_TABLE_MA 表信息元数据
info := strings.SplitN(sql, "\n", 2)
// 原始sql即去除第一行的key与备注信息

View File

@@ -9,7 +9,7 @@ WHERE
SELECT
database ()
)
---------------------------------------
--MYSQL_TABLE_INFO 表详细信息
SELECT
table_name tableName,
@@ -25,7 +25,7 @@ WHERE
SELECT
database ()
)
---------------------------------------
--MYSQL_INDEX_INFO 索引信息
SELECT
index_name indexName,
@@ -45,7 +45,7 @@ WHERE
ORDER BY
index_name asc,
SEQ_IN_INDEX asc
---------------------------------------
--MYSQL_COLUMN_MA 列信息元数据
SELECT
table_name tableName,

View File

@@ -11,7 +11,7 @@ WHERE
current_schema ()
)
AND c.reltype > 0
---------------------------------------
--PGSQL_TABLE_INFO 表详细信息
SELECT
obj_description (c.oid) AS "tableComment",
@@ -28,7 +28,7 @@ WHERE
current_schema ()
)
AND c.reltype > 0
---------------------------------------
--PGSQL_INDEX_INFO 表索引信息
SELECT
indexname AS "indexName",
@@ -41,7 +41,7 @@ WHERE
current_schema ()
)
AND tablename = '%s'
---------------------------------------
--PGSQL_COLUMN_MA 表列信息
SELECT
C.relname AS "tableName",
@@ -64,7 +64,7 @@ WHERE
ORDER BY
C.relname DESC,
A.attnum ASC
---------------------------------------
--PGSQL_TABLE_DDL_FUNC 表ddl函数
CREATE OR REPLACE FUNCTION showcreatetable(namespace character varying, tablename character varying)
RETURNS character varying AS

View File

@@ -103,7 +103,6 @@ func (a *Account) Login(rc *req.Ctx) {
})
biz.ErrIsNilAppendErr(err, "otp生成失败: %s")
res["otpUrl"] = key.URL()
// 使用otpSecret充当token进行二次校验
otpSecret = key.Secret()
}
// 缓存otpInfo, 只有双因素校验通过才可返回真正的accessToken

View File

@@ -1,6 +1,8 @@
package otp
import (
"time"
otp_t "github.com/pquerna/otp"
totp_t "github.com/pquerna/otp/totp"
)
@@ -17,3 +19,7 @@ func Validate(code string, secret string) bool {
}
return totp_t.Validate(code, secret)
}
func GenTotpCode(code string, secret string) (string, error) {
return totp_t.GenerateCode(secret, time.Now())
}

View File

@@ -18,11 +18,11 @@ func StrTrim(str string) string {
return strings.Trim(str, " ")
}
// 去除字符串左右空字符与\n换行
// 去除字符串左右空字符与\n\r换行回车符
func StrTrimSpaceAndBr(str string) string {
return strings.TrimFunc(str, func(r rune) bool {
s := string(r)
return s == " " || s == "\n"
return s == " " || s == "\n" || s == "\r"
})
}