RewardNetworkEndpoint.java

package rewards.ws;

import org.springframework.util.xml.DomUtils;
import org.springframework.ws.server.endpoint.AbstractDomPayloadEndpoint;
import org.w3c.dom.Document;
import org.w3c.dom.Element;

import rewards.Dining;
import rewards.RewardConfirmation;
import rewards.RewardNetwork;
import rewards.AccountContribution.Distribution;

public class RewardNetworkEndpoint extends AbstractDomPayloadEndpoint {

	private static final String NAMESPACE_URI = "http://www.springsource.com/reward-network";

	private RewardNetwork rewardNetwork;

	public RewardNetworkEndpoint(RewardNetwork rewardNetwork) {
		this.rewardNetwork = rewardNetwork;
	}

	@Override
	protected Element invokeInternal(Element requestElement, Document document) throws Exception {
		Element diningElement = DomUtils.getChildElementByTagName(requestElement, "dining");
		Dining dining = mapDining(diningElement);
		RewardConfirmation confirmation = rewardNetwork.rewardAccountFor(dining);
		Element responseElement = document.createElementNS(NAMESPACE_URI, "rewardAccountForDiningResponse");
		responseElement.appendChild(mapRewardConfirmation(document, confirmation));
		return responseElement;
	}

	private Dining mapDining(Element diningElement) {
		String amount = diningElement.getAttribute("amount");
		String creditCardNumber = diningElement.getAttribute("creditCardNumber");
		String merchantNumber = diningElement.getAttribute("merchantNumber");
		Dining dining = Dining.createDining(amount, creditCardNumber, merchantNumber);
		return dining;
	}

	private Element mapRewardConfirmation(Document document, RewardConfirmation confirmation) {
		Element confirmationElement = document.createElementNS(NAMESPACE_URI, "rewardConfirmation");
		confirmationElement.setAttribute("confirmationNumber", confirmation.getConfirmationNumber());
		confirmationElement.setAttribute("accountNumber", confirmation.getAccountContribution().getAccountNumber());
		confirmationElement.setAttribute("amount", confirmation.getAccountContribution().getAmount().toString());
		for (Distribution distribution : confirmation.getAccountContribution().getDistributions()) {
			Element distributionElement = document.createElementNS(NAMESPACE_URI, "distribution");
			distributionElement.setAttribute("beneficiary", distribution.getBeneficiary());
			distributionElement.setAttribute("amount", distribution.getAmount().toString());
			distributionElement.setAttribute("percentage", distribution.getPercentage().toString());
			distributionElement.setAttribute("totalSavings", distribution.getTotalSavings().toString());
			confirmationElement.appendChild(distributionElement);
		}
		return confirmationElement;
	}
}