Cascade Save for spring mongo data

1. we have some entities like below:

    AlarmLogs.java

package com.eifesun.monitor.web.model;

import static com.eifesun.monitor.DeviceType.fromValue;
import static com.eifesun.monitor.inverter.power20k.ErrorCode.getType;
import static org.joda.time.DateTime.now;

import java.util.Date;

import org.bson.types.ObjectId;
import org.joda.time.DateTime;
import org.joda.time.Interval;
import org.joda.time.PeriodType;
import org.springframework.data.annotation.CreatedDate;
import org.springframework.data.annotation.Id;
import org.springframework.data.annotation.LastModifiedDate;
import org.springframework.data.annotation.Transient;
import org.springframework.data.annotation.Version;
import org.springframework.data.mongodb.core.mapping.DBRef;
import org.springframework.data.mongodb.core.mapping.Document;
import org.springframework.format.annotation.DateTimeFormat;
import org.springframework.format.annotation.DateTimeFormat.ISO;

import com.eifesun.monitor.DeviceType;
import com.eifesun.monitor.util.Helper;
import com.eifesun.monitor.web.model.mongodb.CascadeSave;

@Document(collection = "alarm_logs")
public class AlarmLogs {
    @Id
    private ObjectId id;
    @Version
    private Long version;
    @DateTimeFormat(iso = ISO.DATE_TIME)
    private Date alarmDate;
    private String deviceType;
    private String clientId;
    private String errorType;
    private int errorValue;

    @CreatedDate
    private DateTime createdDate;
    @LastModifiedDate
    private DateTime lastModifiedDate;

    @DBRef
    @CascadeSave
    private AlarmDetails alarmDetails;

    @Transient
    private static final String DATE_PARTEN = "MM/dd/yyyy HH:mm";

    public AlarmLogs() {
    }

    public AlarmLogs(DeviceType deviceType, String clientId, String errorType, int errorValue) {
        this.deviceType = deviceType.toValue();
        this.clientId = clientId;
        this.errorType = errorType;
        this.errorValue = errorValue;
        this.alarmDate = now().toDate();
    }

    public String getCreatedDate() {
        return createdDate.toString(DATE_PARTEN);
    }

    public String getAlarmRemainedTime() {
        Interval interval = new Interval(createdDate.getMillis(), createdDate.getMillis());
        DateTime start = this.version > 0 ? lastModifiedDate : now();
        interval.withEndMillis(start.getMillis());
        return Helper.FORMATTER.print(interval.toPeriod(PeriodType.yearMonthDayTime()));
    }

    public ObjectId getId() {
        return id;
    }

    public String getDeviceType() {
        return fromValue(deviceType).toDisplay();
    }

    public String getErrorMessage() {
        return getType(errorType).getDescription(errorValue);
    }

    public String getErrorType() {
        return errorType;
    }

    public int getErrorValue() {
        return errorValue;
    }

    public String getClientId() {
        return clientId;
    }

    public AlarmDetails getAlarmDetails() {
        return alarmDetails;
    }

    public void setAlarmDetails(AlarmDetails alarmDetails) {
        this.alarmDetails = alarmDetails;
    }

    public Long getVersion() {
        return version;
    }

    public void setVersion(Long version) {
        this.version = version;
    }

    public DateTime getLastModifiedDate() {
        return lastModifiedDate;
    }

}

   AlarmDetails.java

package com.eifesun.monitor.web.model;

import org.bson.types.ObjectId;
import org.joda.time.DateTime;
import org.springframework.data.annotation.CreatedDate;
import org.springframework.data.annotation.Id;
import org.springframework.data.mongodb.core.mapping.Document;

@Document(collection = "alarm_details")
public class AlarmDetails {
    @Id
    private ObjectId id;
    private int panel1Power;
    private int panel2Power;
    private int panel1Current;
    private int panel2Current;
    private int panel1Voltage;
    private int panel2Voltage;
    private int outputPower;
    private int lineRAC;
    private int lineRV;
    private int lineSAC;
    private int lineSV;
    private int lineTAC;
    private int lineTV;
    private int gridFrequence;
    private int statusCode;
    private int boxTemperature;
    private int igbtTemperature;
    private int workingHour;
    @CreatedDate
    protected DateTime createdDate;

    public AlarmDetails() {
    }

    public int getPanel1Power() {
        return panel1Power;
    }

    public void setPanel1Power(int panel1Power) {
        this.panel1Power = panel1Power;
    }

    public int getPanel2Power() {
        return panel2Power;
    }

    public void setPanel2Power(int panel2Power) {
        this.panel2Power = panel2Power;
    }

    public int getPanel1Current() {
        return panel1Current;
    }

    public void setPanel1Current(int panel1Current) {
        this.panel1Current = panel1Current;
    }

    public int getPanel2Current() {
        return panel2Current;
    }

    public void setPanel2Current(int panel2Current) {
        this.panel2Current = panel2Current;
    }

    public int getPanel1Voltage() {
        return panel1Voltage;
    }

    public void setPanel1Voltage(int panel1Voltage) {
        this.panel1Voltage = panel1Voltage;
    }

    public int getPanel2Voltage() {
        return panel2Voltage;
    }

    public void setPanel2Voltage(int panel2Voltage) {
        this.panel2Voltage = panel2Voltage;
    }

    public int getOutputPower() {
        return outputPower;
    }

    public void setOutputPower(int outputPower) {
        this.outputPower = outputPower;
    }

    public int getLineRAC() {
        return lineRAC;
    }

    public void setLineRAC(int lineRAC) {
        this.lineRAC = lineRAC;
    }

    public int getLineRV() {
        return lineRV;
    }

    public void setLineRV(int lineRV) {
        this.lineRV = lineRV;
    }

    public int getLineSAC() {
        return lineSAC;
    }

    public void setLineSAC(int lineSAC) {
        this.lineSAC = lineSAC;
    }

    public int getLineSV() {
        return lineSV;
    }

    public void setLineSV(int lineSV) {
        this.lineSV = lineSV;
    }

    public int getLineTAC() {
        return lineTAC;
    }

    public void setLineTAC(int lineTAC) {
        this.lineTAC = lineTAC;
    }

    public int getLineTV() {
        return lineTV;
    }

    public void setLineTV(int lineTV) {
        this.lineTV = lineTV;
    }

    public int getGridFrequence() {
        return gridFrequence;
    }

    public void setGridFrequence(int gridFrequence) {
        this.gridFrequence = gridFrequence;
    }

    public int getStatusCode() {
        return statusCode;
    }

    public void setStatusCode(int statusCode) {
        this.statusCode = statusCode;
    }

    public int getBoxTemperature() {
        return boxTemperature;
    }

    public void setBoxTemperature(int boxTemperature) {
        this.boxTemperature = boxTemperature;
    }

    public int getIgbtTemperature() {
        return igbtTemperature;
    }

    public void setIgbtTemperature(int igbtTemperature) {
        this.igbtTemperature = igbtTemperature;
    }

    public int getWorkingHour() {
        return workingHour;
    }

    public void setWorkingHour(int workingHour) {
        this.workingHour = workingHour;
    }

}

2. here we create a new annotation "CascadeSave" on fields "alarmDetails".

package com.eifesun.monitor.web.model.mongodb;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.FIELD})
public @interface CascadeSave {

}

 

  CascadingMongoEventListener.java

package com.eifesun.monitor.web.model.mongodb;

import java.lang.reflect.Field;

import javax.inject.Inject;

import org.springframework.data.annotation.Id;
import org.springframework.data.mapping.model.MappingException;
import org.springframework.data.mongodb.core.MongoOperations;
import org.springframework.data.mongodb.core.mapping.DBRef;
import org.springframework.data.mongodb.core.mapping.event.AbstractMongoEventListener;
import org.springframework.util.ReflectionUtils;

@SuppressWarnings("rawtypes")
public class CascadingMongoEventListener extends AbstractMongoEventListener {
    @Inject
    private MongoOperations mongoOperations;

    @Override
    public void onBeforeConvert(final Object source) {
        ReflectionUtils.doWithFields(source.getClass(), new ReflectionUtils.FieldCallback() {

            public void doWith(Field field) throws IllegalArgumentException, IllegalAccessException {
                ReflectionUtils.makeAccessible(field);

                if (field.isAnnotationPresent(DBRef.class) && field.isAnnotationPresent(CascadeSave.class)) {
                    final Object fieldValue = field.get(source);
                    if (fieldValue == null) {
                        return;
                    }

                    DbRefFieldCallback callback = new DbRefFieldCallback();
                    ReflectionUtils.doWithFields(fieldValue.getClass(), callback);
                    if (!callback.isIdFound()) {
                        throw new MappingException("Cannot perform cascade save on child object without id set");
                    }

                    mongoOperations.save(fieldValue);
                }
            }
        });
    }

    private static class DbRefFieldCallback implements ReflectionUtils.FieldCallback {
        private boolean idFound;

        public void doWith(Field field) throws IllegalArgumentException, IllegalAccessException {
            ReflectionUtils.makeAccessible(field);

            if (field.isAnnotationPresent(Id.class)) {
                idFound = true;
            }
        }

        public boolean isIdFound() {
            return idFound;
        }
    }
}