com.wordnik.swagger.codegen.CodegenModelFactory Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of swagger-codegen Show documentation

The newest version!

package com.wordnik.swagger.codegen;

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

public final class CodegenModelFactory {

  private static final Map> typeMapping = new HashMap>();

  /**
   * Configure a different implementation class.
   * @param type the type that shall be replaced
   * @param implementation the implementation class must extend the default class and must provide a public no-arg constructor
   */
  public static void setTypeMapping(CodegenModelType type, Class implementation) {
    if (!type.getDefaultImplementation().isAssignableFrom(implementation)) {
      throw new IllegalArgumentException(implementation.getSimpleName() + " doesn't extend " + type.getDefaultImplementation().getSimpleName());
    }
    try {
      implementation.newInstance();
    } catch (Exception e) {
      throw new IllegalArgumentException(e);
    }
    typeMapping.put(type, implementation);
  }

  @SuppressWarnings("unchecked")
  public static  T newInstance(CodegenModelType type) {
    Class classType = typeMapping.get(type);
    try {
      return (T) (classType != null ? classType : type.getDefaultImplementation()).newInstance();
    } catch (InstantiationException e) {
      throw new RuntimeException(e);
    } catch (IllegalAccessException e) {
      throw new RuntimeException(e);
    }
  }
}