com.vip.saturn.job.console.service.helper.ReuseUtils Maven / Gradle / Ivy

Go to download
/**
 * Copyright 2016 vip.com.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
 *  the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
 * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations under the License.
 * 
 **/

package com.vip.saturn.job.console.service.helper;

import com.vip.saturn.job.console.domain.RegistryCenterClient;
import com.vip.saturn.job.console.domain.RegistryCenterConfiguration;
import com.vip.saturn.job.console.exception.SaturnJobConsoleException;
import com.vip.saturn.job.console.repository.zookeeper.CuratorRepository;
import com.vip.saturn.job.console.service.RegistryCenterService;
import com.vip.saturn.job.console.utils.JobNodePath;
import com.vip.saturn.job.console.utils.ThreadLocalCuratorClient;
import org.apache.curator.framework.CuratorFramework;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import static com.vip.saturn.job.console.exception.SaturnJobConsoleException.ERROR_CODE_NOT_EXISTED;

/**
 * @author hebelala
 */
public class ReuseUtils {

	private static final String JOB_NOT_EXIST_TEMPLATE = "The job {%s} does not exists.";

	private static final String NAMESPACE_NOT_EXIST_TEMPLATE = "The namespace {%s} does not exists.";

	private static final Logger log = LoggerFactory.getLogger(ReuseUtils.class);

	public static  T reuse(String namespace, final String jobName, RegistryCenterService registryCenterService,
			CuratorRepository curatorRepository, final ReuseCallBack callBack) throws SaturnJobConsoleException {
		return reuse(namespace, registryCenterService, curatorRepository, new ReuseCallBack() {
			@Override
			public T call(CuratorRepository.CuratorFrameworkOp curatorFrameworkOp) throws SaturnJobConsoleException {
				if (!curatorFrameworkOp.checkExists(JobNodePath.getJobNodePath(jobName))) {
					throw new SaturnJobConsoleException(ERROR_CODE_NOT_EXISTED, String.format(JOB_NOT_EXIST_TEMPLATE, jobName));
				}
				return callBack.call(curatorFrameworkOp);
			}
		});
	}

	public static  T reuse(String namespace, RegistryCenterService registryCenterService,
			CuratorRepository curatorRepository, ReuseCallBack callBack) throws SaturnJobConsoleException {
		try {
			RegistryCenterConfiguration registryCenterConfiguration = registryCenterService
					.findConfigByNamespace(namespace);
			if (registryCenterConfiguration == null) {
				throw new SaturnJobConsoleException(ERROR_CODE_NOT_EXISTED, String.format(NAMESPACE_NOT_EXIST_TEMPLATE, namespace));
			}
			RegistryCenterClient registryCenterClient = registryCenterService.connectByNamespace(namespace);
			if (registryCenterClient != null && registryCenterClient.isConnected()) {
				CuratorFramework curatorClient = registryCenterClient.getCuratorClient();
				CuratorRepository.CuratorFrameworkOp curatorFrameworkOp = curatorRepository
						.newCuratorFrameworkOp(curatorClient);
				ThreadLocalCuratorClient.setCuratorClient(curatorClient);
				return callBack.call(curatorFrameworkOp);
			} else {
				throw new SaturnJobConsoleException("Connect zookeeper failed");
			}
		} catch (SaturnJobConsoleException e) {
			throw e;
		} catch (Throwable t) {
			log.error(t.getMessage(), t);
			throw new SaturnJobConsoleException(t);
		} finally {
			ThreadLocalCuratorClient.clear();
		}
	}

	public static void reuse(String namespace, final String jobName, RegistryCenterService registryCenterService,
			CuratorRepository curatorRepository, final ReuseCallBackWithoutReturn callBack)
			throws SaturnJobConsoleException {
		reuse(namespace, registryCenterService, curatorRepository, new ReuseCallBackWithoutReturn() {
			@Override
			public void call(CuratorRepository.CuratorFrameworkOp curatorFrameworkOp) throws SaturnJobConsoleException {
				if (!curatorFrameworkOp.checkExists(JobNodePath.getJobNodePath(jobName))) {
					throw new SaturnJobConsoleException(ERROR_CODE_NOT_EXISTED, String.format(JOB_NOT_EXIST_TEMPLATE, jobName));
				}
				callBack.call(curatorFrameworkOp);
			}
		});
	}

	public static void reuse(String namespace, RegistryCenterService registryCenterService,
			CuratorRepository curatorRepository, ReuseCallBackWithoutReturn callBack) throws SaturnJobConsoleException {
		try {
			RegistryCenterConfiguration registryCenterConfiguration = registryCenterService
					.findConfigByNamespace(namespace);
			if (registryCenterConfiguration == null) {
				throw new SaturnJobConsoleException(ERROR_CODE_NOT_EXISTED, String.format(NAMESPACE_NOT_EXIST_TEMPLATE, namespace));
			}
			RegistryCenterClient registryCenterClient = registryCenterService.connectByNamespace(namespace);
			if (registryCenterClient != null && registryCenterClient.isConnected()) {
				CuratorFramework curatorClient = registryCenterClient.getCuratorClient();
				CuratorRepository.CuratorFrameworkOp curatorFrameworkOp = curatorRepository
						.newCuratorFrameworkOp(curatorClient);
				ThreadLocalCuratorClient.setCuratorClient(curatorClient);
				callBack.call(curatorFrameworkOp);
			} else {
				throw new SaturnJobConsoleException("Connect zookeeper failed");
			}
		} catch (SaturnJobConsoleException e) {
			throw e;
		} catch (Throwable t) {
			log.error(t.getMessage(), t);
			throw new SaturnJobConsoleException(t);
		} finally {
			ThreadLocalCuratorClient.clear();
		}
	}

}