pub.codex.apix.scan.ApiOperationReader Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of codex-apix Show documentation

Code-X 让程序狗喝茶

There is a newer version: 5.0.2

package pub.codex.apix.scan;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.bind.annotation.RequestMethod;
import pub.codex.apix.context.OperationContext;
import pub.codex.apix.context.RequestMappingContext;
import pub.codex.apix.operation.OperationBuilderPlugin;
import pub.codex.apix.schema.Operation;

import java.util.Arrays;
import java.util.List;
import java.util.Set;

import static com.google.common.collect.Lists.newArrayList;

@Component
public class ApiOperationReader {


    private OperationBuilderPlugin[] operationBuilderPlugins;

    @Autowired
    public ApiOperationReader(OperationBuilderPlugin[] operationBuilderPlugins) {
        this.operationBuilderPlugins = operationBuilderPlugins;
    }

    /**
     * 读取 API 选项的描述信息
     *
     * @param context
     */
    public List read(RequestMappingContext context) {

        Set requestMethods = context.getMethodsCondition();

        List operations = newArrayList();

        for (RequestMethod httpRequestMethod : requestMethods) {

            OperationContext operationContext = new OperationContext(httpRequestMethod, context);

            // 遍历应用方法
            Arrays.stream(operationBuilderPlugins).forEach(plugin -> {
                plugin.apply(operationContext);
            });

            operations.add(operationContext.operationBuilder().build());
        }

        return operations;
    }


}