Make AnnotatedService public

core/src/main/java/com/linecorp/armeria/internal/common/util/ServiceNamingUtil.java

@@ -16,8 +16,8 @@
 package com.linecorp.armeria.internal.common.util;
 
 import com.linecorp.armeria.common.util.Unwrappable;
-import com.linecorp.armeria.internal.server.annotation.AnnotatedService;
 import com.linecorp.armeria.server.HttpService;
+import com.linecorp.armeria.server.annotation.AnnotatedService;
 
 public final class ServiceNamingUtil {
 

core/src/main/java/com/linecorp/armeria/internal/server/annotation/AnnotatedDocServicePlugin.java

@@ -62,6 +62,7 @@
 import com.linecorp.armeria.server.RoutePathType;
 import com.linecorp.armeria.server.Service;
 import com.linecorp.armeria.server.ServiceConfig;
+import com.linecorp.armeria.server.annotation.AnnotatedService;
 import com.linecorp.armeria.server.annotation.Header;
 import com.linecorp.armeria.server.annotation.Param;
 import com.linecorp.armeria.server.annotation.RequestObject;
@@ -144,7 +145,7 @@ public ServiceSpecification generateSpecification(Set<ServiceConfig> serviceConf
         serviceConfigs.forEach(sc -> {
             final AnnotatedService service = sc.service().as(AnnotatedService.class);
             if (service != null) {
-                final Class<?> serviceClass = ClassUtil.getUserClass(service.object().getClass());
+                final Class<?> serviceClass = ClassUtil.getUserClass(service.serviceObject().getClass());
                 final String className = serviceClass.getName();
                 final String methodName = service.method().getName();
                 if (!filter.test(name(), className, methodName)) {
@@ -171,7 +172,8 @@ private static void addMethodInfo(Map<Class<?>, Set<MethodInfo>> methodInfos,
         final Method method = service.method();
         final int overloadId = service.overloadId();
         final TypeSignature returnTypeSignature = getReturnTypeSignature(method);
-        final List<FieldInfo> fieldInfos = fieldInfos(service.annotatedValueResolvers());
+        final List<FieldInfo> fieldInfos = fieldInfos(service.as(DefaultAnnotatedService.class)
+                                                             .annotatedValueResolvers());
         route.methods().forEach(
                 httpMethod -> {
                     final MethodInfo methodInfo = new MethodInfo(

core/src/main/java/com/linecorp/armeria/internal/server/annotation/AnnotatedServiceElement.java

@@ -24,6 +24,7 @@
 
 import com.linecorp.armeria.server.HttpService;
 import com.linecorp.armeria.server.Route;
+import com.linecorp.armeria.server.annotation.AnnotatedService;
 
 /**
  * Details of an annotated HTTP service method.
@@ -78,7 +79,8 @@ public HttpService buildSafeDecoratedService(
         HttpService decoratedService = decorator.apply(service);
         // Apply localDecorator passed in through method parameter
         decoratedService = decoratedService.decorate(localDecorator);
-        return service.withExceptionHandler(decoratedService);
+        return service.as(DefaultAnnotatedService.class)
+                      .withExceptionHandler(decoratedService);
     }
 
     @Override

core/src/main/java/com/linecorp/armeria/internal/server/annotation/AnnotatedServiceFactory.java

@@ -80,6 +80,7 @@
 import com.linecorp.armeria.server.Route;
 import com.linecorp.armeria.server.annotation.AdditionalHeader;
 import com.linecorp.armeria.server.annotation.AdditionalTrailer;
+import com.linecorp.armeria.server.annotation.AnnotatedService;
 import com.linecorp.armeria.server.annotation.Blocking;
 import com.linecorp.armeria.server.annotation.Consumes;
 import com.linecorp.armeria.server.annotation.Decorator;
@@ -291,8 +292,10 @@ static List<AnnotatedServiceElement> create(String pathPrefix, Object object, Me
                                                queryDelimiter);
             return new AnnotatedServiceElement(
                     route,
-                    new AnnotatedService(object, method, overloadId, resolvers, eh, res, route, defaultStatus,
-                                         responseHeaders, responseTrailers, needToUseBlockingTaskExecutor),
+                    new DefaultAnnotatedService(object, method, overloadId,
+                                                resolvers, eh, res, route, defaultStatus,
+                                                responseHeaders, responseTrailers,
+                                                needToUseBlockingTaskExecutor),
                     decorator(method, clazz, dependencyInjector));
         }).collect(toImmutableList());
     }

core/src/main/java/com/linecorp/armeria/internal/server/annotation/AnnotatedValueResolver.java

@@ -84,6 +84,7 @@
 import com.linecorp.armeria.internal.server.FileAggregatedMultipart;
 import com.linecorp.armeria.internal.server.annotation.AnnotatedBeanFactoryRegistry.BeanFactoryId;
 import com.linecorp.armeria.server.ServiceRequestContext;
+import com.linecorp.armeria.server.annotation.AnnotatedService;
 import com.linecorp.armeria.server.annotation.ByteArrayRequestConverterFunction;
 import com.linecorp.armeria.server.annotation.Default;
 import com.linecorp.armeria.server.annotation.Delimiter;

core/src/main/java/com/linecorp/armeria/internal/server/annotation/AnnotatedService.java → core/src/main/java/com/linecorp/armeria/internal/server/annotation/DefaultAnnotatedService.java

@@ -63,6 +63,7 @@
 import com.linecorp.armeria.server.RoutingContext;
 import com.linecorp.armeria.server.ServiceRequestContext;
 import com.linecorp.armeria.server.SimpleDecoratingHttpService;
+import com.linecorp.armeria.server.annotation.AnnotatedService;
 import com.linecorp.armeria.server.annotation.ExceptionHandlerFunction;
 import com.linecorp.armeria.server.annotation.ExceptionVerbosity;
 import com.linecorp.armeria.server.annotation.FallthroughException;
@@ -76,8 +77,8 @@
  * This class is not supposed to be instantiated by a user. Please check out the documentation
  * <a href="https://armeria.dev/docs/server-annotated-service">Annotated HTTP Service</a> to use this.
  */
-public final class AnnotatedService implements HttpService {
-    private static final Logger logger = LoggerFactory.getLogger(AnnotatedService.class);
+public final class DefaultAnnotatedService implements AnnotatedService {
+    private static final Logger logger = LoggerFactory.getLogger(DefaultAnnotatedService.class);
 
     private static final MethodHandles.Lookup lookup = MethodHandles.lookup();
 
@@ -106,18 +107,19 @@ public final class AnnotatedService implements HttpService {
 
     private final ResponseType responseType;
     private final boolean useBlockingTaskExecutor;
-    private final String serviceName;
+    @Nullable
+    private String serviceName;
     private final boolean serviceNameSetByAnnotation;
 
-    AnnotatedService(Object object, Method method,
-                     int overloadId, List<AnnotatedValueResolver> resolvers,
-                     List<ExceptionHandlerFunction> exceptionHandlers,
-                     List<ResponseConverterFunction> responseConverters,
-                     Route route,
-                     HttpStatus defaultStatus,
-                     HttpHeaders defaultHttpHeaders,
-                     HttpHeaders defaultHttpTrailers,
-                     boolean useBlockingTaskExecutor) {
+    DefaultAnnotatedService(Object object, Method method,
+                            int overloadId, List<AnnotatedValueResolver> resolvers,
+                            List<ExceptionHandlerFunction> exceptionHandlers,
+                            List<ResponseConverterFunction> responseConverters,
+                            Route route,
+                            HttpStatus defaultStatus,
+                            HttpHeaders defaultHttpHeaders,
+                            HttpHeaders defaultHttpTrailers,
+                            boolean useBlockingTaskExecutor) {
         this.object = requireNonNull(object, "object");
         this.method = requireNonNull(method, "method");
         checkArgument(overloadId >= 0, "overloadId: %s (expected: >= 0)", overloadId);
@@ -168,7 +170,6 @@ public final class AnnotatedService implements HttpService {
             this.serviceName = serviceName.value();
             serviceNameSetByAnnotation = true;
         } else {
-            this.serviceName = ClassUtil.getUserClass(object.getClass()).getName();
             serviceNameSetByAnnotation = false;
         }
 
@@ -221,39 +222,52 @@ private static void warnIfHttpResponseArgumentExists(Type returnType,
     }
 
     public String serviceName() {
+        if (serviceName == null) {
+            return serviceClass().getName();
+        }
         return serviceName;
     }
 
     public boolean serviceNameSetByAnnotation() {
         return serviceNameSetByAnnotation;
     }
 
+    @Override
     public String methodName() {
         return method.getName();
     }
 
-    Object object() {
+    @Override
+    public Object serviceObject() {
         return object;
     }
 
-    Method method() {
+    @Override
+    public Class<?> serviceClass() {
+        return ClassUtil.getUserClass(object.getClass());
+    }
+
+    @Override
+    public Method method() {
         return method;
     }
 
-    int overloadId() {
+    @Override
+    public int overloadId() {
         return overloadId;
     }
 
     List<AnnotatedValueResolver> annotatedValueResolvers() {
         return resolvers;
     }
 
-    Route route() {
+    @Override
+    public Route route() {
         return route;
     }
 
-    // TODO: Expose through `AnnotatedServiceConfig`, see #5382.
-    HttpStatus defaultStatus() {
+    @Override
+    public HttpStatus defaultStatus() {
         return defaultStatus;
     }
 

core/src/main/java/com/linecorp/armeria/internal/server/annotation/HttpResultUtil.java

@@ -23,6 +23,7 @@
 import com.linecorp.armeria.common.ResponseHeaders;
 import com.linecorp.armeria.common.ResponseHeadersBuilder;
 import com.linecorp.armeria.server.ServiceRequestContext;
+import com.linecorp.armeria.server.annotation.AnnotatedService;
 import com.linecorp.armeria.server.annotation.HttpResult;
 
 final class HttpResultUtil {

core/src/main/java/com/linecorp/armeria/server/ContextPathAnnotatedServiceConfigSetters.java

@@ -27,7 +27,7 @@
 import com.linecorp.armeria.common.SuccessFunction;
 import com.linecorp.armeria.common.annotation.UnstableApi;
 import com.linecorp.armeria.common.util.BlockingTaskExecutor;
-import com.linecorp.armeria.internal.server.annotation.AnnotatedService;
+import com.linecorp.armeria.server.annotation.AnnotatedService;
 import com.linecorp.armeria.server.annotation.ExceptionHandlerFunction;
 import com.linecorp.armeria.server.annotation.RequestConverterFunction;
 import com.linecorp.armeria.server.annotation.ResponseConverterFunction;

core/src/main/java/com/linecorp/armeria/server/DefaultServerErrorHandler.java

@@ -33,7 +33,7 @@
 import com.linecorp.armeria.common.annotation.Nullable;
 import com.linecorp.armeria.common.util.Exceptions;
 import com.linecorp.armeria.internal.common.util.TemporaryThreadLocals;
-import com.linecorp.armeria.internal.server.annotation.AnnotatedService;
+import com.linecorp.armeria.server.annotation.AnnotatedService;
 
 /**
  * The default {@link ServerErrorHandler} that is used when a user didn't specify one.

core/src/main/java/com/linecorp/armeria/server/DefaultServiceConfigSetters.java

@@ -42,7 +42,8 @@
 import com.linecorp.armeria.common.annotation.Nullable;
 import com.linecorp.armeria.common.util.BlockingTaskExecutor;
 import com.linecorp.armeria.common.util.EventLoopGroups;
-import com.linecorp.armeria.internal.server.annotation.AnnotatedService;
+import com.linecorp.armeria.internal.server.annotation.DefaultAnnotatedService;
+import com.linecorp.armeria.server.annotation.AnnotatedService;
 import com.linecorp.armeria.server.logging.AccessLogWriter;
 
 import io.netty.channel.EventLoopGroup;
@@ -338,7 +339,8 @@ ServiceConfigBuilder toServiceConfigBuilder(Route route, String contextPath, Htt
         } else {
             // Set the default service name only when the service name is set using @ServiceName.
             // If it's not, the global defaultServiceNaming is used.
-            if (annotatedService != null && annotatedService.serviceNameSetByAnnotation()) {
+            if (annotatedService != null && annotatedService.as(DefaultAnnotatedService.class)
+                                                            .serviceNameSetByAnnotation()) {
                 serviceConfigBuilder.defaultServiceName(annotatedService.serviceName());
             }
         }

core/src/main/java/com/linecorp/armeria/server/VirtualHostContextPathAnnotatedServiceConfigSetters.java

@@ -26,7 +26,7 @@
 import com.linecorp.armeria.common.SuccessFunction;
 import com.linecorp.armeria.common.annotation.UnstableApi;
 import com.linecorp.armeria.common.util.BlockingTaskExecutor;
-import com.linecorp.armeria.internal.server.annotation.AnnotatedService;
+import com.linecorp.armeria.server.annotation.AnnotatedService;
 import com.linecorp.armeria.server.annotation.ExceptionHandlerFunction;
 import com.linecorp.armeria.server.annotation.RequestConverterFunction;
 import com.linecorp.armeria.server.annotation.ResponseConverterFunction;

core/src/main/java/com/linecorp/armeria/server/annotation/AnnotatedService.java

@@ -0,0 +1,77 @@
+/*
+ * Copyright 2017 LINE Corporation
+ *
+ * LINE Corporation licenses this file to you 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 com.linecorp.armeria.server.annotation;
+
+import java.lang.reflect.Method;
+
+import com.linecorp.armeria.common.HttpStatus;
+import com.linecorp.armeria.server.HttpService;
+import com.linecorp.armeria.server.Route;
+import com.linecorp.armeria.server.ServerBuilder;
+
+/**
+ * An {@link HttpService} which is defined by a {@link Path} or HTTP method annotations.
+ * This class is designed to provide a common interface for {@link AnnotatedService}
+ * internally. Please check out the documentation at
+ * <a href="https://armeria.dev/docs/server-annotated-service">Annotated HTTP Service</a> to use this.
+ */
+public interface AnnotatedService extends HttpService {
+
+    /**
+     * Returns service name for this {@link AnnotatedService}.
+     */
+    String serviceName();
+
+    /**
+     * Returns method name which is annotated in {@link AnnotatedService}.
+     */
+    String methodName();
+
+    /**
+     * Returns the annotated service object specified with {@link ServerBuilder#annotatedService(Object)}.
+     */
+    Object serviceObject();
+
+    /**
+     * Returns the annotated service object specified with {@link ServerBuilder#annotatedService(Object)}.
+     */
+    Class<?> serviceClass();
+
+    /**
+     * Returns the target {@link Method} invoked when the request is received.
+     */
+    Method method();
+
+    /**
+     * Returns a unique ID to distinguish overloaded methods.
+     * If the method is not overloaded, it should return 0.
+     */
+    int overloadId();
+
+    /**
+     * Returns {@link Route} for this {@link AnnotatedService}.
+     */
+    Route route();
+
+    /**
+     * Returns the default {@link HttpStatus} specified with {@link StatusCode}.
+     * If {@link StatusCode} is not given, {@link HttpStatus#OK} is returned by default.
+     * If the method returns a void type such as {@link Void} or Kotlin Unit, {@link HttpStatus#NO_CONTENT} is
+     * returned.
+     */
+    HttpStatus defaultStatus();
+}

kotlin/src/main/java/com/linecorp/armeria/common/kotlin/CoroutineContexts.java

@@ -20,7 +20,7 @@
 
 import com.linecorp.armeria.common.RequestContext;
 import com.linecorp.armeria.common.annotation.Nullable;
-import com.linecorp.armeria.internal.server.annotation.AnnotatedService;
+import com.linecorp.armeria.server.annotation.AnnotatedService;
 import com.linecorp.armeria.server.kotlin.CoroutineContextService;
 
 import io.netty.util.AttributeKey;