import ohos.net.httpurlconnection.HttpURLConnection;
import ohos.net.httpurlconnection.HttpURLConnectionUtils;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.InetSocketAddress;
import java.net.Proxy;
import java.net.ProxySelector;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.List;

public class ProxySelectorExample {
    public static void main(String[] args) {
        try {
            // 设置默认的ProxySelector
            ProxySelector.setDefault(new CustomProxySelector());

            // 创建URL对象
            String urlString = "https://www.example.com";
            URI uri = new URI(urlString);

            // 打开HttpURLConnection连接，使用代理
            HttpURLConnection connection = (HttpURLConnection) uri.toURL().openConnection();

            // 发送GET请求
            connection.setRequestMethod("GET");

            // 获取响应码
            int responseCode = connection.getResponseCode();
            System.out.println("Response Code: " + responseCode);

            // 读取响应数据
            BufferedReader reader = new BufferedReader(new InputStreamReader(connection.getInputStream()));
            StringBuilder response = new StringBuilder();
            String line;
            while ((line = reader.readLine()) != null) {
                response.append(line);
            }
            reader.close();

            // 打印响应数据
            System.out.println("Response Data: " + response.toString());

            // 关闭连接
            connection.disconnect();

        } catch (URISyntaxException | IOException e) {
            e.printStackTrace();
        }
    }

    // 自定义ProxySelector类
    static class CustomProxySelector extends ProxySelector {
        @Override
        public List<Proxy> select(URI uri) {
            // 在这里实现选择代理的逻辑，可以根据 uri 的信息返回不同的代理配置
            // 这里简单地返回一个直接连接的代理
            return List.of(Proxy.NO_PROXY);
        }

        @Override
        public void connectFailed(URI uri, InetSocketAddress sa, IOException ioe) {
            // 处理连接失败的情况
            System.out.println("Failed to connect to: " + uri);
        }
    }
}