[Spring Data MongoDB] - Auto Increment Sequence 초기화하기

📎  [Spring Data MongoDB] - Auto Increment Sequence 초기화하기

 

안녕하세요, 저번 시간에는 Spring Data MongoDB에서 Auto Increment Sequence 를 만드는 방법에 대해 알아보았습니다.

기존 데이터베이스 존재하는 값들을 제거하면서 Auto Increment 값도 초기화 해야 하는 상황이 발생했었습니다.

MySQL에서는 단순 쿼리문으로 Auto Increment를 초기화 할 수 있는데요, MongoDB에서는 구글링을 해봐도 자료가 잘 나오지 않았습니다.(검색 능력이 부족해서...)

따라서 어떻게 초기화를 할 수 있을지 고민을 하다가 두 가지 방법을 찾게되었는데요, 생각외로 훨씬 간단하게 해결할 수 있습니다.

 

 

 

 

1) Sequence 메소드를 증가시키는 함수에서 1로 초기화하기

---

개인적인 생각으로는 좋은 방법은 아닌 것 같지만, 바로 떠오르는 방법이었습니다.

package com.juhyun.shorturl.entity.sequence;

import lombok.RequiredArgsConstructor;
import org.springframework.data.mongodb.core.MongoOperations;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.core.query.Update;
import org.springframework.stereotype.Service;

import java.util.Objects;

import static org.springframework.data.mongodb.core.FindAndModifyOptions.options;
import static org.springframework.data.mongodb.core.query.Criteria.where;

@RequiredArgsConstructor
@Service
public class SequenceGeneratorService {

    private final MongoOperations mongoOperations;

    public Long generateSequence(String key) {
        AutoIncrementSequence counter = mongoOperations.findAndModify(Query.query(where("_id").is(key)),
                new Update().inc("seq", 1), options().returnNew(true).upsert(true), AutoIncrementSequence.class);

        return !Objects.isNull(counter) ? counter.getSeq() : 1;
    }

}

위 SequenceGeneratorService 클래스의 generateSequence() 메소드는 Sequence 객체를 생성하고, 값을 증가시킨 후 리턴하는 함수입니다. 위 함수에서의 리턴 값을 단순히 다음과 같이 바꿔주고 애플리케이션을 실행하고 데이터를 넣어주면 값이 초기화됩니다.

 

return 1L;

 

위와 같이 변경하면 해결은 되지만, 다시 코드를 변경하고 어플리케이션을 실행하는 등 근본적인 해결은 아닌 것 같습니다.

따라서 다음 두 번째 방법에 대해 알아보겠습니다.

 

 

 

 

2) GUI의 sequence collection에서 해당 값을 수정하기

---

전 게시글에서 Auto Increment 를 설정할 때, Entity를 만들어서 관리를 했었습니다.

package com.juhyun.shorturl.entity.sequence;

import lombok.Getter;
import lombok.Setter;
import org.springframework.data.annotation.Id;
import org.springframework.data.mongodb.core.mapping.Document;

@Getter
@Setter
@Document(collection = "auto_sequence")
public class AutoIncrementSequence {

    @Id
    private String id;
    private Long seq;
}

따라서 위 필드는 auto_sequence 라는 이름으로 데이터베이스에 저장이 됩니다.

 

 

따라서 컬렉션에서 seq 값을 변경함으로서 초기화를 할 수 있습니다 😃