• Home
• com.antwerkz.critter
• critter-maven
• 2.0.0
• source code
• UpdaterBuilder.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.antwerkz.critter.UpdaterBuilder Maven / Gradle / Ivy

package com.antwerkz.critter;

import com.mongodb.WriteConcern;
import static java.lang.String.format;
import org.jboss.forge.roaster.Roaster;
import org.jboss.forge.roaster.model.source.FieldSource;
import org.jboss.forge.roaster.model.source.JavaClassSource;
import org.jboss.forge.roaster.model.source.MethodSource;
import org.mongodb.morphia.annotations.Id;
import org.mongodb.morphia.query.UpdateOperations;
import org.mongodb.morphia.query.UpdateResults;

public class UpdaterBuilder {
  public UpdaterBuilder(final CritterClass critterClass, final JavaClassSource criteriaClass) {
    final String type = critterClass.getName() + "Updater";
    final MethodSource getUpdater = criteriaClass.addMethod()
        .setPublic()
        .setName("getUpdater")
        .setReturnType(type);
    getUpdater
        .setBody(format("return new %s();", getUpdater.getReturnType()));

    final JavaClassSource updater = Roaster.create(JavaClassSource.class)
        .setPublic()
        .setName(type);

    criteriaClass.addImport(UpdateOperations.class);
    criteriaClass.addImport(UpdateResults.class);
    criteriaClass.addImport(WriteConcern.class);

    final FieldSource updateOperations = updater.addField()
        .setType(format("UpdateOperations<%s>", critterClass.getName()));
    updateOperations.setName("updateOperations");

    updater.addMethod()
        .setPublic()
        .setName(type)
        .setConstructor(true)
        .setBody(format("updateOperations = ds.createUpdateOperations(%s.class);", critterClass.getName()));

    updater.addMethod()
        .setPublic()
        .setName("update")
        .setReturnType(UpdateResults.class)
        .setBody("return ds.update(query(), updateOperations, false);");

    updater.addMethod()
        .setPublic()
        .setName("update")
        .setReturnType(UpdateResults.class)
        .setBody("return ds.update(query(), updateOperations, false, wc);")
        .addParameter(WriteConcern.class, "wc");

    updater.addMethod()
        .setPublic()
        .setName("upsert")
        .setReturnType(UpdateResults.class)
        .setBody("return ds.update(query(), updateOperations, true);");

    updater.addMethod()
        .setPublic()
        .setName("upsert")
        .setReturnType(UpdateResults.class)
        .setBody("return ds.update(query(), updateOperations, true, wc);")
        .addParameter(WriteConcern.class, "wc");

    for (CritterField field : critterClass.getFields()) {
      if (!field.getParameterTypes().isEmpty()) {
        field.getParameterTypes()
            .stream()
            .forEach(criteriaClass::addImport);
      }

      criteriaClass.addImport(field.getFullType());
      if (!field.hasAnnotation(Id.class)) {
        updater.addMethod()
            .setPublic()
            .setName(field.getName())
            .setReturnType(type)
            .setBody(format("updateOperations.set(\"%s\", value);\nreturn this;", field.getName()))
            .addParameter(field.getParameterizedType(), "value");

        updater.addMethod()
            .setPublic()
            .setName(format("unset%s", nameCase(field.getName())))
            .setReturnType(type)
            .setBody(format("updateOperations.unset(\"%s\");\nreturn this;", field.getName()));

        numerics(type, updater, field);
        containers(type, updater, field);
      }
    }

    criteriaClass.addNestedType(updater);
  }

  private void numerics(final String type, final JavaClassSource updater, final CritterField field) {
    if(field.isNumeric()) {
      updater.addMethod()
          .setPublic()
          .setName(format("dec%s", nameCase(field.getName())))
          .setReturnType(type)
          .setBody(format("updateOperations.dec(\"%s\");\nreturn this;", field.getName()));

      updater.addMethod()
          .setPublic()
          .setName(format("inc%s", nameCase(field.getName())))
          .setReturnType(type)
          .setBody(format("updateOperations.inc(\"%s\");\nreturn this;", field.getName()));


      updater.addMethod()
          .setPublic()
          .setName(format("inc%s", nameCase(field.getName())))
          .setReturnType(type)
          .setBody(format("updateOperations.inc(\"%s\", value);\nreturn this;", field.getName()))
          .addParameter(field.getFullType(), "value");
    }
  }
  private void containers(final String type, final JavaClassSource updater, final CritterField field) {
    if(field.isContainer()) {

      updater.addMethod()
          .setPublic()
          .setName(format("addTo%s", nameCase(field.getName())))
          .setReturnType(type)
          .setBody(format("updateOperations.add(\"%s\", value);\nreturn this;", field.getName()))
          .addParameter(field.getParameterizedType(), "value");

      MethodSource addItems = updater.addMethod()
          .setPublic()
          .setName(format("addTo%s", nameCase(field.getName())))
          .setReturnType(type)
          .setBody(format("updateOperations.add(\"%s\", value, addDups);\nreturn this;", field.getName()));
      addItems
          .addParameter(field.getParameterizedType(), "value");
      addItems
          .addParameter("boolean", "addDups");

      addItems = updater.addMethod()
          .setPublic()
          .setName(format("addAllTo%s", nameCase(field.getName())))
          .setReturnType(type)
          .setBody(format("updateOperations.addAll(\"%s\", values, addDups);\nreturn this;", field.getName()));
      addItems.addParameter(field.getParameterizedType(), "values");
      addItems.addParameter("boolean", "addDups");

      updater.addMethod()
          .setPublic()
          .setName(format("removeFirstFrom%s", nameCase(field.getName())))
          .setReturnType(type)
          .setBody(format("updateOperations.removeFirst(\"%s\");\nreturn this;", field.getName()));

      updater.addMethod()
          .setPublic()
          .setName(format("removeLastFrom%s", nameCase(field.getName())))
          .setReturnType(type)
          .setBody(format("updateOperations.removeLast(\"%s\");\nreturn this;", field.getName()));

      updater.addMethod()
          .setPublic()
          .setName(format("removeFrom%s", nameCase(field.getName())))
          .setReturnType(type)
          .setBody(format("updateOperations.removeAll(\"%s\", value);\nreturn this;", field.getName()))
          .addParameter(field.getParameterizedType(), "value");

      MethodSource removeAll = updater.addMethod()
          .setPublic()
          .setName(format("removeAllFrom%s", nameCase(field.getName())))
          .setReturnType(type)
          .setBody(format("updateOperations.removeAll(\"%s\", values);\nreturn this;", field.getName()));
      removeAll.addParameter(field.getParameterizedType(), "values");
    }
  }

  private String nameCase(final String name) {
    return name.substring(0, 1).toUpperCase() + name.substring(1);
  }
}