modified 统一代码风格
This commit is contained in:
@@ -42,22 +42,22 @@ func CheckFileMd5(md5 string) (err error, uploads []model.ExaSimpleUploader, isD
|
||||
func MergeFileMd5(md5 string, fileName string) (err error) {
|
||||
finishDir := "./finish/"
|
||||
dir := "./chunk/" + md5
|
||||
//如果文件上传成功 不做后续操作 通知成功即可
|
||||
// 如果文件上传成功 不做后续操作 通知成功即可
|
||||
if !errors.Is(global.GVA_DB.First(&model.ExaSimpleUploader{}, "identifier = ? AND is_done = ?", md5, true).Error, gorm.ErrRecordNotFound) {
|
||||
return nil
|
||||
}
|
||||
|
||||
//打开切片文件夹
|
||||
// 打开切片文件夹
|
||||
rd, err := ioutil.ReadDir(dir)
|
||||
_ = os.MkdirAll(finishDir, os.ModePerm)
|
||||
//创建目标文件
|
||||
// 创建目标文件
|
||||
fd, err := os.OpenFile(finishDir+fileName, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0644)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
//关闭文件
|
||||
// 关闭文件
|
||||
defer fd.Close()
|
||||
//将切片文件按照顺序写入
|
||||
// 将切片文件按照顺序写入
|
||||
for k := range rd {
|
||||
content, _ := ioutil.ReadFile(dir + "/" + fileName + strconv.Itoa(k+1))
|
||||
_, err = fd.Write(content)
|
||||
@@ -70,7 +70,7 @@ func MergeFileMd5(md5 string, fileName string) (err error) {
|
||||
return err
|
||||
}
|
||||
err = global.GVA_DB.Transaction(func(tx *gorm.DB) error {
|
||||
//删除切片信息
|
||||
// 删除切片信息
|
||||
if err = tx.Delete(&model.ExaSimpleUploader{}, "identifier = ? AND is_done = ?", md5, false).Error; err != nil {
|
||||
fmt.Println(err)
|
||||
return err
|
||||
@@ -89,6 +89,6 @@ func MergeFileMd5(md5 string, fileName string) (err error) {
|
||||
return nil
|
||||
})
|
||||
|
||||
err = os.RemoveAll(dir) //清除切片
|
||||
err = os.RemoveAll(dir) // 清除切片
|
||||
return err
|
||||
}
|
||||
|
||||
@@ -44,8 +44,8 @@ func GetRedisJWT(userName string) (err error, redisJWT string) {
|
||||
//@author: [piexlmax](https://github.com/piexlmax)
|
||||
//@function: SetRedisJWT
|
||||
//@description: jwt存入redis并设置过期时间
|
||||
//@param: userName string
|
||||
//@return: err error, redisJWT string
|
||||
//@param: jwt string, userName string
|
||||
//@return: err error
|
||||
|
||||
func SetRedisJWT(jwt string, userName string) (err error) {
|
||||
// 此处过期时间等于jwt过期时间
|
||||
|
||||
@@ -59,7 +59,7 @@ func CopyAuthority(copyInfo response.SysAuthorityCopyResponse) (err error, autho
|
||||
//@function: UpdateAuthority
|
||||
//@description: 更改一个角色
|
||||
//@param: auth model.SysAuthority
|
||||
//@return:err error, authority model.SysAuthority
|
||||
//@return: err error, authority model.SysAuthority
|
||||
|
||||
func UpdateAuthority(auth model.SysAuthority) (err error, authority model.SysAuthority) {
|
||||
err = global.GVA_DB.Where("authority_id = ?", auth.AuthorityId).First(&model.SysAuthority{}).Updates(&auth).Error
|
||||
@@ -126,7 +126,7 @@ func GetAuthorityInfo(auth model.SysAuthority) (err error, sa model.SysAuthority
|
||||
//@function: SetDataAuthority
|
||||
//@description: 设置角色资源权限
|
||||
//@param: auth model.SysAuthority
|
||||
//@return:error
|
||||
//@return: error
|
||||
|
||||
func SetDataAuthority(auth model.SysAuthority) error {
|
||||
var s model.SysAuthority
|
||||
|
||||
@@ -96,7 +96,7 @@ func PreviewTemp(autoCode model.AutoCodeStruct) (map[string]string, error) {
|
||||
//@function: CreateTemp
|
||||
//@description: 创建代码
|
||||
//@param: model.AutoCodeStruct
|
||||
//@return: error
|
||||
//@return: err error
|
||||
|
||||
func CreateTemp(autoCode model.AutoCodeStruct) (err error) {
|
||||
dataList, fileList, needMkdir, err := getNeedList(&autoCode)
|
||||
@@ -186,9 +186,8 @@ func GetAllTplFile(pathName string, fileList []string) ([]string, error) {
|
||||
//@author: [piexlmax](https://github.com/piexlmax)
|
||||
//@function: GetTables
|
||||
//@description: 获取数据库的所有表名
|
||||
//@param: pathName string
|
||||
//@param: fileList []string
|
||||
//@return: []string, error
|
||||
//@param: dbName string
|
||||
//@return: err error, TableNames []request.TableReq
|
||||
|
||||
func GetTables(dbName string) (err error, TableNames []request.TableReq) {
|
||||
err = global.GVA_DB.Raw("select table_name as table_name from information_schema.tables where table_schema = ?", dbName).Scan(&TableNames).Error
|
||||
@@ -198,9 +197,7 @@ func GetTables(dbName string) (err error, TableNames []request.TableReq) {
|
||||
//@author: [piexlmax](https://github.com/piexlmax)
|
||||
//@function: GetDB
|
||||
//@description: 获取数据库的所有数据库名
|
||||
//@param: pathName string
|
||||
//@param: fileList []string
|
||||
//@return: []string, error
|
||||
//@return: err error, DBNames []request.DBReq
|
||||
|
||||
func GetDB() (err error, DBNames []request.DBReq) {
|
||||
err = global.GVA_DB.Raw("SELECT SCHEMA_NAME AS `database` FROM INFORMATION_SCHEMA.SCHEMATA;").Scan(&DBNames).Error
|
||||
@@ -210,9 +207,8 @@ func GetDB() (err error, DBNames []request.DBReq) {
|
||||
//@author: [piexlmax](https://github.com/piexlmax)
|
||||
//@function: GetDB
|
||||
//@description: 获取指定数据库和指定数据表的所有字段名,类型值等
|
||||
//@param: pathName string
|
||||
//@param: fileList []string
|
||||
//@return: []string, error
|
||||
//@param: tableName string, dbName string
|
||||
//@return: err error, Columns []request.ColumnReq
|
||||
|
||||
func GetColumn(tableName string, dbName string) (err error, Columns []request.ColumnReq) {
|
||||
err = global.GVA_DB.Raw("SELECT COLUMN_NAME column_name,DATA_TYPE data_type,CASE DATA_TYPE WHEN 'longtext' THEN c.CHARACTER_MAXIMUM_LENGTH WHEN 'varchar' THEN c.CHARACTER_MAXIMUM_LENGTH WHEN 'double' THEN CONCAT_WS( ',', c.NUMERIC_PRECISION, c.NUMERIC_SCALE ) WHEN 'decimal' THEN CONCAT_WS( ',', c.NUMERIC_PRECISION, c.NUMERIC_SCALE ) WHEN 'int' THEN c.NUMERIC_PRECISION WHEN 'bigint' THEN c.NUMERIC_PRECISION ELSE '' END AS data_type_long,COLUMN_COMMENT column_comment FROM INFORMATION_SCHEMA.COLUMNS c WHERE table_name = ? AND table_schema = ?", tableName, dbName).Scan(&Columns).Error
|
||||
@@ -269,7 +265,7 @@ func addAutoMoveFile(data *tplData) {
|
||||
//@function: CreateApi
|
||||
//@description: 自动创建api数据,
|
||||
//@param: a *model.AutoCodeStruct
|
||||
//@return: error
|
||||
//@return: err error
|
||||
|
||||
func AutoCreateApi(a *model.AutoCodeStruct) (err error) {
|
||||
var apiList = []model.SysApi{
|
||||
|
||||
@@ -34,7 +34,7 @@ func DeleteBaseMenu(id float64) (err error) {
|
||||
//@function: UpdateBaseMenu
|
||||
//@description: 更新路由
|
||||
//@param: menu model.SysBaseMenu
|
||||
//@return:err error
|
||||
//@return: err error
|
||||
|
||||
func UpdateBaseMenu(menu model.SysBaseMenu) (err error) {
|
||||
var oldMenu model.SysBaseMenu
|
||||
|
||||
@@ -54,7 +54,7 @@ func GetSysDictionaryDetail(id uint) (err error, sysDictionaryDetail model.SysDi
|
||||
//@function: GetSysDictionaryDetailInfoList
|
||||
//@description: 分页获取字典详情列表
|
||||
//@param: info request.SysDictionaryDetailSearch
|
||||
//@return: err error
|
||||
//@return: err error, list interface{}, total int64
|
||||
|
||||
func GetSysDictionaryDetailInfoList(info request.SysDictionaryDetailSearch) (err error, list interface{}, total int64) {
|
||||
limit := info.PageSize
|
||||
|
||||
@@ -18,7 +18,7 @@ import (
|
||||
//@author: [songzhibin97](https://github.com/songzhibin97)
|
||||
//@function: writeConfig
|
||||
//@description: 回写配置
|
||||
//@param:
|
||||
//@param: viper *viper.Viper, mysql config.Mysql
|
||||
//@return: error
|
||||
|
||||
func writeConfig(viper *viper.Viper, mysql config.Mysql) error {
|
||||
@@ -67,8 +67,8 @@ func initDB(InitDBFunctions ...model.InitDBFunc) (err error) {
|
||||
//@author: [songzhibin97](https://github.com/songzhibin97)
|
||||
//@function: InitDB
|
||||
//@description: 创建数据库并初始化
|
||||
//@param: authorityId string
|
||||
//@return: err error, treeMap map[string][]model.SysMenu
|
||||
//@param: conf request.InitDB
|
||||
//@return: error
|
||||
|
||||
func InitDB(conf request.InitDB) error {
|
||||
BaseMysql := config.Mysql{
|
||||
@@ -118,7 +118,7 @@ func InitDB(conf request.InitDB) error {
|
||||
SkipInitializeWithVersion: false, // 根据版本自动配置
|
||||
}
|
||||
if db, err := gorm.Open(mysql.New(mysqlConfig), &gorm.Config{DisableForeignKeyConstraintWhenMigrating: true}); err != nil {
|
||||
//global.GVA_LOG.Error("MySQL启动异常", zap.Any("err", err))
|
||||
//global.GVA_LOG.Error("MySQL启动异常!", zap.Any("err", err))
|
||||
//os.Exit(0)
|
||||
//return nil
|
||||
_ = writeConfig(global.GVA_VP, BaseMysql)
|
||||
|
||||
@@ -87,7 +87,7 @@ func getBaseChildrenList(menu *model.SysBaseMenu, treeMap map[string][]model.Sys
|
||||
//@function: AddBaseMenu
|
||||
//@description: 添加基础路由
|
||||
//@param: menu model.SysBaseMenu
|
||||
//@return: err error
|
||||
//@return: error
|
||||
|
||||
func AddBaseMenu(menu model.SysBaseMenu) error {
|
||||
if !errors.Is(global.GVA_DB.Where("name = ?", menu.Name).First(&model.SysBaseMenu{}).Error, gorm.ErrRecordNotFound) {
|
||||
|
||||
@@ -42,15 +42,15 @@ func GetServerInfo() (server *utils.Server, err error) {
|
||||
var s utils.Server
|
||||
s.Os = utils.InitOS()
|
||||
if s.Cpu, err = utils.InitCPU(); err != nil {
|
||||
global.GVA_LOG.Error("func utils.InitCPU() Failed!", zap.String("err", err.Error()))
|
||||
global.GVA_LOG.Error("func utils.InitCPU() Failed", zap.String("err", err.Error()))
|
||||
return &s, err
|
||||
}
|
||||
if s.Rrm, err = utils.InitRAM(); err != nil {
|
||||
global.GVA_LOG.Error("func utils.InitRAM() Failed!", zap.String("err", err.Error()))
|
||||
global.GVA_LOG.Error("func utils.InitRAM() Failed", zap.String("err", err.Error()))
|
||||
return &s, err
|
||||
}
|
||||
if s.Disk, err = utils.InitDisk(); err != nil {
|
||||
global.GVA_LOG.Error("func utils.InitDisk() Failed!", zap.String("err", err.Error()))
|
||||
global.GVA_LOG.Error("func utils.InitDisk() Failed", zap.String("err", err.Error()))
|
||||
return &s, err
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user