Kaynağa Gözat

增加接收功图诊断请求队列

chenwen 4 ay önce
ebeveyn
işleme
4e07eb620a

+ 13 - 0
src/main/java/com/hb/proj/api/controller/APIController.java

@@ -26,6 +26,7 @@ import com.hb.proj.model.Hyperparam;
 import com.hb.proj.model.TrainLog;
 import com.hb.proj.model.TrainParam;
 import com.hb.proj.rep.DiagramSampleService;
+import com.hb.proj.third.RedisMsgProcessor;
 import com.hb.proj.utils.JacksonUtils;
 import com.hb.proj.utils.RespVO;
 import com.hb.proj.utils.RespVOBuilder;
@@ -48,6 +49,18 @@ public class APIController {
 	@Autowired
 	private XGBoostEngine   xgbEngine;
 	
+	@Autowired
+	private RedisMsgProcessor  msgProcessor;
+	
+	/**
+	 * 获得批量诊断队列的当前大小
+	 * @return
+	 */
+	@RequestMapping("/buffer/size")
+	public RespVO<Object> getBufferSize(){
+		return RespVOBuilder.ok(msgProcessor.getQueueSize());
+	}
+	
 	
 	/**
 	 * 获得功图绘制数据(样本功图)

+ 3 - 3
src/main/java/com/hb/proj/diagnose/FeatureExtractor.java

@@ -67,7 +67,7 @@ public class FeatureExtractor {
 		
 		float glbRange=glbYMax-glbYMin;
 		
-		logger.info("glbYMax:{}, glbYMin:{},glbRange:{}",glbYMax,glbYMin,glbRange);
+		//logger.info("glbYMax:{}, glbYMin:{},glbRange:{}",glbYMax,glbYMin,glbRange);
 		
 		//起始方向1:up;0:down
 		boolean startTag=(glbXMax-disps.get(0).floatValue()) > (disps.get(0).floatValue()-0);
@@ -99,9 +99,9 @@ public class FeatureExtractor {
 		}
 		
 		
-		logger.info("距离:{}",midMinDiffs);
+		//logger.info("距离:{}",midMinDiffs);
 		
-		logger.info("斜率:{},{}",gradients.size(),gradients);
+		//logger.info("斜率:{},{}",gradients.size(),gradients);
 		
 		DiagramFeatureSample  sample=diagram.build();
 		sample.setStartTag(startTag);