package com.trendmicro.grid.acl.ds.msmq;
   
   import com.trendmicro.grid.acl.ds.ProcessingInitiator;
   import com.trendmicro.grid.acl.l0.datatypes.ProcessPackageDataSet;
   import net.sf.tinyjee.config.Connection;
   import org.slf4j.Logger;
   import org.slf4j.LoggerFactory;
   import org.springframework.stereotype.Service;
   
  import javax.annotation.PostConstruct;
  import javax.xml.bind.JAXBException;
  import javax.xml.bind.Marshaller;
  import java.io.IOException;
  import java.io.StringWriter;
  import java.io.Writer;
  import java.net.URI;
  import java.util.ArrayList;
  import java.util.Collection;
  import java.util.List;
  
  import static net.sf.tinyjee.util.Assert.assertNotNull;
  
  /**
   * Implements {@link ProcessingInitiator} as a MSMQ bound service.
   *
   * @author juergen_kellerer, werner_watzka, 2010-06-02
   * @version 1.0
   */
  @Service
  public class MSMQProcessingInitiator implements ProcessingInitiator {
  
  	private static final Logger log = LoggerFactory.getLogger(MSMQProcessingInitiator.class);
  
  	private static final String SENDER_IMPLEMENTATION = System.getProperty("gacl.msmq.sender",
  			"com.trendmicro.grid.acl.ds.msmq.PlainRoundRobinSender");
  
  	List<URI> serviceEndpoints;
  	String targetQueueName;
  	Sender sender;
  
  	@PostConstruct
  	void initializeInitiator() throws Exception {
  		Collection<Connection> connections = MSMQPropertySection.getMSMQConnections();
  
  		// Abort if config is incorrect.
  		int connectionCount = connections.size();
  		if (connectionCount != 1) {
  			if (connectionCount == 0)
  				log.warn("TMACL-00790:No MSQM connection is configured. MSMQ sending will be disabled.");
  			else
  				log.warn("TMACL-00760:More than one MSMQ connection is configured. MSMQ sending will be disabled.");
  			return;
  		}
  
  		// Define fields
  		serviceEndpoints = new ArrayList<URI>();
  		for (Connection c : connections)
  			serviceEndpoints.add(c.getURI());
  
  		if (!serviceEndpoints.isEmpty()) {
  			targetQueueName = MSMQPropertySection.getTargetQueue(MSMQPropertySection.QUEUE_PROCESS_REQUEST);
  			sender = (Sender) Class.forName(SENDER_IMPLEMENTATION).newInstance();
  			sender.reset(serviceEndpoints, targetQueueName);
  		}
  	}
  
  	/**
  	 * {@inheritDoc}
  	 * <p/>
  	 * Note: This method is called by the web-service implementation
  	 * after a job was assembled and produced this initial package.
  	 * <p/>
  	 * The implementation of this method sends the given payload against
  	 * the MSMQ queue "al_out".
  	 */
  	public void initiate(ProcessPackageDataSet dataSet) {
  		assertNotNull("ProcessPackageDataSet", dataSet);
  
  		if (serviceEndpoints.isEmpty()) {
  			log.error("TMACL-00800:Not initiating the processing for job " +
  					"'{}' as the MSMQ sending is disabled.", dataSet.getReferringJob());
  			return;
  		}
  
  		String messageBody = toXmlString(dataSet);
  		try {
  			sender.send(dataSet.getProcessPriority(), messageBody);
  		} catch (IOException e) {
  			// TODO: Note: In theory we'd need to store the messages in between in this case and
  			// TODO: send later when the MSMQ server is available again.
  			log.error("TMACL-00810:Failed initiating the processing for job '" +
  					dataSet.getReferringJob() + "' as sending to MSMQ failed.", e);
  			throw new RuntimeException(e);
  		}
  	}
  
  	String toXmlString(ProcessPackageDataSet dataSet) {
  		try {
  			Writer w = new StringWriter();
 			Marshaller m = ProcessPackageDataSet.getXmlSerializer().getJaxbContext().createMarshaller();
 			m.marshal(dataSet, w);
 			return w.toString();
 		} catch (JAXBException e) {
 			throw new RuntimeException(e);
 		}
 	}
 }