diff --git a/OCRTool/src/main/java/com/bonus/core/OCRRecognition.java b/OCRTool/src/main/java/com/bonus/core/OCRRecognition.java index b127bc6..7ac22b7 100644 --- a/OCRTool/src/main/java/com/bonus/core/OCRRecognition.java +++ b/OCRTool/src/main/java/com/bonus/core/OCRRecognition.java @@ -32,8 +32,15 @@ public class OCRRecognition { logger.info(pythonEnv); logger.info(scriptEnv); logger.info("开始打印从请求里带过来的参数:"); + String typeStr = Integer.toString(recognitionParams.getType()); logger.info(typeStr); + + int type = recognitionParams.getType(); + if (type < 0 || type >= 2) { + logger.info("参数检查错误"); + return ErrorHandler.getErrorInfo(ErrorHandler.RECOGNITION_PARAM_ERROR); + } File file = new File(pythonEnv); if (!file.exists()) { logger.info("pythonEnv不存在");