<!-- 百度AI SDK -->
<dependency>
    <groupId>com.baidu.aip</groupId>
    <artifactId>java-sdk</artifactId>
    <version>4.16.19</version>
    <exclusions>
        <exclusion>
            <groupId>org.slf4j</groupId>
            <artifactId>slf4j-simple</artifactId>
        </exclusion>
    </exclusions>
</dependency>
<!-- Spring Boot核心依賴 -->
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-web</artifactId>
</dependency>
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-redis</artifactId>
</dependency>
<!-- 數(shù)據(jù)訪問 -->
<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-boot-starter</artifactId>
    <version>3.5.1</version>
</dependency>
<dependency>
    <groupId>mysql</groupId>
    <artifactId>mysql-connector-java</artifactId>
    <scope>runtime</scope>
</dependency>
<!-- 工具類 -->
<dependency>
    <groupId>org.projectlombok</groupId>
    <artifactId>lombok</artifactId>
    <optional>true</optional>
</dependency>
<dependency>
    <groupId>cn.hutool</groupId>
    <artifactId>hutool-all</artifactId>
    <version>5.2.4</version>
</dependency>
<!-- 文件處理 -->
<dependency>
    <groupId>commons-fileupload</groupId>
    <artifactId>commons-fileupload</artifactId>
    <version>1.3.1</version>
</dependency>

@Data
@TableName("in_out_record")
public class InOutRecordEntity implements Serializable {
    private static final long serialVersionUID = 1L;
    @TableId(value = "in_out_record_id", type = IdType.AUTO)
    private Integer inOutRecordId;
    @TableField("person_id")
    private Integer personId;
    @TableField("community_id")
    private Integer communityId;
    @TableField("in_time")
    private LocalDateTime inTime;
    @TableField("out_time")
    private LocalDateTime outTime;
    @TableField("in_pic")
    private String inPic;
    @TableField("out_pic")
    private String outPic;
}

@Data
public class InOutForm {
    private Long page;
    private Long limit;
    private String userName;
    @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
    private LocalDateTime startDate;
    @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
    private LocalDateTime endDate;
}

@Data
public class InOutRecordVO {
    @TableId(value = "in_out_record_id", type = IdType.AUTO)
    private Integer inOutRecordId;
    private Integer personId;
    private Integer communityId;
    private LocalDateTime inTime;
    private LocalDateTime outTime;
    private String inPic;
    private String outPic;
    // 擴展字段，用于展示
    private String userName;
    private String communityName;
    private String termName;
    private String houseNo;
}

@Component
@Slf4j
public class BaiduAiUtils {
    @Value("${baidu.face.appId}")
    private String APP_ID;
    @Value("${baidu.face.apiKey}")
    private String API_KEY;
    @Value("${baidu.face.secretKey}")
    private String SECRET_KEY;
    @Value("${baidu.face.imageType}")
    private String IMAGE_TYPE;
    @Value("${baidu.face.groupId}")
    private String groupId;
    private AipFace client;
    private HashMap<String, Object> options = new HashMap<>();
    public BaiduAiUtils() {
        // 設(shè)置圖像質(zhì)量控制
        options.put("quality_control", "NORMAL");
        // 設(shè)置活體檢測控制級別
        options.put("liveness_control", "LOW");
    }
    @PostConstruct
    public void init() {
        // 初始化百度AI客戶端
        client = new AipFace(APP_ID, API_KEY, SECRET_KEY);
    }
    /**
     * 人臉檢測（檢查是否有且僅有一張人臉）
     */
    public Boolean faceCheck(String image) {
        JSONObject res = client.detect(image, IMAGE_TYPE, options);
        log.info("detect result ：{}", res);
        if (res.has("error_code") && res.getInt("error_code") == 0) {
            JSONObject resultObject = res.getJSONObject("result");
            Integer faceNum = resultObject.getInt("face_num");
            return faceNum == 1;
        }
        return false;
    }
    /**
     * 人臉?biāo)阉鳎ㄆヅ溆脩簦?
     */
    public String faceSearch(String image) {
        JSONObject res = client.search(image, IMAGE_TYPE, groupId, options);
        log.info("search result ：{}", res);
        if (res.has("error_code") && res.getInt("error_code") == 0) {
            JSONObject result = res.getJSONObject("result");
            JSONArray userList = result.getJSONArray("user_list");
            if (userList.length() > 0) {
                JSONObject user = userList.getJSONObject(0);
                double score = user.getDouble("score");
                // 置信度大于80分認(rèn)為匹配成功
                if (score > 80) {
                    return user.getString("user_id");
                }
            }
        }
        return null;
    }
}

@Service
public class InOutRecordServiceImpl extends ServiceImpl<InOutRecordMapper, InOutRecordEntity> implements InOutRecordService {
    @Autowired
    private InOutRecordMapper inOutRecordMapper;
    @Autowired
    private PersonMapper personMapper;
    @Override
    public InOutPageListVO getInOutList(InOutForm form) {
        Page<InOutRecordEntity> page = new Page<>(form.getPage(), form.getLimit());
        QueryWrapper<InOutRecordEntity> queryWrapper = new QueryWrapper<>();
        // 時間范圍查詢
        if (form.getStartDate() != null && form.getEndDate() != null) {
            queryWrapper.between("in_time", form.getStartDate(), form.getEndDate());
        }
        // 如果需要按用戶名查詢，可以在這里添加關(guān)聯(lián)查詢條件
        Page<InOutRecordEntity> pages = inOutRecordMapper.selectPage(page, queryWrapper);
        List<InOutRecordVO> inOutRecordVOList = new ArrayList<>();
        // 轉(zhuǎn)換為VO對象并補充關(guān)聯(lián)信息
        for(InOutRecordEntity entity : pages.getRecords()){
            InOutRecordVO vo = new InOutRecordVO();
            BeanUtils.copyProperties(entity, vo);
            PersonEntity person = personMapper.selectById(entity.getPersonId());
            if (person != null) {
                vo.setUserName(person.getUserName());
                vo.setHouseNo(person.getHouseNo());
            }
            // 獲取小區(qū)名稱
            vo.setCommunityName(personMapper.selectCommunityNameByID(entity.getCommunityId()));
            inOutRecordVOList.add(vo);
        }
        // 封裝分頁結(jié)果
        InOutPageListVO result = new InOutPageListVO();
        result.setRecords(inOutRecordVOList);
        result.setTotalCount(pages.getTotal());
        result.setPageSize(pages.getSize());
        result.setTotalPage(pages.getPages());
        result.setCurrPage(pages.getCurrent());
        return result;
    }
    @Override
    public InOutRecordEntity findLatestRecord(Integer personId) {
        QueryWrapper<InOutRecordEntity> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("person_id", personId)
                .orderByDesc("in_time")
                .last("LIMIT 1");
        return this.getOne(queryWrapper);
    }
}

@RestController
@RequestMapping("/sys/inOut")
public class InOutRecordController {
    @Autowired
    private BaiduAiUtils baiduAiUtils;
    @Autowired
    private PersonService personService;
    @Autowired
    private InOutRecordService recordService;
    @Value("${file.upload-dir}")
    private String uploadDir;
    /**
     * 人臉識別接口
     */
    @PostMapping("/add")
    public Result add(@RequestBody FaceForm faceForm) {
        // 提取Base64圖像數(shù)據(jù)
        String fileBase64 = faceForm.getFileBase64();
        if (fileBase64.contains(",")) {
            fileBase64 = fileBase64.split(",")[1];
        }
        // 1. 檢測人臉
        boolean hasValidFace = baiduAiUtils.faceCheck(fileBase64);
        if (!hasValidFace) {
            return Result.error("人臉檢測失敗");
        }
        // 2. 人臉?biāo)阉髌ヅ溆脩?
        String userId = baiduAiUtils.faceSearch(fileBase64);
        if (userId == null) {
            return Result.ok().put("data", "人員信息不存在").put("status", "fail");
        }
        // 3. 查詢用戶信息
        int personId;
        try {
            personId = Integer.parseInt(userId);
        } catch (NumberFormatException e) {
            return Result.error("用戶ID格式錯誤");
        }
        PersonEntity person = personService.getById(personId);
        if (person == null) {
            return Result.ok().put("data", "人員信息不存在").put("status", "fail");
        }
        try {
            // 4. 保存圖片到本地
            String fileName = System.currentTimeMillis() + ".png";
            String filePath = Paths.get(uploadDir, fileName).toString();
            // 確保目錄存在
            File dir = new File(uploadDir);
            if (!dir.exists()) {
                dir.mkdirs();
            }
            // 解碼并保存圖片
            byte[] imageBytes = Base64.getDecoder().decode(fileBase64);
            Files.write(Paths.get(filePath), imageBytes);
            // 構(gòu)建圖片URL
            String fullUrl = "http://localhost:8080/photos/" + fileName;
            // 5. 查找最近記錄判斷是入場還是出場
            InOutRecordEntity latestRecord = recordService.findLatestRecord(personId);
            if (latestRecord == null || latestRecord.getOutTime() != null) {
                // 入場記錄
                InOutRecordEntity newRecord = new InOutRecordEntity();
                newRecord.setCommunityId(faceForm.getCommunityId());
                newRecord.setPersonId(personId);
                newRecord.setInTime(LocalDateTime.now());
                newRecord.setInPic(fullUrl);
                recordService.save(newRecord);
                return Result.ok().put("data", person.getUserName() + "進入小區(qū)").put("status", "success");
            } else {
                // 出場記錄
                latestRecord.setOutTime(LocalDateTime.now());
                latestRecord.setOutPic(fullUrl);
                recordService.updateById(latestRecord);
                return Result.ok().put("data", person.getUserName() + "離開小區(qū)").put("status", "success");
            }
        } catch (Exception e) {
            e.printStackTrace();
            return Result.error("操作失敗: " + e.getMessage());
        }
    }
    /**
     * 出入記錄查詢接口
     */
    @GetMapping("/list")
    public Result list(InOutForm form) {
        // 獲取分頁數(shù)據(jù)
        InOutPageListVO pageListVO = inOutRecordService.getInOutList(form);
        // 構(gòu)建返回結(jié)構(gòu)
        Map<String, Object> pageListMap = new HashMap<>();
        pageListMap.put("totalCount", pageListVO.getTotalCount());
        pageListMap.put("pageSize", pageListVO.getPageSize());
        pageListMap.put("totalPage", pageListVO.getTotalPage());
        pageListMap.put("currPage", pageListVO.getCurrPage());
        pageListMap.put("list", pageListVO.getRecords());
        Map<String, Object> dataMap = new HashMap<>();
        dataMap.put("pageList", pageListMap);
        return Result.ok().put("data", dataMap);
    }
}

{
    "communityId": 2,
    "extName": "png",
    "fileBase64": "iVBORw0KGgoAAAANSUhEUgAAAoAAAAHgCAYAAAA10dzkAAAAAXNSR0IArs4c6QAAIABJREFUeF7sXe2OHblu3BljTX1lQUgyxEvv9HMQ5yIZq239/8B8gtdpbgl/6cAAAAASUVORK5CYII="
}

{
    "msg": "操作成功",
    "code": 200,
    "data": "張三進入小區(qū)",
    "status": "success"
}

{
    "msg": "操作成功",
    "code": 200,
    "data": "人員信息不存在",
    "status": "fail"
}

{
    "page": 1,
    "limit": 10,
    "userName": "張三",
    "startDate": "2023-07-20 12:59:54",
    "endDate": "2023-07-20 23:00:00"
}

{
    "msg": "操作成功",
    "code": 200,
    "data": {
        "pageList": {
            "totalCount": 1,
            "pageSize": 10,
            "totalPage": 1,
            "currPage": 1,
            "list": [
                {
                    "inOutRecordId": 44,
                    "inTime": "2023-07-19 16:51:55",
                    "outTime": "2023-07-19 16:52:07",
                    "inPic": "http://localhost:8181/villegePic/face/47b49187-a5e9-486a-b8ac-4409710b3323.png",
                    "outPic": "http://localhost:8181/villegePic/face/4cbfb2b9-a691-4d0a-a4d4-4bf602cb33ac.png",
                    "communityName": "棲海澐頌",
                    "termName": "8棟",
                    "houseNo": "802",
                    "userName": "麗麗"
                }
            ]
        }
    }
}