Java 按后缀规则匹配获取指定服务关联的PORT、HOST环境变量值

核心结论 #

完全不需要对生成的map二次执行筛选操作，甚至连中间map的生成都可以省略。基于现有的后缀匹配逻辑，在过滤阶段直接提取目标PORT、HOST值，是性能和可维护性最高的实现。

注意：由于你采用的是后缀匹配规则，环境变量的完整键是「未知前缀 + _ + 传入serviceName + _PORT/HOST」，你根本没法预知完整键名，因此不能直接用固定键名从map里取值，这也是你最初需要做筛选的核心原因。

方案对比与选择 #

1. 已保留中间map场景下的最低成本取值 #

如果你已经写好了生成map的逻辑，不想做大段修改，最划算的方式是直接对这个小map做一次遍历：

• 这个map的最大size只有2，单次遍历的循环开销可以忽略
• 不需要重复编写stream的过滤判断逻辑，避免冗余代码
  示例代码：

String targetPort = null;
String targetHost = null;
for (Map.Entry<String, String> entry : map.entrySet()) {
    String key = entry.getKey();
    // 匹配带下划线边界的后缀，避免误判
    if (key.endsWith("_" + serviceName + "_PORT")) {
        targetPort = entry.getValue();
    } else if (key.endsWith("_" + serviceName + "_HOST")) {
        targetHost = entry.getValue();
    }
}

为什么不推荐二次跑stream filter的写法：很多人会图省事写两轮stream分别找PORT、HOST，这种方式相当于把小map遍历了两次，还得重复写两遍后缀判断逻辑，后续修改匹配规则时很容易漏改，维护成本更高，完全不划算。

2. 源头优化的推荐方案 #

既然最终只需要PORT、HOST两个值，完全没必要把筛选结果收集成HashMap。在遍历环境变量的阶段就直接提取值，全程只遍历一次System.getenv()，没有多余的对象构建开销。
可以用一个简单的载体类存两个值（Java16+可以直接用record简化）：

// 定义结果载体，简化取值后的传递
record ServiceEndpoint(String host, String port) {}

直接遍历环境变量完成提取即可，比原来的stream写法逻辑更直接：

String port = null;
String host = null;
for (Map.Entry<String, String> entry : System.getenv().entrySet()) {
    String key = entry.getKey();
    if (key.endsWith("_" + serviceName + "_PORT")) {
        port = entry.getValue();
    } else if (key.endsWith("_" + serviceName + "_HOST")) {
        host = entry.getValue();
    }
}
ServiceEndpoint endpoint = new ServiceEndpoint(host, port);

这个方案的优势：

• 省去了Collectors.toMap构建HashMap的内存开销
• 没有后续从map中查找、二次遍历的步骤
• 对重复匹配的场景可以灵活处理（比如遇到多个同后缀的值时直接覆盖、主动抛业务异常，不需要依赖toMap的默认行为抛出模糊的重复键运行时异常）

原代码的两个优化提示 #

• 原匹配逻辑endsWith(serviceName + "_PORT")存在边界缺陷：比如传入serviceName为"SERVICE1"时，FAKE_SERVICE11_PORT这类不符合预期的键也会被误匹配，建议加上前置下划线边界，改成endsWith("_" + serviceName + "_PORT")。
• 原Collectors.toMap的收集逻辑，在遇到多个同后缀的匹配键时，会直接抛出无上下文的重复键运行时异常，如果你的场景可能出现多匹配，需要提前加上重复值的处理逻辑。

内容的提问来源于stack exchange，提问作者wazza1010