Sfoglia il codice sorgente

上传获得similarScore的方法

MengyangDuan 5 anni fa
parent
commit
cc6ba23c33

+ 13 - 0
src/main/java/edu/nju/dao/BugSimilarScoreDao.java

@@ -3,8 +3,12 @@ package edu.nju.dao;
 import edu.nju.entities.BugSimilarScore;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.data.mongodb.core.MongoOperations;
+import org.springframework.data.mongodb.core.query.Criteria;
+import org.springframework.data.mongodb.core.query.Query;
 import org.springframework.stereotype.Repository;
 
+import java.util.List;
+
 @Repository
 public class BugSimilarScoreDao {
 
@@ -14,4 +18,13 @@ public class BugSimilarScoreDao {
     public void save(BugSimilarScore score) {
         mongoOperations.save(score);
     }
+
+    public BugSimilarScore findById(String id) {
+        Query query = new Query();
+        query.addCriteria(Criteria.where("_id").is(id));
+        List<BugSimilarScore> list = mongoOperations.find(query, BugSimilarScore.class);
+        if(list != null && list.size() != 0) {return list.get(0);}
+        return null;
+    }
+
 }

+ 5 - 0
src/main/java/edu/nju/service/AnalyzeService.java

@@ -86,6 +86,11 @@ public class AnalyzeService {
 		}
 	}
 
+	public BugSimilarScore getSimilarScore(String id) {
+		BugSimilarScore bss = bugSimilarScoreDao.findById(id);
+		return bss;
+	}
+
 	public boolean saveGradeByWorker(String id, String workerId, int grade) {
 		try {
 			bsdao.save(new BugScore(id, grade, 0));