java-design-patterns/extension-objects/src/main/java/units/CommanderUnit.java

package units;

import abstractextensions.UnitExtension;
import concreteextensions.Commander;

/**
 * Class defining CommanderUnit
 */
public class CommanderUnit extends Unit {

  public CommanderUnit(String name) {
    super(name);
  }

  @Override
  public UnitExtension getUnitExtension(String extensionName) {

    if (extensionName.equals("CommanderExtension")) {
      if (unitExtension == null) {
        unitExtension = new Commander(this);
      }
      return unitExtension;
    }

    return super.getUnitExtension(extensionName);
  }
}