MongoDB Spring Data创建索引工具

package com.man.tools.mongodb.factory;

import cn.hutool.core.annotation.AnnotationUtil;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.theone.date.util.DateUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.index.CompoundIndex;
import org.springframework.data.mongodb.core.index.CompoundIndexes;
import org.springframework.data.mongodb.core.index.Index;
import org.springframework.data.mongodb.core.index.IndexInfo;

import java.util.Date;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

public class MongoJpaCollectionFactory {
    protected static final Logger LOGGER = LoggerFactory.getLogger(MongoJpaCollectionFactory.class);

    
    public static String getCollectionName(String docName, Date date) {
        return docName + DateUtil.format(date, DateUtil.yyyyMM);
    }

    public static void creatIndex(MongoTemplate mongoTemplate, String collectionName, Class aClass) {
        Set indexSet = new HashSet<>();
        for (IndexInfo indexInfo : mongoTemplate.indexOps(collectionName).getIndexInfo()) {
            indexSet.add(indexInfo.getName());
        }
        CompoundIndex compoundIndex1 = AnnotationUtil.getAnnotation(aClass, CompoundIndex.class);
        if (compoundIndex1 != null) {
            index(collectionName, mongoTemplate, indexSet, compoundIndex1);
            return;
        }
        CompoundIndexes compoundIndexes = AnnotationUtil.getAnnotation(aClass, CompoundIndexes.class);
        for (CompoundIndex compoundIndex : compoundIndexes.value()) {
            index(collectionName, mongoTemplate, indexSet, compoundIndex);
        }
    }

    private static void index(String collectionName, MongoTemplate mongoTemplate, Set indexSet, CompoundIndex compoundIndex) {
        if (indexSet.contains(compoundIndex.name())) {
            return;
        }
        mongoTemplate.indexOps(collectionName).ensureIndex(getIndex(compoundIndex));
    }

    private static Index getIndex(CompoundIndex compoundIndex) {
        String name = compoundIndex.name();
        String def = compoundIndex.def();
        Index index = new Index();
        index.named(name);
        index.background();
        JSONObject indexMap = JSONUtil.parseObj(def);
        for (Map.Entry keyOrder : indexMap.entrySet()) {
            String key = keyOrder.getKey();
            String order = String.valueOf(keyOrder.getValue());
            index.on(key, "1".equals(order) ? Sort.Direction.ASC : Sort.Direction.DESC);
        }
        return index;
    }
}

package com.man.tools.mongodb;

import org.springframework.data.annotation.Id;
import org.springframework.data.mongodb.core.index.CompoundIndex;
import org.springframework.data.mongodb.core.index.CompoundIndexes;
import org.springframework.data.mongodb.core.mapping.document;

@document(collection = "TestDoc")
@CompoundIndexes({
        @CompoundIndex(name = "age_-1_name_1", def = "{age:-1,name:1}", background = true)
})
public class TestDoc{
    @Id
    private String id;
    
    private Integer age;
    
    private String name;

    public String getId() {
        return id;
    }

    public void setId(String id) {
        this.id = id;
    }

    public Integer getAge() {
        return age;
    }

    public void setAge(Integer age) {
        this.age = age;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }
}

@RunWith(SpringJUnit4ClassRunner.class)
//启动Spring
@SpringBootTest(classes = Application.class)
public class baseTest {
	@Resource 
	private MongoTemplate mongoTemplate;
	@Test
	public void testIndexOp() {
	    // 按年分表
		ArrayList> list = CollectionUtil.newArrayList(TestDoc.class);
		for (Class aClass : list) {
			String collectionNamebase = mongoTemplate.getCollectionName(aClass);
			String collectionName =collectionNamebase + "2020";
			if (mongoTemplate.collectionExists(collectionName)) {
				MongoJpaCollectionFactory.creatIndex(mongoTemplate, collectionName, aClass);
			}
		}
	}
}