diff --git a/core/src/main/java/ru/trader/core/Place.java b/core/src/main/java/ru/trader/core/Place.java index aef92b1..7b52001 100644 --- a/core/src/main/java/ru/trader/core/Place.java +++ b/core/src/main/java/ru/trader/core/Place.java @@ -70,4 +70,8 @@ public interface Place extends Connectable { vendor.clear(); } } + + default Vendor asTransit(){ + return new TransitVendor(this); + } } diff --git a/core/src/main/java/ru/trader/core/VendorsIterator.java b/core/src/main/java/ru/trader/core/VendorsIterator.java index da96c63..54423f8 100644 --- a/core/src/main/java/ru/trader/core/VendorsIterator.java +++ b/core/src/main/java/ru/trader/core/VendorsIterator.java @@ -31,7 +31,7 @@ public class VendorsIterator implements Iterator { nextVendor(); } else { if (includeTransit) - next = new TransitVendor(place); + next = place.asTransit(); else nextPlace(); }