Java tutorial
/** * Copyright (C) 2009 - present by OpenGamma Inc. and the OpenGamma group of companies * * Please see distribution for license. */ package com.opengamma.web.portfolio; import java.net.URI; import java.util.LinkedList; import java.util.List; import javax.ws.rs.Consumes; import javax.ws.rs.DELETE; import javax.ws.rs.FormParam; import javax.ws.rs.GET; import javax.ws.rs.POST; import javax.ws.rs.PUT; import javax.ws.rs.Path; import javax.ws.rs.Produces; import javax.ws.rs.core.MediaType; import javax.ws.rs.core.Response; import javax.ws.rs.core.Response.Status; import org.apache.commons.lang.StringUtils; import org.joda.beans.impl.flexi.FlexiBean; import com.google.common.base.Objects; import com.opengamma.DataNotFoundException; import com.opengamma.id.UniqueId; import com.opengamma.master.portfolio.ManageablePortfolioNode; import com.opengamma.master.portfolio.PortfolioDocument; import com.opengamma.master.position.PositionSearchRequest; import com.opengamma.master.position.PositionSearchResult; import com.opengamma.util.tuple.ObjectsPair; /** * RESTful resource for a node in a portfolio. */ @Path("/portfolios/{portfolioId}/nodes/{nodeId}") public class WebPortfolioNodeResource extends AbstractWebPortfolioResource { /** * Creates the resource. * @param parent the parent resource, not null */ public WebPortfolioNodeResource(final AbstractWebPortfolioResource parent) { super(parent); } //------------------------------------------------------------------------- @GET @Produces(MediaType.TEXT_HTML) public String getHTML() { FlexiBean out = createPortfolioNodeData(); return getFreemarker().build(HTML_DIR + "portfolionode.ftl", out); } @GET @Produces(MediaType.APPLICATION_JSON) public Response getJSON() { FlexiBean out = createPortfolioNodeData(); PortfolioDocument doc = data().getPortfolio(); if (!doc.isLatest()) { return Response.status(Status.NOT_FOUND).build(); } String s = getFreemarker().build(JSON_DIR + "portfolionode.ftl", out); return Response.ok(s).build(); } private FlexiBean createPortfolioNodeData() { ManageablePortfolioNode node = data().getNode(); PositionSearchRequest positionSearch = new PositionSearchRequest(); positionSearch.setPositionObjectIds(node.getPositionIds()); PositionSearchResult positionsResult = data().getPositionMaster().search(positionSearch); resolveSecurities(positionsResult.getPositions()); FlexiBean out = createRootData(); out.put("positionsResult", positionsResult); out.put("positions", positionsResult.getPositions()); return out; } //------------------------------------------------------------------------- @POST @Consumes(MediaType.APPLICATION_FORM_URLENCODED) @Produces(MediaType.TEXT_HTML) public Response postHTML(@FormParam("name") String name) { name = StringUtils.trimToNull(name); if (name == null) { FlexiBean out = createRootData(); out.put("err_nameMissing", true); String html = getFreemarker().build(HTML_DIR + "portfolionode-add.ftl", out); return Response.ok(html).build(); } URI uri = createPortfolioNode(name); return Response.seeOther(uri).build(); } @POST @Consumes(MediaType.APPLICATION_FORM_URLENCODED) @Produces(MediaType.APPLICATION_JSON) public Response postJSON(@FormParam("name") String name) { name = StringUtils.trimToNull(name); URI uri = createPortfolioNode(name); return Response.created(uri).build(); } private URI createPortfolioNode(String name) { ManageablePortfolioNode newNode = new ManageablePortfolioNode(name); ManageablePortfolioNode node = data().getNode(); node.addChildNode(newNode); PortfolioDocument doc = data().getPortfolio(); doc = data().getPortfolioMaster().update(doc); URI uri = WebPortfolioNodeResource.uri(data()); // lock URI before updating data() data().setPortfolio(doc); return uri; } //------------------------------------------------------------------------- @PUT @Consumes(MediaType.APPLICATION_FORM_URLENCODED) @Produces(MediaType.TEXT_HTML) public Response putHTML(@FormParam("name") String name) { PortfolioDocument doc = data().getPortfolio(); if (doc.isLatest() == false) { return Response.status(Status.FORBIDDEN).entity(getHTML()).build(); } name = StringUtils.trimToNull(name); if (name == null) { FlexiBean out = createRootData(); out.put("err_nameMissing", true); String html = getFreemarker().build(HTML_DIR + "portfolionode-update.ftl", out); return Response.ok(html).build(); } URI uri = updatePortfolioNode(name, doc); return Response.seeOther(uri).build(); } @PUT @Consumes(MediaType.APPLICATION_FORM_URLENCODED) @Produces(MediaType.APPLICATION_JSON) public Response putJSON(@FormParam("name") String name) { PortfolioDocument doc = data().getPortfolio(); if (doc.isLatest() == false) { return Response.status(Status.FORBIDDEN).entity(getHTML()).build(); } name = StringUtils.trimToNull(name); updatePortfolioNode(name, doc); return Response.ok().build(); } private URI updatePortfolioNode(String name, PortfolioDocument doc) { ManageablePortfolioNode node = data().getNode(); URI uri = WebPortfolioNodeResource.uri(data()); // lock URI before updating data() if (Objects.equal(node.getName(), name) == false) { node.setName(name); doc = data().getPortfolioMaster().update(doc); data().setPortfolio(doc); } return uri; } //------------------------------------------------------------------------- @DELETE @Produces(MediaType.TEXT_HTML) public Response deleteHTML() { PortfolioDocument doc = data().getPortfolio(); if (doc.isLatest() == false) { Response.status(Status.FORBIDDEN).entity(getHTML()).build(); } if (data().getParentNode() == null) { throw new IllegalArgumentException("Root node cannot be deleted"); } if (data().getParentNode().removeNode(data().getNode().getUniqueId()) == false) { throw new DataNotFoundException("PortfolioNode not found: " + data().getNode().getUniqueId()); } doc = data().getPortfolioMaster().update(doc); data().setPortfolio(doc); URI uri = WebPortfolioNodeResource.uri(data(), data().getParentNode().getUniqueId()); return Response.seeOther(uri).build(); } @DELETE @Produces(MediaType.APPLICATION_JSON) public Response deleteJSON() { PortfolioDocument doc = data().getPortfolio(); if (doc.isLatest()) { if (data().getParentNode() == null) { throw new IllegalArgumentException("Root node cannot be deleted"); } if (data().getParentNode().removeNode(data().getNode().getUniqueId()) == false) { throw new DataNotFoundException("PortfolioNode not found: " + data().getNode().getUniqueId()); } doc = data().getPortfolioMaster().update(doc); } return Response.ok().build(); } //------------------------------------------------------------------------- @Path("positions") public WebPortfolioNodePositionsResource findPositions() { return new WebPortfolioNodePositionsResource(this); } //------------------------------------------------------------------------- /** * Creates the output root data. * @return the output root data, not null */ protected FlexiBean createRootData() { FlexiBean out = super.createRootData(); PortfolioDocument doc = data().getPortfolio(); ManageablePortfolioNode node = data().getNode(); out.put("portfolioDoc", doc); out.put("portfolio", doc.getPortfolio()); out.put("parentNode", data().getParentNode()); out.put("node", node); out.put("childNodes", node.getChildNodes()); out.put("deleted", !doc.isLatest()); out.put("pathNodes", getPathNodes(node)); return out; } /** * Extracts the path from the root node to the current node, for web breadcrumb display. * The nodes are returned in a list, ordered from root to current node. * Not using findNodeStackByObjectId(), which traverses the tree exhaustively until it finds the required path. * @param node the current node * @return a list of <UniqueId, String> pairs denoting all nodes on the path from root to current node */ protected List<ObjectsPair<UniqueId, String>> getPathNodes(ManageablePortfolioNode node) { LinkedList<ObjectsPair<UniqueId, String>> result = new LinkedList<ObjectsPair<UniqueId, String>>(); ManageablePortfolioNode currentNode = node; while (currentNode != null) { result.addFirst(new ObjectsPair<UniqueId, String>(currentNode.getUniqueId(), currentNode.getName())); currentNode = (currentNode.getParentNodeId() == null) ? null : data().getPortfolioMaster().getNode(currentNode.getParentNodeId()); } return result; } //------------------------------------------------------------------------- /** * Builds a URI for this resource. * @param data the data, not null * @return the URI, not null */ public static URI uri(final WebPortfoliosData data) { return uri(data, null); } /** * Builds a URI for this resource. * @param data the data, not null * @param overrideNodeId the override node id, null uses information from data * @return the URI, not null */ public static URI uri(final WebPortfoliosData data, final UniqueId overrideNodeId) { String portfolioId = data.getBestPortfolioUriId(null); String nodeId = data.getBestNodeUriId(overrideNodeId); return data.getUriInfo().getBaseUriBuilder().path(WebPortfolioNodeResource.class).build(portfolioId, nodeId); } }