cn.msuno.swagger.spring.boot.autoconfigure.mappers.InheritanceDeterminer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of swagger-spring-boot-starter Show documentation

Project for Spring Boot

There is a newer version: 2.5.2

Show newest version

/*
 *
 *  Copyright 2017-2019 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
 *
 *         http://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 cn.msuno.swagger.spring.boot.autoconfigure.mappers;

import static springfox.documentation.builders.BuilderDefaults.nullToEmptyList;

import java.util.HashMap;
import java.util.Map;

import io.swagger.models.RefModel;
import springfox.documentation.schema.Model;
import springfox.documentation.schema.ModelReference;

class InheritanceDeterminer {
  private final Map parentLookup = new HashMap();

  public InheritanceDeterminer(Map models) {
    for (Model each : models.values()) {
      for (ModelReference modelReference : nullToEmptyList(each.getSubTypes())) {
        parentLookup.put(modelReference.getType(), toRefModel(each));
      }
    }
  }

  public boolean hasParent(Model model) {
    return parentLookup.containsKey(model.getName());
  }

  public RefModel parent(Model model) {
    return parentLookup.get(model.getName());
  }

  private RefModel toRefModel(Model model) {
    return new RefModel(model.getName());
  }
}