org.yx.http.select.DefaultHttpActionSelector 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.http.select;

import static org.yx.common.matcher.Matchers.WILDCARD;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.function.Function;

import org.yx.common.KeyValuePair;
import org.yx.conf.AppInfo;
import org.yx.exception.SumkException;
import org.yx.http.act.HttpActionInfo;
import org.yx.http.act.HttpActionNode;
import org.yx.http.act.HttpActions;
import org.yx.http.act.IngoreNameActionInfo;
import org.yx.http.act.PrefixMappingActionInfo;
import org.yx.log.Logs;

public class DefaultHttpActionSelector implements HttpActionSelector {

	private Map actMap = Collections.emptyMap();
	private HttpActionInfo[] starts;
	private HttpActionInfo[] lastMappings;

	@Override
	public HttpActionInfo getHttpInfo(String act, String method) {
		method = method.toUpperCase();
		HttpActionInfo info = this.getAcceptedAction(actMap.get(act), act, method);
		if (info != null) {
			return info;
		}

		info = this.getAcceptedAction(this.starts, act, method);
		if (info != null) {
			return info;
		}

		return this.getAcceptedAction(this.lastMappings, act, method);
	}

	protected HttpActionInfo getAcceptedAction(HttpActionInfo[] infos, String act, String method) {
		if (infos == null) {
			return null;
		}
		for (HttpActionInfo info : infos) {
			if (info.match(act, method)) {
				return info;
			}
		}
		return null;
	}

	@Override
	public Collection actions() {
		List list = new ArrayList<>(200);
		for (HttpActionInfo[] infos : this.actMap.values()) {
			for (HttpActionInfo info : infos) {
				list.add(info);
			}
		}
		list.sort(null);

		if (this.starts != null) {
			list.addAll(Arrays.asList(this.starts));
		}
		if (this.lastMappings != null) {
			list.addAll(Arrays.asList(this.lastMappings));
		}
		return list;
	}

	protected boolean isDuplicate(HttpActionInfo info, HttpActionInfo[] old) {
		for (String m : info.node().methods()) {
			for (HttpActionInfo o : old) {
				if (o.node().acceptMethod(m)) {
					return true;
				}
			}
		}
		return false;
	}

	protected void appendStart(PrefixMappingActionInfo info, List startList) {
		for (PrefixMappingActionInfo old : startList) {
			if (!Objects.equals(info.getUrlStart(), old.getUrlStart())) {
				continue;
			}
			this.urlDuplicate(info.formalName());
			for (String m : info.node().methods()) {
				if (old.node().acceptMethod(m)) {
					throw new SumkException(-345647432, "web接口" + old.formalName() + "重复了,http方法是" + m);
				}
			}
		}
		startList.add(info);
	}

	protected void urlDuplicate(String url) {
		String key = "sumk.http.url.duplicate";
		if (!AppInfo.getBoolean(key, false)) {
			throw new SumkException(-345647431, "web接口" + url + "重复了，设置" + key + "=1可以开启url重名功能");
		}
	}

	@Override
	public void init(List> infos, Function nameResolver) {
		Map actMap = new HashMap<>(infos.size() * 2);
		List startList = new ArrayList<>();
		for (KeyValuePair kv : infos) {
			String parsedName = nameResolver.apply(kv.key());
			if (parsedName.endsWith(HttpActions.PREFIX_MATCH_ENDING)) {
				this.appendStart(new PrefixMappingActionInfo(kv.key(), kv.value(), parsedName,
						parsedName.substring(0, parsedName.length() - 1)), startList);
				continue;
			}
			HttpActionInfo info = new IngoreNameActionInfo(kv.key(), kv.value(), parsedName);
			HttpActionInfo[] old = actMap.get(parsedName);
			if (old == null) {
				actMap.put(parsedName, new HttpActionInfo[] { info });
				continue;
			}

			this.urlDuplicate(parsedName);
			if (this.isDuplicate(info, old)) {
				throw new SumkException(-345647432, "web接口" + parsedName + "重复了");
			}
			HttpActionInfo[] newInfos = Arrays.copyOf(old, old.length + 1);
			newInfos[newInfos.length - 1] = info;
			actMap.put(parsedName, newInfos);
		}

		if (startList.size() > 0) {
			String key = "sumk.http.url.match.prefix";
			if (!AppInfo.getBoolean(key, false)) {
				Logs.http().error("{}接口是前缀匹配，但是本系统没有开启前缀匹配功能。前缀匹配接口共有{}个", startList.get(0).rawAct(), startList.size());
				throw new SumkException(-345647431, "需要设置" + key + "=1才能开启前缀匹配的功能");
			}
		}
		startList.sort(null);
		Collections.reverse(startList);
		this.lastMappings = actMap.remove(WILDCARD);
		this.actMap = new HashMap<>(actMap);
		this.starts = startList.isEmpty() ? null : startList.toArray(new PrefixMappingActionInfo[startList.size()]);
	}

}