package com.demo.mo.dao.impl;

import java.util.List;
import java.util.regex.Pattern;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.core.query.Update;
import org.springframework.stereotype.Repository;

import com.demo.mo.dao.UserDao;
import com.demo.mo.entity.User;


@Repository
public class UserDaoImpl implements UserDao {
	
	@Autowired
	private MongoTemplate mongoTemplate;
	
	public MongoTemplate getMongoTemplate() {
		return mongoTemplate;
	}

	public void setMongoTemplate(MongoTemplate mongoTemplate) {
		this.mongoTemplate = mongoTemplate;
	}

	@Override
	public void insert(User user) {
		getMongoTemplate().insert(user);
	}

	@Override
	public User findOne(String id) {
		return getMongoTemplate().findOne(new Query(Criteria.where("uid").is(id)),User.class);
	}

	@Override
	public List<User> findAll() {
		return getMongoTemplate().findAll(User.class);
	}

	@Override
	public List<User> findByRegex(String regex) {
		Pattern pattern = Pattern.compile(regex,Pattern.CASE_INSENSITIVE);   
		Criteria criteria = new Criteria("name").regex(pattern.toString());   
		return getMongoTemplate().find(new Query(criteria), User.class);
	}

	@Override
	public void removeOne(User us) {
		getMongoTemplate().remove(us);
	}

	@Override
	public void removeAll() {
		List<User> list = this.findAll();   
		if(list != null){   
			for(User user : list){   
				getMongoTemplate().remove(user);   
			}   
		}   
	}

	@Override
	public void findAndModify(User u) {
		Query query = new Query();
		Criteria criteria = new Criteria();
		criteria.where("id").is(u.getUid());
		query.addCriteria(criteria);
		Update update = new Update();
		update.set("name",u.getName());
		update.set("age", u.getAge());
		getMongoTemplate().updateFirst(query,update, User.class);
	}

}