org.openas2.processor.msgtracking.BaseMsgTrackingModule Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of openas2-server Show documentation
Open source implementation of the AS2 standard for signed encrypted and compressed document transfer
There is a newer version: 2.10.1
package org.openas2.processor.msgtracking;

import java.util.HashMap;
import java.util.Map;

import org.openas2.OpenAS2Exception;
import org.openas2.Session;
import org.openas2.message.AS2MessageMDN;
import org.openas2.message.Message;
import org.openas2.message.MessageMDN;
import org.openas2.partner.Partnership;
import org.openas2.processor.BaseProcessorModule;
import org.openas2.processor.resender.ResenderModule;

public abstract class BaseMsgTrackingModule extends BaseProcessorModule implements TrackingModule {
	
    public void handle(String action, Message msg, Map options) throws OpenAS2Exception {
        
    	Map fields = buildMap(msg, options);
    	persist(msg, fields);
        
    }

    public boolean canHandle(String action, Message msg, Map options) {
        return action.equals(getModuleAction());
    }

    public void init(Session session, Map options) throws OpenAS2Exception {
        super.init(session, options);
    }

    protected abstract String getModuleAction();

    protected abstract void persist(Message msg, Map map);

    protected Map buildMap(Message msg, Map options)
    {
    	Map map = new HashMap();
    	String msgId = msg.getMessageID();
    	MessageMDN mdn = msg.getMDN();
    	if (mdn != null)
    	{
    		map.put(FIELDS.MDN_ID, mdn.getMessageID());
        	map.put(FIELDS.MDN_RESPONSE, msg.getMDN().getText());
        	// Make sure we log against the original message ID since MDN can have different ID
        	String originalMsgId = mdn.getAttribute(AS2MessageMDN.MDNA_ORIG_MESSAGEID);
        	if (originalMsgId != null && !msgId.equals(originalMsgId)) msgId = originalMsgId;
    	}
    	map.put(FIELDS.MSG_ID, msgId);
    	map.put(FIELDS.PRIOR_MSG_ID, msg.getAttribute(FIELDS.PRIOR_MSG_ID));
    	// Default DIRECTION to SEND for now...
    	String direction =  (String) options.get("DIRECTION");
    	map.put(FIELDS.DIRECTION,direction==null?"SEND":direction);
    	String isResend =  (String) options.get("IS_RESEND");
    	if (isResend != null)
    	{
    		map.put(FIELDS.IS_RESEND, isResend);
    		map.put(FIELDS.RESEND_COUNT, (String) options.get(ResenderModule.OPTION_RETRIES));
    	}
    	//map.put(FIELDS.RESEND_COUNT, );
    	String sender = msg.getPartnership().getSenderID(Partnership.PID_AS2);
    	if (sender == null) sender = mdn.getPartnership().getSenderID(Partnership.PID_AS2);
    	map.put(FIELDS.SENDER_ID, sender);
       	String receiver = msg.getPartnership().getReceiverID(Partnership.PID_AS2);
    	if (receiver == null) receiver = mdn.getPartnership().getReceiverID(Partnership.PID_AS2);
    	map.put(FIELDS.RECEIVER_ID, receiver);
    	map.put(FIELDS.STATUS, msg.getStatus());
    	String state = (String) options.get("STATE");
    	map.put(FIELDS.STATE,  state);
        map.put(FIELDS.STATE_MSG, Message.STATE_MSGS.get(state));
        map.put(FIELDS.SIGNATURE_ALGORITHM, msg.getPartnership().getAttribute(Partnership.PA_SIGNATURE_ALGORITHM));
    	map.put(FIELDS.ENCRYPTION_ALGORITHM, msg.getPartnership().getAttribute(Partnership.PA_ENCRYPTION_ALGORITHM));
    	map.put(FIELDS.COMPRESSION, msg.getPartnership().getAttribute(Partnership.PA_COMPRESSION_TYPE));
    	map.put(FIELDS.FILE_NAME, msg.getPayloadFilename());
    	map.put(FIELDS.CONTENT_TYPE, msg.getContentType());
    	map.put(FIELDS.CONTENT_TRANSFER_ENCODING, msg.getHeader("Content-Transfer-Encoding"));
    	map.put(FIELDS.MDN_MODE, (msg.getPartnership().isAsyncMDN()?"ASYNC":"SYNC"));
    	
    	return map;
    }

    public static class FIELDS {
        public static final String MSG_ID = "MSG_ID";
        public static final String PRIOR_MSG_ID = "PRIOR_MSG_ID";
        public static final String MDN_ID = "MDN_ID";
        public static final String DIRECTION = "DIRECTION";
        public static final String IS_RESEND = "IS_RESEND";
        public static final String RESEND_COUNT = "RESEND_COUNT";
        public static final String SENDER_ID = "SENDER_ID";
        public static final String RECEIVER_ID = "RECEIVER_ID";
        public static final String STATUS = "STATUS";
        public static final String STATE = "STATE";
        public static final String SIGNATURE_ALGORITHM = "SIGNATURE_ALGORITHM";
        public static final String ENCRYPTION_ALGORITHM = "ENCRYPTION_ALGORITHM";
        public static final String COMPRESSION = "COMPRESSION";
        public static final String FILE_NAME = "FILE_NAME";
        public static final String CONTENT_TYPE = "CONTENT_TYPE";
        public static final String CONTENT_TRANSFER_ENCODING = "CONTENT_TRANSFER_ENCODING";
        public static final String MDN_MODE = "MDN_MODE";
        public static final String MDN_RESPONSE = "MDN_RESPONSE";
        public static final String STATE_MSG = "STATE_MSG";
        public static final String CREATE_DT = "CREATE_DT";
        public static final String UPDATE_DT = "UPDATE_DT";
    }
    
}