org.dromara.jpom.system.init.ProxySelectorConfig Maven / Gradle / Ivy
/*
* The MIT License (MIT)
*
* Copyright (c) 2019 Code Technology Studio
*
* Permission is hereby granted, free of charge, to any person obtaining a copy of
* this software and associated documentation files (the "Software"), to deal in
* the Software without restriction, including without limitation the rights to
* use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
* the Software, and to permit persons to whom the Software is furnished to do so,
* subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
* FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
* COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
* IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
package org.dromara.jpom.system.init;
import cn.hutool.core.util.ReUtil;
import cn.hutool.core.util.StrUtil;
import cn.keepbx.jpom.event.ICacheTask;
import com.alibaba.fastjson2.JSONArray;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.dromara.jpom.common.ILoadEvent;
import org.dromara.jpom.common.forward.NodeForward;
import org.dromara.jpom.service.system.SystemParametersServer;
import org.springframework.context.ApplicationContext;
import org.springframework.context.annotation.Configuration;
import java.io.IOException;
import java.net.Proxy;
import java.net.ProxySelector;
import java.net.SocketAddress;
import java.net.URI;
import java.util.Collections;
import java.util.List;
import java.util.Objects;
import java.util.Optional;
import java.util.stream.Collectors;
/**
* 全局代理配置
*
* @author bwcx_jzy
* @since 2022/7/4
*/
@Slf4j
@Configuration
public class ProxySelectorConfig extends ProxySelector implements ILoadEvent, ICacheTask {
public static final String KEY = "global_proxy";
private final SystemParametersServer systemParametersServer;
private volatile List proxyConfigItems;
private ProxySelector defaultProxySelector;
public ProxySelectorConfig(SystemParametersServer systemParametersServer) {
this.systemParametersServer = systemParametersServer;
}
@Override
public List select(URI uri) {
String url = uri.toString();
return Optional.ofNullable(proxyConfigItems)
.flatMap(proxyConfigItems -> proxyConfigItems.stream()
.filter(proxyConfigItem -> {
if (StrUtil.equals(proxyConfigItem.getPattern(), "*")) {
return true;
}
if (ReUtil.isMatch(proxyConfigItem.getPattern(), url)) {
// 满足正则条件
return true;
}
return StrUtil.containsIgnoreCase(url, proxyConfigItem.getPattern());
})
.map(proxyConfigItem -> NodeForward.crateProxy(proxyConfigItem.getProxyType(), proxyConfigItem.getProxyAddress()))
.filter(Objects::nonNull)
.findFirst()
.map(Collections::singletonList)
).orElseGet(() -> {
// revert to the default behaviour
return defaultProxySelector == null ? Collections.singletonList(Proxy.NO_PROXY) : defaultProxySelector.select(uri);
});
}
@Override
public void connectFailed(URI uri, SocketAddress sa, IOException ioe) {
if (uri == null || sa == null || ioe == null) {
throw new IllegalArgumentException(
"Arguments can't be null.");
}
}
/**
* 刷新
*/
@Override
public void refreshCache() {
JSONArray array = systemParametersServer.getConfigDefNewInstance(ProxySelectorConfig.KEY, JSONArray.class);
proxyConfigItems = array.toJavaList(ProxyConfigItem.class)
.stream()
.filter(proxyConfigItem -> StrUtil.isAllNotEmpty(proxyConfigItem.pattern, proxyConfigItem.proxyAddress, proxyConfigItem.proxyType))
.collect(Collectors.toList());
}
@Override
public void afterPropertiesSet(ApplicationContext applicationContext) throws Exception {
if (ProxySelector.getDefault() != this) {
defaultProxySelector = ProxySelector.getDefault();
//
ProxySelector.setDefault(this);
}
// 立马配置 全局代理
this.refreshCache();
}
/**
* @author bwcx_jzy
* @since 2022/7/4
*/
@Data
public static class ProxyConfigItem {
private String pattern;
/**
* @see Proxy.Type
*/
private String proxyType;
/**
* 127.0.0.1:8888
*/
private String proxyAddress;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy