eu.unicore.services.registry.RegistryEntryUpdater Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of use-rest Show documentation
There is a newer version: 5.2.0-2
package eu.unicore.services.registry;

import java.util.Calendar;
import java.util.Map;

import org.apache.logging.log4j.Logger;

import eu.unicore.services.Home;
import eu.unicore.services.Kernel;
import eu.unicore.services.exceptions.ResourceUnknownException;
import eu.unicore.services.impl.InstanceChecker;
import eu.unicore.services.rest.client.RegistryClient;
import eu.unicore.util.Log;

/**
 * Checks and refreshes a registry entry in the local registry.
 * 
 * @author demuth
 * @author schuller
 */
public class RegistryEntryUpdater implements InstanceChecker {

	private static final Logger logger = Log.getLogger(Log.SERVICES+".registry", RegistryEntryUpdater.class);
	
	public boolean check(Home home, String id)throws Exception{
		Calendar c=home.getTerminationTime(id);
		logger.debug("Checking <{} {}> TT = {}", home.getServiceName(), id, (c!=null?c.getTime():"none"));
		//if for some reason the TT is null, force a refresh (in contrast to the usual expiry check)
		return c==null? true : (c.compareTo(Calendar.getInstance())<=0);
	}

	/**
	 * For the current registry entry, it is checked whether the 
	 * corresponding member service (i.e. the service registered in the local registry)
	 * is still alive. 
	 * 
	 * 
	 *  If the member service is no longer alive, its entry is refreshed, i.e. re-added 
	 * to the local registry.
	 *
	 *  
If the member service is gone, the registry entry is removed.
	 *
	 *
	 * @return true if the registry entry is still valid, false 
	 * if the service is gone and the registry entry was removed 
	 */
	@Override
	public boolean process(Home home, String id) {
		if(home.isShuttingDown())return true;
		Kernel kernel = home.getKernel();
		String serviceName = home.getServiceName();
		try{
			RegistryEntryImpl entry=(RegistryEntryImpl)home.get(id);
			String memberAddress = entry.getModel().getEndpoint();
			
			// check that URL is still basically correct (e.g. hostname, port) 
			if(!checkBasicCorrectness(memberAddress, kernel)){
				logger.info("Member address <{}> is no longer valid, destroying registry entry.", memberAddress);
				entry.destroy();
				home.destroyResource(id);
				//instance is invalid and should be removed from all checks
				return false;
			}
			Mapcontent = entry.getModel().getContent();
			// if server key is present in content, check that it is still up to date
			checkAndUpdateServerPublicKey(kernel, content);
			try
			{
				reAdd(kernel, memberAddress, content);
				logger.debug("Refreshed registry entry for: {}", memberAddress);
			}
			catch(Exception e)
			{						
				Log.logException("Error refreshing service entry for: "+memberAddress,e,logger);
			}		
		}
		catch(ResourceUnknownException rue) {
			// entry is gone
			return false;
		}
		catch(Exception e){
			Log.logException("Could not update registry entry "+serviceName+"/"+id,e,logger);
		}

		// instance is still valid
		return true;
	}
	
	/**
	 * 
	 * @param url - the member service URL to check
	 * @return true if URL looks OK
	 */
	protected boolean checkBasicCorrectness(String url, Kernel k){
		String baseURL = k.getContainerProperties().getContainerURL();
		return url!=null && url.startsWith(baseURL);
	}

	protected void checkAndUpdateServerPublicKey(Kernel k, Mapcontent) {
		String dn = content.get(RegistryClient.SERVER_IDENTITY);
		String pem = content.get(RegistryClient.SERVER_PUBKEY);
		if(dn!=null && pem!=null) {
			String dn1 = k.getContainerSecurityConfiguration().getCredential().getSubjectName();
			String pem1 = k.getContainerSecurityConfiguration().getServerCertificateAsPEM();
			content.put(RegistryClient.SERVER_IDENTITY, dn1);
			content.put(RegistryClient.SERVER_PUBKEY, pem1);
		}
	}
	
	protected void reAdd(Kernel kernel, String endpoint, Mapcontent) throws Exception
	{
		RegistryHandler handler = kernel.getAttribute(RegistryHandler.class);
		handler.getRegistryClient().refresh(endpoint, content);
	}

}