ebike-share/ebike-operate/src/main/java/com/cdzy/ebikeoperate/controller/EbikeHelmetQrcodeController.java

package com.cdzy.ebikeoperate.controller;

import com.cdzy.common.model.JsonResult;
import com.cdzy.ebikeoperate.model.dto.request.ReqEbikeHelmetQrcodeBatchDto;
import com.cdzy.ebikeoperate.model.dto.request.ReqEbikeHelmetQrcodeDto;
import com.mybatisflex.core.paginate.Page;
import com.mybatisflex.core.query.QueryWrapper;
import org.springframework.web.bind.annotation.*;
import org.springframework.beans.factory.annotation.Autowired;
import com.cdzy.ebikeoperate.model.pojo.EbikeHelmetQrcode;
import com.cdzy.ebikeoperate.service.EbikeHelmetQrcodeService;

import java.util.List;

/**
 * 头盔二维码 控制层。
 *
 * @author dingchao
 * @since 2025-03-25
 */
@RestController
@RequestMapping("/ebikeHelmetQrcode")
public class EbikeHelmetQrcodeController {

    @Autowired
    private EbikeHelmetQrcodeService ebikeHelmetQrcodeService;

    /**
     * 添加头盔二维码。
     *
     * @param ebikeHelmetQrcode 头盔二维码
     * @return {@code true} 添加成功，{@code false} 添加失败
     */
    @PostMapping("save")
    public JsonResult<?> save(@RequestBody EbikeHelmetQrcode ebikeHelmetQrcode) {
        boolean r = ebikeHelmetQrcodeService.save(ebikeHelmetQrcode);
        return r? JsonResult.success() : JsonResult.failed("添加头盔二维码失败");
    }

    /**
     * 批量添加头盔二维码。
     *
     * @param ebikeHelmetQrcodes 头盔二维码列表
     * @return {@code true} 添加成功，{@code false} 添加失败
     */
    @PostMapping("batchSave")
    public JsonResult<?> batchSave(@RequestBody ReqEbikeHelmetQrcodeBatchDto ebikeHelmetQrcodes) {
        List<EbikeHelmetQrcode> ebikeHelmetQrcodeList = ebikeHelmetQrcodes.getHelmetCodes().stream().map(code -> {
            EbikeHelmetQrcode ebikeHelmetQrcode = new EbikeHelmetQrcode();
            ebikeHelmetQrcode.setComponentProId(ebikeHelmetQrcodes.getComponentProId());
            ebikeHelmetQrcode.setHelmetCode(code);
            return ebikeHelmetQrcode;
        }).toList();
        boolean r = ebikeHelmetQrcodeService.saveBatch(ebikeHelmetQrcodeList);
        return r? JsonResult.success() : JsonResult.failed("添加头盔二维码失败");
    }

    /**
     * 根据主键更新头盔二维码。
     *
     * @param ebikeHelmetQrcode 头盔二维码
     * @return {@code true} 更新成功，{@code false} 更新失败
     */
    @PostMapping("update")
    public JsonResult<?> update(@RequestBody EbikeHelmetQrcode ebikeHelmetQrcode) {
        boolean r = ebikeHelmetQrcodeService.updateById(ebikeHelmetQrcode);
        return r? JsonResult.success() : JsonResult.failed("更新头盔二维码失败");
    }

    /**
     * 批量更新头盔二维码。
     *
     * @param ebikeHelmetQrcodes 头盔二维码列表
     * @return
     */
    @PostMapping("batchUpdate")
    public JsonResult<?> batchUpdate(@RequestBody List<EbikeHelmetQrcode> ebikeHelmetQrcodes) {
        boolean r = ebikeHelmetQrcodeService.updateBatch(ebikeHelmetQrcodes);
        return r? JsonResult.success() : JsonResult.failed("更新头盔二维码失败");
    }

    /**
     * 根据头盔二维码主键获取详细信息。
     *
     * @param id 头盔二维码主键
     * @return 头盔二维码详情
     */
    @GetMapping("getInfo")
    public JsonResult<?> getInfo(@RequestParam(name = "id") String id) {
        EbikeHelmetQrcode r = ebikeHelmetQrcodeService.getById(id);
        return r != null ? JsonResult.success(r) : JsonResult.failed("获取头盔二维码详情失败");
    }

    /**
     * 分页查询头盔二维码。
     *
     * @param reqEbikeHelmetQrcodeDto 分页查询条件
     * @return 分页对象
     */
    @PostMapping("list")
    public JsonResult<?> list(@RequestBody ReqEbikeHelmetQrcodeDto reqEbikeHelmetQrcodeDto) {
        Page<EbikeHelmetQrcode> pageRecords = ebikeHelmetQrcodeService.getPageRecords(reqEbikeHelmetQrcodeDto);
        return JsonResult.success(pageRecords);
    }
    /**
     * 根据头盔编号判断头盔二维码是否存在。
     *
     * @param helmetCode 头盔二维码
     * @return JsonResult 包含查询结果的响应
     */
    @GetMapping("/checkhelmetCodeExistence")
    public JsonResult<?> checkhelmetCodeExistence(@RequestParam(name = "helmetCode") String helmetCode) {
        try {
            QueryWrapper queryWrapper = new QueryWrapper();
            queryWrapper.eq("helmet_code", helmetCode);
            long count = ebikeHelmetQrcodeService.count(queryWrapper);
            return JsonResult.success(count);
        } catch (Exception e) {
            // 记录异常信息
            return JsonResult.failed("查询失败，请稍后再试");
        }
    }
}