org.yx.conf.AbstractUrlConfig Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sumk Show documentation
A quick developing framewort for internet company
There is a newer version: 4.0.2
/**
 * Copyright (C) 2016 - 2030 youtongluan.
 *
 * 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 org.yx.conf;

import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.util.Arrays;

import org.yx.log.RawLog;
import org.yx.main.SumkServer;
import org.yx.util.StreamUtil;

public abstract class AbstractUrlConfig extends AbstractRefreshableSystemConfig {

	protected final URL url;

	public AbstractUrlConfig(URL url) {
		this.url = url;
	}

	protected long period = 1000L * 60 * 5;
	protected int timeout = 5000;
	protected int readTimeout = 5000;
	protected String method = "GET";
	protected Charset charset = StandardCharsets.UTF_8;
	protected byte[] rawData;

	private volatile boolean threadStarted;

	public final synchronized void stop() {
		this.started = false;
		this.onStop();
	}

	protected void onStop() {

	}

	/**
	 * 初始化
	 */
	@Override
	public void init() {
		extractData();
		if (threadStarted) {
			return;
		}
		new Thread(() -> {
			synchronized (AbstractUrlConfig.this) {
				if (threadStarted) {
					return;
				}
				threadStarted = true;
			}
			while (!SumkServer.isDestoryed() && started) {
				try {
					Thread.sleep(period);
				} catch (InterruptedException e) {
					RawLog.debug("sumk.conf", "url config exited because interrupted");
					return;
				}
				if (extractData()) {
					this.onRefresh();
				}
			}
		}).start();
	}

	protected boolean extractData() {
		HttpURLConnection conn = null;
		try {

			conn = (HttpURLConnection) url.openConnection();
			conn.setConnectTimeout(timeout);
			conn.setRequestMethod(method);
			conn.setReadTimeout(readTimeout);
			conn.setDoOutput(true);
			conn.connect();
			if (conn.getResponseCode() != 200) {
				return false;
			}
			InputStream in = conn.getInputStream();
			byte[] data = StreamUtil.readAllBytes(in, true);
			if (Arrays.equals(rawData, data)) {
				return false;
			}
			handleData(data);
			this.rawData = data;
		} catch (Exception e) {
			RawLog.error("sumk.conf", e);
			return false;
		} finally {
			if (conn != null) {
				try {
					conn.disconnect();
				} catch (Exception e2) {
				}
			}
		}
		return true;
	}

	protected abstract void handleData(byte[] data);

	public int getTimeout() {
		return timeout;
	}

	public void setTimeout(int timeout) {
		this.timeout = timeout;
	}

	public int getReadTimeout() {
		return readTimeout;
	}

	public void setReadTimeout(int readTimeout) {
		this.readTimeout = readTimeout;
	}

	public String getMethod() {
		return method;
	}

	public void setMethod(String method) {
		this.method = method;
	}

	public Charset getCharset() {
		return charset;
	}

	public void setCharset(Charset charset) {
		this.charset = charset;
	}

	public long getPeriod() {
		return period;
	}

	public void setPeriod(long period) {
		this.period = period;
	}

}