PIbd-23_Dolgov_D.A._Airbus._Hard/EntityWithIlumCreator.java

import java.util.ArrayList;
import java.util.Random;

public class EntityWithIlumCreator <T extends EntityAirbus, U extends IDrawingIlum>
{
    private final ArrayList<T> entityArr;
    private final ArrayList<U> IlumArr;

    public EntityWithIlumCreator() {
        entityArr = new ArrayList<T>();
        IlumArr =  new ArrayList<U>();
    }

    public void Insert(T entityAirbus) {
        entityArr.add(entityAirbus);
    }

    public void Insert (U ilum) {
        IlumArr.add(ilum);
    }

    public DrawingAirbus getEntityWithExtra() {
        Random random = new Random();
        int getEntityRandomIndex = random.nextInt(entityArr.size());
        int getExtraRandomIndex = random.nextInt(IlumArr.size());

        EntityAirbus airbus = entityArr.get(getEntityRandomIndex);
        IDrawingIlum ilum = IlumArr.get(getExtraRandomIndex);

        if (airbus instanceof EntityAdvancedAirbus) {
            return  new DrawingAdvancedAirbus(airbus, ilum);
        }
        return new DrawingAirbus(airbus, ilum);
    }
}