jp.go.nict.langrid.cosee.DynamicBindingRewriter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jp.go.nict.langrid.servicecontainer Show documentation
Service container including JSON-RPC handler for the Service Grid Server Software and java web services.
There is a newer version: 1.1.3
Show newest version
/*
 * This is a program for Language Grid Core Node. This combines multiple language resources and provides composite language services.
 * Copyright (C) 2005-2010 NICT Language Grid Project.
 *
 * This program is free software: you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 2.1 of the License, or (at
 * your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser
 * General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program. If not, see .
 */
package jp.go.nict.langrid.cosee;

import java.net.URI;
import java.text.ParseException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.logging.Level;
import java.util.logging.Logger;

import jp.go.nict.langrid.commons.cs.binding.BindingNode;
import jp.go.nict.langrid.commons.cs.binding.DynamicBindingUtil;
import jp.go.nict.langrid.commons.rpc.RpcHeader;
import jp.go.nict.langrid.commons.rpc.TransportHeader;
import jp.go.nict.langrid.commons.ws.LangridConstants;
import jp.go.nict.langrid.commons.ws.ServiceContext;
import jp.go.nict.langrid.cosee.binding.TreeBindings;
import jp.go.nict.langrid.repackaged.net.arnx.jsonic.JSON;

/**
 * 
 * 
 * @author Takao Nakaguchi
 */
public class DynamicBindingRewriter extends AbstractEndpointRewriter{
	@SuppressWarnings("unchecked")
	public void extractProperties(
			ServiceContext serviceContext
			, Map properties)
	{
		super.extractProperties(
				serviceContext, properties);

		List bindings = null;
		try{
			TreeBindings treeBindings = (TreeBindings)properties.get("treeBindings");
			if(treeBindings == null){
				treeBindings = new TreeBindings();
			}
			bindings = (List)properties.get(
					LangridConstants.ACTOR_SERVICE_TREEBINDING
					);
			treeBindings.merge(new TreeBindings(bindings));
			properties.put("treeBindings", treeBindings);
			properties.put("treeBindings.authKey", serviceContext.getInitParameter("langrid.appAuthKey"));
			properties.put("treeBindings.authUserId", serviceContext.getAuthUser());
		} catch(ParseException e){
			logger.log(Level.WARNING
					, "failed to parse binding information: " + JSON.encode(bindings, true)
					, e);
			properties.put("treeBindings", new TreeBindings());
		}
	}

	public Endpoint rewrite(
			Endpoint original, Map properties
			, URI processNamespace, String partnerLinkName, URI serviceNamespace
			)
	{
		TreeBindings tb = (TreeBindings)properties.get("treeBindings");
		String serviceId = tb.getBindingFor(partnerLinkName);
		if(serviceId == null || serviceId.trim().length() == 0){
			return original;
		}

		Endpoint ep = makeEndpoint(
				original, serviceId, properties
				);
		if(!serviceId.contains("://")){
			String key = (String)properties.get("treeBindings.authKey");
			String userId = (String)properties.get("treeBindings.authUserId");
			if(key != null && userId != null){
				ep.setUserName(key);
				ep.setPassword(userId);
			}
		}
		return ep;
	}

	@Override
	public void adjustProperties(
			Map properties
			, String partnerLinkName)
	{
		super.adjustProperties(properties, partnerLinkName);
		TreeBindings tb = (TreeBindings)properties.get("treeBindings");
		BindingNode node = tb.getBindingNodeFor(partnerLinkName);
		if(node == null) return;

		{
			@SuppressWarnings("unchecked")
			List additionalHeaders = (List)properties.get("additionalMimeHeaders");
			for(TransportHeader th : node.getTransportHeaders()){
				additionalHeaders.add(th);
			}
		}
		{
			@SuppressWarnings("unchecked")
			List additionalHeaders = (List)properties.get("additionalRpcHeaders");
			for(RpcHeader th : node.getRpcHeaders()){
				additionalHeaders.add(th);
			}
		}

		Collection children = node.getChildren();
		if(children != null && children.size() > 0){
			List headers = new ArrayList();
			headers.add(new RpcHeader(
					LangridConstants.ACTOR_SERVICE_TREEBINDING
					, "binding"
					, DynamicBindingUtil.encodeTree(children)
					));
			properties.put(
					LangridConstants.ACTOR_SERVICE_TREEBINDING
					, headers);
		} else{
			properties.remove(
					LangridConstants.ACTOR_SERVICE_TREEBINDING
					);
		}
	}

	private static Logger logger = LoggerFactory.create();
}