import com.github.mongobee.changeset.ChangeSet; //导入依赖的package包/类
/**
 * update v5: {@link UserDocument} has changed; String-Field 'email' was replaced with custom Type 'Email'.
 * find all user with an email String-value, replace String-Type with Custom-Type and update all UserDocuments.
 * if email-Field is Empty or null, you don't have to update user.
 *
 * @since V6
 */
@ChangeSet(order = "006", id = "updateUserChangeEmailFromStringToCustomClass", author = "admin")
public void updateUserChangeEmailField(final MongoTemplate template) {
  final Criteria isEmptyCriteria = new Criteria().orOperator(Criteria.where("email").is(""), Criteria.where("email").is(null));
  while (true) {
    final UserDocument result = template.findAndModify(new Query(isEmptyCriteria), Update.update("email", new Email()), UserDocument.class);
    if (result == null)
      break;
  }

  /**
   * if email not null -> Field will be cast to specific class and updates will create correct entries
   */
  // final Criteria isNotEmptyCriteria = new Criteria().orOperator(Criteria.where("email").ne(""), Criteria.where("email").ne(null));
  // final List<UserDocument> userDocumentList = template.find(new Query(isNotEmptyCriteria), UserDocument.class);

}
 

import com.github.mongobee.changeset.ChangeSet; //导入依赖的package包/类
/**
 * update 6: {@link UserDocument} has changed; 'name' and 'surname' will be concat to 'name'.
 * for each every user document get 'name' and 'surname', concat them, update 'name', remove field surname and update document.
 *
 * @since V7
 */
@ChangeSet(order = "008", id = "updateUserChangeNameAndSurnameToName", author = "admin")
public void updateUserChangeNameAndSurnametoName(final MongoTemplate template) {
  final DBCollection userCollection = template.getCollection("user");

  final Iterator<DBObject> cursor = userCollection.find();
  while (cursor.hasNext()) {
    final DBObject current = cursor.next();

    final Object nameObj = current.get("name");
    final Object surnameObj = current.get("surname");
    final String updateName = (nameObj != null ? nameObj.toString() : "") + " " + (surnameObj != null ? surnameObj.toString() : "");

    final BasicDBObject updateQuery = new BasicDBObject();
    updateQuery.append("$set", new BasicDBObject("name", updateName));
    updateQuery.append("$unset", new BasicDBObject("surname", ""));

    final BasicDBObject searchQuery = new BasicDBObject();
    searchQuery.put("_id", current.get("_id"));

    userCollection.update(searchQuery, updateQuery);
  }
}
 

import com.github.mongobee.changeset.ChangeSet; //导入依赖的package包/类
@ChangeSet(order = "02", author = "initiator", id = "02-addUsers")
public void addUsers(MongoTemplate mongoTemplate) {
    Authority adminAuthority = new Authority();
    adminAuthority.setName(AuthoritiesConstants.ADMIN);
    Authority userAuthority = new Authority();
    userAuthority.setName(AuthoritiesConstants.USER);

    User adminUser = new User();
    adminUser.setId("user-0");
    adminUser.setLogin("admin");
    adminUser.setPassword("$2a$10$gSAhZrxMllrbgj/kkK9UceBPpChGWJA7SYIb1Mqo.n5aNLq1/oRrC");
    adminUser.setFirstName("admin");
    adminUser.setLastName("Administrator");
    adminUser.setEmail("[email protected]");
    adminUser.setActivated(true);
    adminUser.setLangKey("en");
    adminUser.getAuthorities().add(userAuthority);
    adminUser.setCreatedBy(adminUser.getLogin());
    adminUser.setCreatedDate(ZonedDateTime.now());
    adminUser.setLastModifiedBy(adminUser.getLogin());
    adminUser.setLastModifiedDate(ZonedDateTime.now());
    mongoTemplate.save(adminUser);
}
 

import com.github.mongobee.changeset.ChangeSet; //导入依赖的package包/类
/**
 * update v3: {@link UserDocument} has changed; an embedded Document 'address' was added.
 *
 * @since V4
 */
@ChangeSet(order = "004", id = "addAddressToUser", author = "admin")
public void updateUserAddAddress(final MongoTemplate template) {
  final List<UserDocument> userList = template.find(Query.query(Criteria.where("email").is("[email protected]")), UserDocument.class);
  userList.stream()
      .map(userDocument -> {
        userDocument.getUserAddress()
            .addAll(
                Stream.iterate(0, n -> n + 1)
                    .limit(new Random().nextInt(5))
                    .map(it -> new Address("Hamburger City", "Random Muster Straße " + it, it + "5678"))
                    .collect(Collectors.toList()));
        return userDocument;
      })
      .forEach(template::save);
}
 

import com.github.mongobee.changeset.ChangeSet; //导入依赖的package包/类
@ChangeSet(order = "01",
           author = "developer",
           id = "01-addUsers")
public void addUsers(final DB db) {
    final DBCollection userCollection = db.getCollection(User.COLLECTION_NAME);

    userCollection.insert(BasicDBObjectBuilder
                                  .start()
                                  .add(FIELD_NAME_ID, new ObjectId("590f86d92449343841cc2c3f"))
                                  .add(User.FIELD_NAME_FIRST_NAME, "User")
                                  .add(User.FIELD_NAME_LAST_NAME, "One")
                                  .add(User.FIELD_NAME_EMAIL, "[email protected]")
                                  .get());

    userCollection.insert(BasicDBObjectBuilder
                                  .start()
                                  .add(FIELD_NAME_ID, new ObjectId("590f86d92449343841cc2c40"))
                                  .add(User.FIELD_NAME_FIRST_NAME, "User")
                                  .add(User.FIELD_NAME_LAST_NAME, "Two")
                                  .add(User.FIELD_NAME_EMAIL, "[email protected]")
                                  .get());
}
 

import com.github.mongobee.changeset.ChangeSet; //导入依赖的package包/类
@ChangeSet(order = "03", author = "initiator", id = "03-addSocialUserConnection")
public void addSocialUserConnection(DB db) {
    DBCollection socialUserConnectionCollection = db.getCollection("jhi_social_user_connection");
    socialUserConnectionCollection.createIndex(BasicDBObjectBuilder
            .start("user_id", 1)
            .add("provider_id", 1)
            .add("provider_user_id", 1)
            .get(),
        "user-prov-provusr-idx", true);
}
 

import com.github.mongobee.changeset.ChangeSet; //导入依赖的package包/类
@ChangeSet(order = "01", author = "initiator", id = "01-addAuthorities")
public void addAuthorities(DB db) {
    DBCollection authorityCollection = db.getCollection("jhi_authority");
    authorityCollection.insert(
        BasicDBObjectBuilder.start()
            .add("_id", "ROLE_ADMIN")
            .get());
    authorityCollection.insert(
        BasicDBObjectBuilder.start()
            .add("_id", "ROLE_USER")
            .get());
}
 

import com.github.mongobee.changeset.ChangeSet; //导入依赖的package包/类
@ChangeSet(order = "02", author = "initiator", id = "02-addUsers")
public void addUsers(DB db) {
    DBCollection usersCollection = db.getCollection("jhi_user");
    usersCollection.createIndex("login");
    usersCollection.createIndex("email");
    usersCollection.insert(BasicDBObjectBuilder.start()
        .add("_id", "user-0")
        .add("login", "system")
        .add("password", "*************")
        .add("first_name", "")
        .add("last_name", "System")
        .add("email", "[email protected]")
        .add("activated", "true")
        .add("lang_key", "en")
        .add("created_by", "system")
        .add("created_date", new Date())
        .add("authorities", authoritiesAdminAndUser)
        .get()
    );
    usersCollection.insert(BasicDBObjectBuilder.start()
        .add("_id", "user-1")
        .add("login", "anonymousUser")
        .add("password", "*************")
        .add("first_name", "Anonymous")
        .add("last_name", "User")
        .add("email", "[email protected]")
        .add("activated", "true")
        .add("lang_key", "en")
        .add("created_by", "system")
        .add("created_date", new Date())
        .add("authorities", new Map[]{})
        .get()
    );
}
 

import com.github.mongobee.changeset.ChangeSet; //导入依赖的package包/类
@ChangeSet(order = "04", author = "user", id = "04-addAuthorities-2")
public void addAuthorities2(DB db) {
    DBCollection authorityCollection = db.getCollection("jhi_authority");
    authorityCollection.insert(
        BasicDBObjectBuilder.start()
            .add("_id", "ROLE_MANAGER")
            .get());
    authorityCollection.insert(
        BasicDBObjectBuilder.start()
            .add("_id", "ROLE_SUPPORT")
            .get());
}
 

import com.github.mongobee.changeset.ChangeSet; //导入依赖的package包/类
@ChangeSet(order = "05", author = "user", id = "05-addAuthorities-3")
public void addAuthorities3(DB db) {
    DBCollection authorityCollection = db.getCollection("jhi_authority");
    authorityCollection.insert(
        BasicDBObjectBuilder.start()
            .add("_id", "ROLE_ACTUATOR")
            .get());
}
 

import com.github.mongobee.changeset.ChangeSet; //导入依赖的package包/类
/**
 * update v1: {@link UserDocument} has changed; Date-Field 'creationDate' was added and initialized with current date.
 *
 * @since V2
 */
@ChangeSet(order = "002", id = "updateUserAddCreationDate", author = "admin")
public void updateUserAndSetDate(final MongoTemplate template) {
  template.findAll(UserDocument.class, "user")
      .stream()
      .map(userDocument -> {
        if (userDocument.getCreateDate() == null)
          userDocument.setCreateDate(new Date());
        return userDocument;
      })
      .forEach(template::save);
}
 

import com.github.mongobee.changeset.ChangeSet; //导入依赖的package包/类
/**
 * update v2: {@link UserDocument} has changed; Date-Field 'modificationDate' was added and initialized with currentDate.
 * Also an Integer-Field 'version' was added and initialized with 2.
 * <p>
 * Keep in mind, that you have to add {@link org.springframework.data.annotation.Version} annotation, after you have created and initialized the Field.
 *
 * @since V3
 */
@ChangeSet(order = "003", id = "updateUserAddVersionAndModificationDate", author = "admin")
public void updateUserAndSetVersionAndModificationDate(final MongoTemplate template) {
  template.findAll(UserDocument.class, "user")
      .stream()
      .map(userDocument -> {
        userDocument.setModificationDate(new Date());
        userDocument.setVersion(2); // workaround; set version value manually; add annotation afterwards
        return userDocument;
      })
      .forEach(template::save);
}
 

import com.github.mongobee.changeset.ChangeSet; //导入依赖的package包/类
/**
 * update 8: {@Link UserDocument} has changed; add String-Fields 'createdBy' and 'modifiedBy' and initialize with String 'admin'
 *
 * @since V8
 */
@ChangeSet(order = "009", id = "updateUserAddCreatedByAndLastModifiedBy", author = "admin")
public void updateUserAddCreatedByAndLastModifiedBy(final MongoTemplate template) {
  final Update update = new Update();
  update.set("createdBy", "admin");
  update.set("modifiedBy", "admin");
  template.updateMulti(null, update, UserDocument.class);
}
 

import com.github.mongobee.changeset.ChangeSet; //导入依赖的package包/类
@ChangeSet(order = "01", author = "initiator", id = "01-addAuthorities")
public void addAuthorities(MongoTemplate mongoTemplate) {
    Authority adminAuthority = new Authority();
    adminAuthority.setName(AuthoritiesConstants.ADMIN);
    Authority userAuthority = new Authority();
    userAuthority.setName(AuthoritiesConstants.USER);
    mongoTemplate.save(adminAuthority);
    mongoTemplate.save(userAuthority);
}
 

import com.github.mongobee.changeset.ChangeSet; //导入依赖的package包/类
@ChangeSet(author = "initiator", id = "03-addSocialUserConnection", order = "03")
public void addSocialUserConnection(DB db) {
    DBCollection socialUserConnectionCollection = db.getCollection("jhi_social_user_connection");
    socialUserConnectionCollection.createIndex(BasicDBObjectBuilder
            .start("user_id", 1)
            .add("provider_id", 1)
            .add("provider_user_id", 1)
            .get(),
        "user-prov-provusr-idx", true);
}
 

import com.github.mongobee.changeset.ChangeSet; //导入依赖的package包/类
@ChangeSet(order = "01", author = "initiator", id = "01-addAuthorities")
public void addAuthorities(MongoTemplate mongoTemplate) {
	Authority adminAuthority = new Authority();
	adminAuthority.setName(AuthoritiesConstants.ADMIN);
	Authority userAuthority = new Authority();
	userAuthority.setName(AuthoritiesConstants.USER);
	mongoTemplate.save(adminAuthority);
	mongoTemplate.save(userAuthority);
}
 

import com.github.mongobee.changeset.ChangeSet; //导入依赖的package包/类
public boolean isRunAlwaysChangeSet(Method changesetMethod){
  if (changesetMethod.isAnnotationPresent(ChangeSet.class)){
    ChangeSet annotation = changesetMethod.getAnnotation(ChangeSet.class);
    return annotation.runAlways();
  } else {
    return false;
  }
}
 

import com.github.mongobee.changeset.ChangeSet; //导入依赖的package包/类
public ChangeEntry createChangeEntry(Method changesetMethod){
  if (changesetMethod.isAnnotationPresent(ChangeSet.class)){
    ChangeSet annotation = changesetMethod.getAnnotation(ChangeSet.class);

    return new ChangeEntry(
        annotation.id(),
        annotation.author(),
        new Date(),
        changesetMethod.getDeclaringClass().getName(),
        changesetMethod.getName());
  } else {
    return null;
  }
}
 

import com.github.mongobee.changeset.ChangeSet; //导入依赖的package包/类
private List<Method> filterChangeSetAnnotation(List<Method> allMethods) throws MongobeeChangeSetException {
  final Set<String> changeSetIds = new HashSet<>();
  final List<Method> changesetMethods = new ArrayList<>();
  for (final Method method : allMethods) {
    if (method.isAnnotationPresent(ChangeSet.class)) {
      String id = method.getAnnotation(ChangeSet.class).id();
      if (changeSetIds.contains(id)) {
        throw new MongobeeChangeSetException(String.format("Duplicated changeset id found: '%s'", id));
      }
      changeSetIds.add(id);
      changesetMethods.add(method);
    }
  }
  return changesetMethods;
}
 

import com.github.mongobee.changeset.ChangeSet; //导入依赖的package包/类
@ChangeSet(order = "02", author = "initiator", id = "02-addUsers")
public void addUsers(DB db) {
    DBCollection usersCollection = db.getCollection("jhi_user");
    usersCollection.createIndex("login");
    usersCollection.createIndex("email");
    usersCollection.insert(BasicDBObjectBuilder.start()
        .add("_id", "user-0")
        .add("login", "system")
        .add("password", "$2a$10$mE.qmcV0mFU5NcKh73TZx.z4ueI/.bDWbj0T1BYyqP481kGGarKLG")
        .add("first_name", "")
        .add("last_name", "System")
        .add("email", "[email protected]")
        .add("activated", "true")
        .add("lang_key", "en")
        .add("created_by", "system")
        .add("created_date", new Date())
        .add("authorities", authoritiesAdminAndUser)
        .get()
    );
    usersCollection.insert(BasicDBObjectBuilder.start()
        .add("_id", "user-1")
        .add("login", "anonymousUser")
        .add("password", "$2a$10$j8S5d7Sr7.8VTOYNviDPOeWX8KcYILUVJBsYV83Y5NtECayypx9lO")
        .add("first_name", "Anonymous")
        .add("last_name", "User")
        .add("email", "[email protected]")
        .add("activated", "true")
        .add("lang_key", "en")
        .add("created_by", "system")
        .add("created_date", new Date())
        .add("authorities", new Map[]{})
        .get()
    );
    usersCollection.insert(BasicDBObjectBuilder.start()
        .add("_id", "user-2")
        .add("login", "admin")
        .add("password", "$2a$10$gSAhZrxMllrbgj/kkK9UceBPpChGWJA7SYIb1Mqo.n5aNLq1/oRrC")
        .add("first_name", "admin")
        .add("last_name", "Administrator")
        .add("email", "[email protected]")
        .add("activated", "true")
        .add("lang_key", "en")
        .add("created_by", "system")
        .add("created_date", new Date())
        .add("authorities", authoritiesAdminAndUser)
        .get()
    );
    usersCollection.insert(BasicDBObjectBuilder.start()
        .add("_id", "user-3")
        .add("login", "user")
        .add("password", "$2a$10$VEjxo0jq2YG9Rbk2HmX9S.k1uZBGYUHdUcid3g/vfiEl7lwWgOH/K")
        .add("first_name", "")
        .add("last_name", "User")
        .add("email", "[email protected]")
        .add("activated", "true")
        .add("lang_key", "en")
        .add("created_by", "system")
        .add("created_date", new Date())
        .add("authorities", authoritiesUser)
        .get()
    );
}