crowdsource-backend/src/main/java/edu/nju/dao/ReviewWorkerDao.java

package edu.nju.dao;

import edu.nju.entities.ReviewWorker;
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 ReviewWorkerDao {

    @Autowired
    private MongoOperations mongoOperations;

    public String save(ReviewWorker worker) {
        mongoOperations.save(worker);
        return worker.getId();
    }

    public ReviewWorker getWorkerByOriginalId(String originalId,String examId){
        Query query = new Query();
        query.addCriteria(Criteria.where("original_id").is(originalId).and("exam_id").is(examId));
        List<ReviewWorker> list = mongoOperations.find(query,ReviewWorker.class);
        if(list.size() == 0 || list == null) {return null;}
        return list.get(0);
    }
}