com.github.shoothzj.sdk.distribute.impl.mongo.repo.MongoLockRepoImpl Maven / Gradle / Ivy

package com.github.shoothzj.sdk.distribute.impl.mongo.repo;

import com.github.shoothzj.sdk.distribute.impl.mongo.dao.MongoLock;
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.data.mongodb.core.query.Update;
import org.springframework.stereotype.Service;

@Service
public class MongoLockRepoImpl implements MongoLockRepoCustom {

    @Autowired
    private MongoOperations mongoOperations;

    @Override
    public void findAndModify(String scene, String oldId, String uniqId, long expireTime) {
        Query query = new Query();
        query.addCriteria(Criteria.where("scene").is(scene)).addCriteria(Criteria.where("uniqId").is(oldId));

        Update update = new Update();
        update.set("scene", scene);
        update.set("uniqId", uniqId);
        update.set("expireTime", expireTime);

        mongoOperations.findAndModify(query, update, MongoLock.class);
    }

}