- segments = Segment.find(:all, :condtions => ['node_a in ? or node_b in ?', node_ids, node_ids])
- render :text => node_ids.join(',')
- return
+ # get the referenced segments
+ segments = Segment.find_by_sql "select * from segments where node_a in #{node_ids_sql} or node_b in #{node_ids_sql}"
+
+ # see if we have nay missing nodes
+ segments_nodes = segments.collect {|segment| segment.node_a }
+ segments_nodes += segments.collect {|segment| segment.node_b }
+
+ segments_nodes.uniq!
+
+ missing_nodes = segments_nodes - node_ids
+
+ # get missing nodes if there are any
+ nodes += Node.find(missing_nodes) if missing_nodes.length > 0