fix FleetBuilder FlightTime with tx

alaingilbert · Sep 15, 2022 · 5537c50 · 5537c50
1 parent 7b3f580
commit 5537c50
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pkg/wrapper/fleetBuilder.go b/pkg/wrapper/fleetBuilder.go
@@ -202,6 +202,9 @@ func (f *FleetBuilder) FlightTime() (secs, fuel int64) {
 			ships, _ = f.b.GetShips(f.origin.GetID())
 		}
 	}
+	if f.tx != nil {
+		return f.tx.FlightTime(f.origin.GetCoordinate(), f.destination, f.speed, ships, f.mission)
+	}
 	return f.b.FlightTime(f.origin.GetCoordinate(), f.destination, f.speed, ships, f.mission)
 }