用户登录限制接口

doc-admin/src/main/java/com/doc/web/controller/system/SysUserExpandController.java

@@ -19,15 +19,14 @@ import java.util.List;
 
 /**
  * 用户登录限制Controller
- * 
+ *
  * @author wukai
  * @date 2023-10-11
  */
-@Api(tags="用户登录限制")
+@Api(tags = "用户登录限制")
 @RestController
-@RequestMapping("/system/expand")
-public class SysUserExpandController extends BaseController
-{
+@RequestMapping("/system/user/expand")
+public class SysUserExpandController extends BaseController {
     @Resource
     private ISysUserExpandService sysUserExpandService;
 
@@ -37,8 +36,7 @@ public class SysUserExpandController extends BaseController
     @ApiOperation("查询用户登录限制列表")
     @PreAuthorize("@ss.hasPermi('system:expand:list')")
     @GetMapping("/list")
-    public TableDataInfo list(SysUserExpand sysUserExpand)
-    {
+    public TableDataInfo list(SysUserExpand sysUserExpand) {
         startPage();
         List<SysUserExpand> list = sysUserExpandService.selectSysUserExpandList(sysUserExpand);
         return getDataTable(list);
@@ -51,8 +49,7 @@ public class SysUserExpandController extends BaseController
     @PreAuthorize("@ss.hasPermi('system:expand:export')")
     @Log(title = "用户登录限制", businessType = BusinessType.EXPORT)
     @PostMapping("/export")
-    public void export(HttpServletResponse response, SysUserExpand sysUserExpand)
-    {
+    public void export(HttpServletResponse response, SysUserExpand sysUserExpand) {
         List<SysUserExpand> list = sysUserExpandService.selectSysUserExpandList(sysUserExpand);
         ExcelUtil<SysUserExpand> util = new ExcelUtil<SysUserExpand>(SysUserExpand.class);
         util.exportExcel(response, list, "用户登录限制数据");
@@ -64,8 +61,7 @@ public class SysUserExpandController extends BaseController
     @ApiOperation("获取用户登录限制详细信息")
     @PreAuthorize("@ss.hasPermi('system:expand:query')")
     @GetMapping(value = "/{userId}")
-    public AjaxResult getInfo(@PathVariable("userId") Long userId)
-    {
+    public AjaxResult getInfo(@PathVariable("userId") Long userId) {
         return success(sysUserExpandService.selectSysUserExpandByUserId(userId));
     }
 
@@ -76,8 +72,7 @@ public class SysUserExpandController extends BaseController
     @PreAuthorize("@ss.hasPermi('system:expand:add')")
     @Log(title = "用户登录限制", businessType = BusinessType.INSERT)
     @PostMapping
-    public AjaxResult add(@RequestBody SysUserExpand sysUserExpand)
-    {
+    public AjaxResult add(@RequestBody SysUserExpand sysUserExpand) {
         return toAjax(sysUserExpandService.insertSysUserExpand(sysUserExpand));
     }
 
@@ -88,8 +83,7 @@ public class SysUserExpandController extends BaseController
     @PreAuthorize("@ss.hasPermi('system:expand:edit')")
     @Log(title = "用户登录限制", businessType = BusinessType.UPDATE)
     @PutMapping
-    public AjaxResult edit(@RequestBody SysUserExpand sysUserExpand)
-    {
+    public AjaxResult edit(@RequestBody SysUserExpand sysUserExpand) {
         return toAjax(sysUserExpandService.updateSysUserExpand(sysUserExpand));
     }
 
@@ -99,9 +93,8 @@ public class SysUserExpandController extends BaseController
     @ApiOperation("删除用户登录限制")
     @PreAuthorize("@ss.hasPermi('system:expand:remove')")
     @Log(title = "用户登录限制", businessType = BusinessType.DELETE)
-	@DeleteMapping("/{userIds}")
-    public AjaxResult remove(@PathVariable Long[] userIds)
-    {
+    @DeleteMapping("/{userIds}")
+    public AjaxResult remove(@PathVariable Long[] userIds) {
         return toAjax(sysUserExpandService.deleteSysUserExpandByUserIds(userIds));
     }
 }

doc-system/src/main/java/com/doc/system/service/impl/SysUserExpandServiceImpl.java

@@ -49,7 +49,7 @@ public class SysUserExpandServiceImpl implements ISysUserExpandService {
      */
     @Override
     public int insertSysUserExpand(SysUserExpand sysUserExpand) {
-            return sysUserExpandMapper.insertSysUserExpand(sysUserExpand);
+        return sysUserExpandMapper.insertSysUserExpand(sysUserExpand);
     }
 
     /**
@@ -60,7 +60,15 @@ public class SysUserExpandServiceImpl implements ISysUserExpandService {
      */
     @Override
     public int updateSysUserExpand(SysUserExpand sysUserExpand) {
-        return sysUserExpandMapper.updateSysUserExpand(sysUserExpand);
+        SysUserExpand u = sysUserExpandMapper.selectSysUserExpandByUserId(sysUserExpand.getUserId());
+        /**
+         * 查询用户扩展信息是否存在，如果不存在则创建
+         */
+        if (u != null) {
+            return sysUserExpandMapper.updateSysUserExpand(sysUserExpand);
+        } else {
+            return sysUserExpandMapper.insertSysUserExpand(sysUserExpand);
+        }
     }
 
     /**