Union type returns the results of all union type members

[shafreenAnfar]

Query,

{
  search(status:OPEN) {
    ... on Trail {
      name
      trees
    }
  }
}

The result,

Here the third result should not be there. Because it is part of Lifts. Hence, trees end up being null.

{
  "data": {
    "search": [
      {
        "name": "Blue Bird",
        "trees": false
      },
      {
        "name": "Blackhawk",
        "trees": false
      },
      {
        "name": "Astra Express",
        "trees": null
      }
    ]
  }
}

Implementation,

resource function get search (ds:Status status) returns SearchResult[] {
        ds:TrailRecord[] trails = from var trail in ds:trailTable where trail.status == status select trail;
        SearchResult[] searchResults = trails.map(function (ds:TrailRecord trail) returns Trail => new Trail(trail));

        ds:LiftRecord[] lifts = from var lift in ds:liftTable where lift.status == status select lift;
        lifts.forEach(function (ds:LiftRecord lift) returns () => searchResults.push(new Lift(lift)));

        return searchResults;
    }