Spring REST Docs によるAPI仕様書自動化

2016.10.12

こんにちは。サーバサイドエンジニアをしている大堀です。

今回は、私が所属しているプロジェクトで導入しているSpring REST Docsによる、
API仕様書自動化方法に関して紹介します。

導入するに至った背景

私が今までのプロジェクトでは、APIドキュメントを手動で作成していましたが、
記入ミスや、時間が経つにつれて更新が滞ったり、実装と異なってくるという問題が生じていました。

そこで、Spring MVCの枠組みで作成されたアプリケーションから、
自動でドキュメントを生成してくれるSpring REST Docsを導入することにしました。

Spring REST Docsについて

Spring REST DocsはRESTfulなサービスのドキュメントを生成するツールです。

Spring MVC Testで書かれたテストを通った場合、HTTPリクエストやレスポンス等が記載されたsnippetsをasciidoctorで生成します。
テストを通らなかった場合は、ドキュメントは生成されません。

asciidoctorはhtmlを生成することができ、また、必要に応じてUIをカスタマイズすることができます。

Spring REST Docsを導入することで、以下のメリットがあります。

API仕様書を自動作成するため、仕様書がコードと同様に保たれる。また作成の手間も省ける。
見やすいAPI仕様書が手軽に作成でき、またUIのカスタマイズも容易である。

利用方法について

次に、Spring REST Docsの導入方法と利用方法を紹介します。

インストール

Spring REST Docsをインストールするには、pom.xmlに以下を追記します。


    ${project.build.directory}/generated-snippets



    org.apache.maven.plugins
    maven-site-plugin
    3.2
    
        
        org.asciidoctor
        asciidoctor-maven-plugin
        1.5.2
        
    



    org.asciidoctor
    asciidoctor-maven-plugin
    1.5.2
    
        
            generate-docs
            package ー①
            
            process-asciidoc
            
            
            true ー②
            html ー③
            book
            
            ${basedir}/src/main/asciidoc/asciidoc.css ー④
            ${project.build.directory}/generated-snippets ー⑤
            
            
        
    



    org.springframework.restdocs
    spring-restdocs-mockmvc
    1.0.2.BUILD-SNAPSHOT
    test

上記pom.xmlについて説明します。

いつドキュメントを生成するか指定します。今回はパッケージング時に生成しています。
asciidocのディレクトリ構成と同じディレクトリ構成でhtml格納ディレクトリを生成します。
ドキュメントの出力形式を指定します。今回はhtml形式で出力しています。
デフォルトで読み込むcssを指定します。
自動生成するasciidoc(request header,response header,request body, response body等毎)の格納フォルダを指定します。

テストを書く

APIドキュメントを生成するためのテストを書きます。

テストを通った時にgenerated-snippets配下へasciidocファイルが生成されます。
テストを通らないと生成されないので頑張って通します。

以下が、テスト例となります。

public class RestDocsSampleTest extends ApiTestBase {

    /**
     * 初期設定
     *
     * @throws Exception
     */
    @Before
    public void buildMockMvcBefore() throws Exception {
        super.before(); ー①
    }

    /**
     * GetAPI仕様書生成例
     *
     * @throws Exception
     */
    @Test
    public void testRequestMapping() throws Exception { ー②
        // Snippetへ記載されるリクエストボディ
        // リクエストで使用していて仕様書へ記載する必要ないフィールドは末尾へ.ignore()を追記。（しないとエラーが出ます）
        List responseFieldDescriptorList = super.addBaseResponseFieldList();
        responseFieldDescriptorList.add(fieldWithPath("userId").description("The persons' ID")); ー③
            responseFieldDescriptorList.add(fieldWithPath("name").description("The persons' last name")).ignored());ー④

            // Snippetを作成します
            // Headerの内容を仕様書へ記載したくない場合はcreateHeaderSnippetsBooleanをfalseにして渡して下さい。
            super.createResponseFieldSnippets(responseFieldDescriptorList, super.createHeaderSnippetsBoolean);

        // Getリクエスト正常テスト
        super.getRequest("/hoge/fuga/1");
    }

    /**
     * PostAPI仕様書生成例
     *
     * @throws Exception
     */
    @Test
    public void testPostRequestMapping() throws Exception {
        // Snippetへ記載されるリクエストボディ
        // リクエストで使用していて仕様書へ記載する必要ないフィールドは末尾へ.ignore()を追記。（しないとエラーが出ます）
        List requestFieldDescriptorList = super.addBaseRequestFieldList();
        requestFieldDescriptorList.add(fieldWithPath("userId").description("The persons' ID"));
        requestFieldDescriptorList.add(fieldWithPath("name").description("The persons' ID").ignored());

        // Snippetへ記載するレスポンスボディ
        // レスポンスで使用していて仕様書へ記載する必要ないフィールドは末尾へ.ignore()を追記。（しないとエラーが出ます）
        List responseFieldDescriptorList = super.addBaseResponseFieldList();
        responseFieldDescriptorList.add(fieldWithPath("userId").description("The persons' ID"));
        responseFieldDescriptorList.add(fieldWithPath("name").description("The persons' last name").type("HogeObject")); ー⑤

            // Snippetを作成します
            // Headerの内容を仕様書へ記載したくない場合はcreateHeaderSnippetsBooleanをfalseにして渡して下さい。
            super.createRequestAndResponseFieldSnippets(requestFieldDescriptorList, responseFieldDescriptorList, !super.createHeaderSnippetsBoolean);

        SampleRequest sampleRequest = new SampleRequest();
        sampleRequest.setUserId(1);
        sampleRequest.setName("taro");

        // Postリクエスト正常テスト
        super.postRequest("/hoge/fuga/1", sampleRequest);
    }
}

上記コードについて説明します。

asciidoc出力設定を継承元クラスで行っています。継承元クラスは下部に記載があります。
API仕様書を作成するメソッド名はアルファベットで指定します。日本語でも指定できますが、asciidocが格納されるフォルダ名になるため、アルファベットが望ましいかと思います。
自動で生成されるAPI仕様書には以下が記載できます

fieldWithPath:フィールド名（Null不可）
description:フィールド説明（Null不可）
type:フィールド型
optional:descriptionやtypeを上書き
ignore:仕様書へ記載しないフィールド（仕様書へ記載しないフィールドも、フィールド名と説明は設定した上でこのメソッドを付与する必要があります）

リクエストやレスポンスのフィールドはAPI仕様書へ記載するしないに関わらず全てテスト上へ書いておく必要があります。書いていない場合エラーが出力されます。
リクエストフィールドやレスポンスフィールドの型は自動で読み込まれますが、独自型を設定している場合はtypeメソッドへStringで記載することで仕様書への出力が可能です。

ApiTestBase.Java

asciidocが生成されるフォルダの命名規則や出力フォーマットなどをまとめてあります。

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration("classpath:TestApplicationContext.xml")
@WebAppConfiguration
public class ApiTestBase {

    @Rule
    public final RestDocumentation restDocumentation = new RestDocumentation("target/generated-snippets");

    @Resource(name = "jsonObjectMapperFactory")
        protected ObjectMapper jsonObjectMapper;

    protected RestDocumentationResultHandler document;

    protected MockMvc mockMvc;

    protected boolean createHeaderSnippetsBoolean = true;

    /**
     * 初期設定
     *
     * @throws Exception
     */
    public void before() throws Exception {
        this.document = document("{class-name}/{method-name}", preprocessRequest(prettyPrint()), preprocessResponse(prettyPrint()));
        mockMvc = builder.apply(documentationConfiguration(this.restDocumentation)).alwaysDo(this.document).build();
    }

    private List addBaseRequestHeaderList() {
        List requestHeaderDescriptorList = new ArrayList();

        requestHeaderDescriptorList.add(headerWithName("X-Hoge").description("hoge"));

        return requestHeaderDescriptorList;
    }

    private List addBaseResponseHeaderList() {
        List responseHeaderDescriptorList = new ArrayList();

        responseHeaderDescriptorList.add(headerWithName("X-Fuga").description("fuga"));

        return responseHeaderDescriptorList;
    }

    protected List addBaseRequestFieldList() {
        List requestFieldDescriptorList = new ArrayList();

        return requestFieldDescriptorList;
    }

    protected List addBaseResponseFieldList() {
        List responseFieldDescriptorList = new ArrayList();

        responseFieldDescriptorList.add(fieldWithPath("statusCode").ignored());

        return responseFieldDescriptorList;
    }

    public void createRequestFieldSnippets(List requestFieldDescriptorList, boolean createHeaderSnippetsBoolean) {
        his.createRequestAndResponseFieldSnippets(requestFieldDescriptorList, null, createHeaderSnippetsBoolean);
    }

    public void createResponseFieldSnippets(List responseFieldDescriptorList, boolean createHeaderSnippetsBoolean) {
        this.createRequestAndResponseFieldSnippets(null, responseFieldDescriptorList, createHeaderSnippetsBoolean);
    }

    public void createRequestAndResponseFieldSnippets(List requestFieldDescriptorList, List responseFieldDescriptorList, boolean createHeaderSnippetsBoolean) {
        List snippetList = new ArrayList();

        // Headerも仕様書へ出力する
        if (createHeaderSnippetsBoolean) {
            snippetList.add(requestHeaders(this.addBaseRequestHeaderList().toArray(new HeaderDescriptor[0])));
            snippetList.add(responseHeaders(this.addBaseResponseHeaderList().toArray(new HeaderDescriptor[0])));
        }

        // リクエスト、レスポンスをテーブルへ出力する
        if (requestFieldDescriptorList != null &amp;&amp; responseFieldDescriptorList != null) {
            snippetList.add(requestFields(requestFieldDescriptorList.toArray(new FieldDescriptor[0])));
            snippetList.add(responseFields(responseFieldDescriptorList.toArray(new FieldDescriptor[0])));
        } else if (requestFieldDescriptorList != null) {
            snippetList.add(requestFields(requestFieldDescriptorList.toArray(new FieldDescriptor[0])));
        } else if (responseFieldDescriptorList != null) {
            snippetList.add(responseFields(responseFieldDescriptorList.toArray(new FieldDescriptor[0])));
        }

        this.document.snippets(snippetList.toArray(new Snippet[0]));
    }

    public void getRequest(String uri) throws Exception {
        this.mockMvc.perform(
                get(uri)
                .header("X-Hoge", "1")
                .accept(MediaType.APPLICATION_JSON)
                )
            .andExpect(status().isOk());
    }

    public void postRequest(String uri, Object object) throws Exception {
        this.mockMvc.perform(
                post(uri)
                .header("X-Hoge", "1")
                .contentType(MediaType.APPLICATION_JSON)
                .content(this.jsonObjectMapper.writeValueAsString(object))
                .accept(MediaType.APPLICATION_JSON)
                )
            .andExpect(status().isOk());
    }
}

base.adoc設定

生成されたasciidocを繋ぎ合わせるasciidocです。
プロジェクトでは、base.adocを元に、
テストから生成されたasciidoc(curl-request.adoc,http-request.adocなど)を繋ぎ合わせたasciidocをシェルスクリプトで作成しています。

= RESTful Notes API Guide

Sample;
:doctype: book
:icons: font
:source-highlighter: highlightjs
:sectlinks:

=== About API

API概要

.リクエスト構造

include::{snippets}/controller-name/method-name/curl-request.adoc[]

.リクエスト例

include::{snippets}/controller-name/method-name/http-request.adoc[]

.レスポンス例

include::{snippets}/controller-name/method-name/http-response.adoc[]

[caption=]
.リクエストヘッダー例

include::{snippets}/controller-name/method-name/request-headers.adoc[]

[caption=]
.レスポンスヘッダー例

include::{snippets}/controller-name/method-name/response-headers.adoc[]

[caption=]
.リクエストフィールド例

include::{snippets}/controller-name/method-name/request-fields.adoc[]

[caption=]
.レスポンスフィールド例

include::{snippets}/controller-name/method-name/response-fields.adoc[]