refactor code

src/main/java/com/peanut/modules/book/controller/UserAddressController.java

@@ -1,133 +1,84 @@
 package com.peanut.modules.book.controller;
 
-import java.util.Arrays;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.peanut.common.utils.PageUtils;
+import com.peanut.common.utils.R;
+import com.peanut.modules.book.entity.UserAddress;
+import com.peanut.modules.book.service.UserAddressService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.*;
+
 import java.util.List;
 import java.util.Map;
 
-import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
-import org.apache.shiro.authz.annotation.RequiresPermissions;
-import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.PathVariable;
-import org.springframework.web.bind.annotation.RequestBody;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RequestParam;
-import org.springframework.web.bind.annotation.RestController;
 
-import com.peanut.modules.book.entity.UserAddressEntity;
-import com.peanut.modules.book.service.UserAddressService;
-import com.peanut.common.utils.PageUtils;
-import com.peanut.common.utils.R;
-
-
-
-/**
- * 
- *
- * @author yl
- * @email yl328572838@163.com
- * @date 2022-10-31 11:20:32
- */
 @RestController
-@RequestMapping("book/useraddress")
+@RequestMapping("book/userAddress")
 public class UserAddressController {
     @Autowired
     private UserAddressService userAddressService;
 
     /**
-     * 列表
+     * 用户地址列表
+     *
+     * @param params param
+     * @return R
      */
     @RequestMapping("/list")
-//    @RequiresPermissions("book:useraddress:list")
-    public R list(@RequestParam Map<String, Object> params){
+    public R list(@RequestParam Map<String, Object> params) {
         PageUtils page = userAddressService.queryPage(params);
-
         return R.ok().put("page", page);
     }
 
 
     /**
-     * 信息
+     * 获取地址信息
+     *
+     * @param addressId address ID
+     * @return R
      */
-    @RequestMapping("/info/{addressid}")
-//    @RequiresPermissions("book:useraddress:info")
-    public R info(@PathVariable("addressid") Integer addressid){
-		UserAddressEntity userAddress = userAddressService.getById(addressid);
-
+    @RequestMapping("/info/{addressId}")
+    public R info(@PathVariable("addressId") Integer addressId) {
+        UserAddress userAddress = userAddressService.getById(addressId);
         return R.ok().put("userAddress", userAddress);
     }
 
     /**
-     * 保存
+     * 保存、更新
+     *
+     * @param userAddress 用户地址
+     * @return R
      */
     @RequestMapping("/save")
-//    @RequiresPermissions("book:useraddress:save")
-    public R save(@RequestBody UserAddressEntity userAddress){
-        Integer isdefault = userAddress.getIsdefault();
-        if (isdefault == 1) {
-            Integer userid = userAddress.getUserid();
-            List<UserAddressEntity> list = userAddressService.list(new QueryWrapper<UserAddressEntity>().eq("userId", userid));
-            for (UserAddressEntity userAddressEntity : list) {
-                if (userAddressEntity.getIsdefault() == 1) {
-                    userAddressEntity.setIsdefault(0);
-                }
-            }
-            userAddressService.updateBatchById(list);
-        }
-        userAddressService.save(userAddress);
+    public R save(@RequestBody UserAddress userAddress) {
+        userAddressService.saveOrUpdate(userAddress);
         return R.ok();
     }
 
     /**
-     * 修改
+     * 删除用户地址
+     *
+     * @param id address ID
+     * @return R
      */
-    @RequestMapping("/update")
-//    @RequiresPermissions("book:useraddress:update")
-    public R update(@RequestBody UserAddressEntity userAddress){
-
-        Integer isdefault = userAddress.getIsdefault();
-
-        if (isdefault == 1) {
-
-            Integer userid = userAddress.getUserid();
-
-            List<UserAddressEntity> list = userAddressService.list(new QueryWrapper<UserAddressEntity>().eq("userId", userid));
-
-            for (UserAddressEntity userAddressEntity : list) {
-                if (userAddressEntity.getIsdefault() == 1) {
-                    userAddressEntity.setIsdefault(0);
-                    userAddressService.updateById(userAddressEntity);
-                }
-            }
-
-        }
-
-		userAddressService.updateById(userAddress);
-
+    @RequestMapping(value = "/delete", method = RequestMethod.GET)
+    public R delete(@RequestParam("id") Integer id) {
+        userAddressService.removeById(id);
         return R.ok();
     }
 
     /**
-     * 删除
-     */
-    @RequestMapping("/delete")
-//    @RequiresPermissions("book:useraddress:delete")
-    public R delete(@RequestBody Integer[] addressids){
-		userAddressService.removeByIds(Arrays.asList(addressids));
-
-        return R.ok();
-    }
-
-    /**
-     * app获取用户 收货地址
+     * 获取用户收货地址列表
+     *
+     * @param userId 用户 ID
+     * @return R
      */
     @RequestMapping("/getUserAddress")
-//    @RequiresPermissions("book:useraddress:delete")
-    public R getUserAddress(@RequestParam("userId") Integer userId){
-
-        List<UserAddressEntity> list = userAddressService.list(new QueryWrapper<UserAddressEntity>().eq("userId", userId).orderByDesc("isDefault"));
-
-        return R.ok().put("list",list);
+    public R getUserAddress(@RequestParam("userId") Integer userId) {
+        QueryWrapper<UserAddress> queryWrapper = new QueryWrapper<>();
+        queryWrapper.eq("user_id", userId);
+        queryWrapper.orderByDesc("is_default");
+        List<UserAddress> userAddressList = userAddressService.list(queryWrapper);
+        return R.ok().put("list", userAddressList);
     }
-
-
 }