From ae9fe08a1f9fb73fa31cf3bdb737f5a5d6eb8ced Mon Sep 17 00:00:00 2001 From: TsMask <340112800@qq.com> Date: Thu, 29 Feb 2024 21:00:28 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E6=9F=A5=E8=AF=A2SQLmap=E5=B1=9E?= =?UTF-8?q?=E6=80=A7=E5=80=BC=E8=BD=AC=E6=8D=A2=E7=A9=BA=E5=AD=97=E7=AC=A6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- features/sys_config/service/repo_sys_config.go | 12 ++---------- features/sys_dict_data/repo/repo_sys_dict_data.go | 6 +----- features/sys_dict_type/repo/repo_sys_dict_type.go | 6 +----- features/sys_menu/service/repo_sys_menu.go | 6 +----- features/sys_role/service/repo_sys_role.go | 6 +----- features/sys_user/service/repo_sys_user.go | 6 +----- src/modules/monitor/repository/sys_job.impl.go | 7 ++----- .../network_element/repository/ne_host.impl.go | 7 ++----- .../network_element/repository/ne_host_cmd.impl.go | 7 ++----- src/modules/system/repository/sys_config.impl.go | 13 +++---------- src/modules/system/repository/sys_dept.impl.go | 6 +----- src/modules/system/repository/sys_dict_data.impl.go | 7 ++----- src/modules/system/repository/sys_dict_type.impl.go | 7 ++----- src/modules/system/repository/sys_menu.impl.go | 6 +----- src/modules/system/repository/sys_post.impl.go | 7 ++----- src/modules/system/repository/sys_role.impl.go | 7 ++----- src/modules/system/repository/sys_user.impl.go | 7 ++----- 17 files changed, 28 insertions(+), 95 deletions(-) diff --git a/features/sys_config/service/repo_sys_config.go b/features/sys_config/service/repo_sys_config.go index 5b8ecdb4..7a4f0ec9 100644 --- a/features/sys_config/service/repo_sys_config.go +++ b/features/sys_config/service/repo_sys_config.go @@ -181,11 +181,7 @@ func (r *RepoSysConfig) SelectConfigValueByKey(configKey string) string { return "" } if len(results) > 0 { - v, ok := results[0]["str"].(string) - if ok { - return v - } - return "" + return fmt.Sprint(results[0]["str"]) } return "" } @@ -230,11 +226,7 @@ func (r *RepoSysConfig) CheckUniqueConfig(sysConfig model.SysConfig) string { return "" } if len(results) > 0 { - v, ok := results[0]["str"].(string) - if ok { - return v - } - return "" + return fmt.Sprint(results[0]["str"]) } return "" } diff --git a/features/sys_dict_data/repo/repo_sys_dict_data.go b/features/sys_dict_data/repo/repo_sys_dict_data.go index 6ae2631c..346dc5fa 100644 --- a/features/sys_dict_data/repo/repo_sys_dict_data.go +++ b/features/sys_dict_data/repo/repo_sys_dict_data.go @@ -215,11 +215,7 @@ func (r *RepoSysDictData) CheckUniqueDictData(sysDictData model.SysDictData) str return "" } if len(results) > 0 { - v, ok := results[0]["str"].(string) - if ok { - return v - } - return "" + return fmt.Sprint(results[0]["str"]) } return "" } diff --git a/features/sys_dict_type/repo/repo_sys_dict_type.go b/features/sys_dict_type/repo/repo_sys_dict_type.go index 5e88e992..55c3d070 100644 --- a/features/sys_dict_type/repo/repo_sys_dict_type.go +++ b/features/sys_dict_type/repo/repo_sys_dict_type.go @@ -227,11 +227,7 @@ func (r *RepoSysDictType) CheckUniqueDictType(sysDictType model.SysDictType) str return "" } if len(results) > 0 { - v, ok := results[0]["str"].(string) - if ok { - return v - } - return "" + return fmt.Sprint(results[0]["str"]) } return "" } diff --git a/features/sys_menu/service/repo_sys_menu.go b/features/sys_menu/service/repo_sys_menu.go index eae567d7..c06395aa 100644 --- a/features/sys_menu/service/repo_sys_menu.go +++ b/features/sys_menu/service/repo_sys_menu.go @@ -469,11 +469,7 @@ func (r *RepoSysMenu) CheckUniqueMenu(sysMenu model.SysMenu) string { return "" } if len(results) > 0 { - v, ok := results[0]["str"].(string) - if ok { - return v - } - return "" + return fmt.Sprint(results[0]["str"]) } return "" } diff --git a/features/sys_role/service/repo_sys_role.go b/features/sys_role/service/repo_sys_role.go index 5997d67c..6e6bda02 100644 --- a/features/sys_role/service/repo_sys_role.go +++ b/features/sys_role/service/repo_sys_role.go @@ -356,11 +356,7 @@ func (r *RepoSysRole) CheckUniqueRole(sysRole model.SysRole) string { return "" } if len(results) > 0 { - v, ok := results[0]["str"].(string) - if ok { - return v - } - return "" + return fmt.Sprint(results[0]["str"]) } return "" } diff --git a/features/sys_user/service/repo_sys_user.go b/features/sys_user/service/repo_sys_user.go index 1dff7c89..c31bda41 100644 --- a/features/sys_user/service/repo_sys_user.go +++ b/features/sys_user/service/repo_sys_user.go @@ -572,11 +572,7 @@ func (r *RepoSysUser) CheckUniqueUser(sysUser sysUserModel.SysUser) string { log.Errorf("query err %v", err) } if len(results) > 0 { - v, ok := results[0]["str"].(string) - if ok { - return v - } - return "" + return fmt.Sprint(results[0]["str"]) } return "" } diff --git a/src/modules/monitor/repository/sys_job.impl.go b/src/modules/monitor/repository/sys_job.impl.go index 196601fb..4ad09104 100644 --- a/src/modules/monitor/repository/sys_job.impl.go +++ b/src/modules/monitor/repository/sys_job.impl.go @@ -1,6 +1,7 @@ package repository import ( + "fmt" "strings" "time" @@ -209,11 +210,7 @@ func (r *SysJobImpl) CheckUniqueJob(sysJob model.SysJob) string { return "" } if len(results) > 0 { - v, ok := results[0]["str"].(string) - if ok { - return v - } - return "" + return fmt.Sprint(results[0]["str"]) } return "" } diff --git a/src/modules/network_element/repository/ne_host.impl.go b/src/modules/network_element/repository/ne_host.impl.go index 3a5775c9..41649d91 100644 --- a/src/modules/network_element/repository/ne_host.impl.go +++ b/src/modules/network_element/repository/ne_host.impl.go @@ -1,6 +1,7 @@ package repository import ( + "fmt" "strings" "time" @@ -231,11 +232,7 @@ func (r *NeHostImpl) CheckUniqueNeHost(neHost model.NeHost) string { return "" } if len(results) > 0 { - v, ok := results[0]["str"].(string) - if ok { - return v - } - return "" + return fmt.Sprint(results[0]["str"]) } return "" } diff --git a/src/modules/network_element/repository/ne_host_cmd.impl.go b/src/modules/network_element/repository/ne_host_cmd.impl.go index b831dad2..c5d7f51c 100644 --- a/src/modules/network_element/repository/ne_host_cmd.impl.go +++ b/src/modules/network_element/repository/ne_host_cmd.impl.go @@ -1,6 +1,7 @@ package repository import ( + "fmt" "strings" "time" @@ -199,11 +200,7 @@ func (r *NeHostCmd) CheckUniqueGroupTitle(neHostCmd model.NeHostCmd) string { return "" } if len(results) > 0 { - v, ok := results[0]["str"].(string) - if ok { - return v - } - return "" + return fmt.Sprint(results[0]["str"]) } return "" } diff --git a/src/modules/system/repository/sys_config.impl.go b/src/modules/system/repository/sys_config.impl.go index 90d0d3d8..b0960530 100644 --- a/src/modules/system/repository/sys_config.impl.go +++ b/src/modules/system/repository/sys_config.impl.go @@ -1,6 +1,7 @@ package repository import ( + "fmt" "strings" "time" @@ -185,11 +186,7 @@ func (r *SysConfigImpl) SelectConfigValueByKey(configKey string) string { return "" } if len(results) > 0 { - v, ok := results[0]["str"].(string) - if ok { - return v - } - return "" + return fmt.Sprint(results[0]["str"]) } return "" } @@ -234,11 +231,7 @@ func (r *SysConfigImpl) CheckUniqueConfig(sysConfig model.SysConfig) string { return "" } if len(results) > 0 { - v, ok := results[0]["str"].(string) - if ok { - return v - } - return "" + return fmt.Sprint(results[0]["str"]) } return "" } diff --git a/src/modules/system/repository/sys_dept.impl.go b/src/modules/system/repository/sys_dept.impl.go index 550d0eb8..6d754ed1 100644 --- a/src/modules/system/repository/sys_dept.impl.go +++ b/src/modules/system/repository/sys_dept.impl.go @@ -226,11 +226,7 @@ func (r *SysDeptImpl) CheckUniqueDept(sysDept model.SysDept) string { return "" } if len(results) > 0 { - v, ok := results[0]["str"].(string) - if ok { - return v - } - return "" + return fmt.Sprint(results[0]["str"]) } return "" } diff --git a/src/modules/system/repository/sys_dict_data.impl.go b/src/modules/system/repository/sys_dict_data.impl.go index 055762a2..221de3fd 100644 --- a/src/modules/system/repository/sys_dict_data.impl.go +++ b/src/modules/system/repository/sys_dict_data.impl.go @@ -1,6 +1,7 @@ package repository import ( + "fmt" "strings" "time" @@ -219,11 +220,7 @@ func (r *SysDictDataImpl) CheckUniqueDictData(sysDictData model.SysDictData) str return "" } if len(results) > 0 { - v, ok := results[0]["str"].(string) - if ok { - return v - } - return "" + return fmt.Sprint(results[0]["str"]) } return "" } diff --git a/src/modules/system/repository/sys_dict_type.impl.go b/src/modules/system/repository/sys_dict_type.impl.go index 0dd63b25..19a74016 100644 --- a/src/modules/system/repository/sys_dict_type.impl.go +++ b/src/modules/system/repository/sys_dict_type.impl.go @@ -1,6 +1,7 @@ package repository import ( + "fmt" "strings" "time" @@ -231,11 +232,7 @@ func (r *SysDictTypeImpl) CheckUniqueDictType(sysDictType model.SysDictType) str return "" } if len(results) > 0 { - v, ok := results[0]["str"].(string) - if ok { - return v - } - return "" + return fmt.Sprint(results[0]["str"]) } return "" } diff --git a/src/modules/system/repository/sys_menu.impl.go b/src/modules/system/repository/sys_menu.impl.go index 72975c0e..55efdd53 100644 --- a/src/modules/system/repository/sys_menu.impl.go +++ b/src/modules/system/repository/sys_menu.impl.go @@ -468,11 +468,7 @@ func (r *SysMenuImpl) CheckUniqueMenu(sysMenu model.SysMenu) string { return "" } if len(results) > 0 { - v, ok := results[0]["str"].(string) - if ok { - return v - } - return "" + return fmt.Sprint(results[0]["str"]) } return "" } diff --git a/src/modules/system/repository/sys_post.impl.go b/src/modules/system/repository/sys_post.impl.go index 184c98b4..0df71e05 100644 --- a/src/modules/system/repository/sys_post.impl.go +++ b/src/modules/system/repository/sys_post.impl.go @@ -1,6 +1,7 @@ package repository import ( + "fmt" "strings" "time" @@ -314,11 +315,7 @@ func (r *SysPostImpl) CheckUniquePost(sysPost model.SysPost) string { logger.Errorf("query err %v", err) } if len(results) > 0 { - v, ok := results[0]["str"].(string) - if ok { - return v - } - return "" + return fmt.Sprint(results[0]["str"]) } return "" } diff --git a/src/modules/system/repository/sys_role.impl.go b/src/modules/system/repository/sys_role.impl.go index 719ee4e0..2aa07dbf 100644 --- a/src/modules/system/repository/sys_role.impl.go +++ b/src/modules/system/repository/sys_role.impl.go @@ -1,6 +1,7 @@ package repository import ( + "fmt" "strings" "time" @@ -371,11 +372,7 @@ func (r *SysRoleImpl) CheckUniqueRole(sysRole model.SysRole) string { return "" } if len(results) > 0 { - v, ok := results[0]["str"].(string) - if ok { - return v - } - return "" + return fmt.Sprint(results[0]["str"]) } return "" } diff --git a/src/modules/system/repository/sys_user.impl.go b/src/modules/system/repository/sys_user.impl.go index 05d47c94..e54628b8 100644 --- a/src/modules/system/repository/sys_user.impl.go +++ b/src/modules/system/repository/sys_user.impl.go @@ -1,6 +1,7 @@ package repository import ( + "fmt" "strings" "time" @@ -557,11 +558,7 @@ func (r *SysUserImpl) CheckUniqueUser(sysUser model.SysUser) string { logger.Errorf("query err %v", err) } if len(results) > 0 { - v, ok := results[0]["str"].(string) - if ok { - return v - } - return "" + return fmt.Sprint(results[0]["str"]) } return "" }