rstoyanchev
1 year ago
5 changed files with 211 additions and 147 deletions
@ -0,0 +1,160 @@
@@ -0,0 +1,160 @@
|
||||
/* |
||||
* Copyright 2002-2025 the original author or authors. |
||||
* |
||||
* 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 |
||||
* |
||||
* https://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.springframework.web.service.registry; |
||||
|
||||
import java.util.Collection; |
||||
import java.util.LinkedHashMap; |
||||
import java.util.LinkedHashSet; |
||||
import java.util.Map; |
||||
import java.util.Set; |
||||
import java.util.function.Consumer; |
||||
import java.util.stream.Collectors; |
||||
|
||||
import org.springframework.util.Assert; |
||||
import org.springframework.util.ClassUtils; |
||||
|
||||
/** |
||||
* Container for HTTP Service type registrations, initially storing HTTP Service |
||||
* type names as {@link Registration}s, and later exposing access to those |
||||
* registrations as {@link HttpServiceGroup}s via {@link #groups()}. |
||||
* |
||||
* @author Rossen Stoyanchev |
||||
* @since 7.0 |
||||
*/ |
||||
final class GroupsMetadata { |
||||
|
||||
private final Map<String, DefaultRegistration> groupMap = new LinkedHashMap<>(); |
||||
|
||||
|
||||
/** |
||||
* Create a registration for the given group name, or return an existing |
||||
* registration. If there is an existing registration, merge the client |
||||
* types after checking they don't conflict. |
||||
*/ |
||||
public Registration getOrCreateGroup(String groupName, HttpServiceGroup.ClientType clientType) { |
||||
return this.groupMap.computeIfAbsent(groupName, name -> new DefaultRegistration(name, clientType)) |
||||
.clientType(clientType); |
||||
} |
||||
|
||||
/** |
||||
* Merge all registrations from the given {@link GroupsMetadata} into this one. |
||||
*/ |
||||
public void mergeWith(GroupsMetadata other) { |
||||
other.forEachRegistration(registration -> |
||||
getOrCreateGroup(registration.name(), registration.clientType()) |
||||
.httpServiceTypeNames() |
||||
.addAll(registration.httpServiceTypeNames())); |
||||
} |
||||
|
||||
public void forEachRegistration(Consumer<Registration> consumer) { |
||||
this.groupMap.values().forEach(consumer); |
||||
} |
||||
|
||||
public Collection<HttpServiceGroup> groups() { |
||||
return this.groupMap.values().stream().map(DefaultRegistration::toHttpServiceGroup).toList(); |
||||
} |
||||
|
||||
public static Class<?> loadClass(String type) { |
||||
try { |
||||
return ClassUtils.forName(type, GroupsMetadata.class.getClassLoader()); |
||||
} |
||||
catch (ClassNotFoundException ex) { |
||||
throw new IllegalStateException("Failed to load '" + type + "'", ex); |
||||
} |
||||
} |
||||
|
||||
|
||||
/** |
||||
* Registration metadata for an {@link HttpServiceGroup}. |
||||
*/ |
||||
interface Registration { |
||||
|
||||
String name(); |
||||
|
||||
HttpServiceGroup.ClientType clientType(); |
||||
|
||||
Set<String> httpServiceTypeNames(); |
||||
} |
||||
|
||||
|
||||
/** |
||||
* Default implementation of {@link Registration}. |
||||
*/ |
||||
private static class DefaultRegistration implements Registration { |
||||
|
||||
private final String name; |
||||
|
||||
private HttpServiceGroup.ClientType clientType; |
||||
|
||||
private final Set<String> typeNames = new LinkedHashSet<>(); |
||||
|
||||
DefaultRegistration(String name, HttpServiceGroup.ClientType clientType) { |
||||
this.name = name; |
||||
this.clientType = clientType; |
||||
} |
||||
|
||||
@Override |
||||
public String name() { |
||||
return this.name; |
||||
} |
||||
|
||||
@Override |
||||
public HttpServiceGroup.ClientType clientType() { |
||||
return this.clientType; |
||||
} |
||||
|
||||
@Override |
||||
public Set<String> httpServiceTypeNames() { |
||||
return this.typeNames; |
||||
} |
||||
|
||||
/** |
||||
* Update the client type if it does not conflict with the existing value. |
||||
*/ |
||||
public DefaultRegistration clientType(HttpServiceGroup.ClientType other) { |
||||
if (this.clientType.isUnspecified()) { |
||||
this.clientType = other; |
||||
} |
||||
else { |
||||
Assert.isTrue(this.clientType == other || other.isUnspecified(), |
||||
"ClientType conflict for HttpServiceGroup '" + this.name + "'"); |
||||
} |
||||
return this; |
||||
} |
||||
|
||||
/** |
||||
* Create the {@link HttpServiceGroup} from the metadata. |
||||
*/ |
||||
public HttpServiceGroup toHttpServiceGroup() { |
||||
return new RegisteredGroup(this.name, |
||||
(this.clientType.isUnspecified() ? HttpServiceGroup.ClientType.REST_CLIENT : this.clientType), |
||||
this.typeNames.stream().map(GroupsMetadata::loadClass).collect(Collectors.toSet())); |
||||
} |
||||
|
||||
@Override |
||||
public String toString() { |
||||
return "Group '" + this.name + "', ClientType." + this.clientType + ", " + this.typeNames; |
||||
} |
||||
} |
||||
|
||||
|
||||
private record RegisteredGroup(String name, ClientType clientType, Set<Class<?>> httpServiceTypes) |
||||
implements HttpServiceGroup { |
||||
|
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue