CODEWITHSUNDEEP
ruby-on-railsrubyarraysruby-on-rails-3
Pi Horse
Pi Horse

Reputation: 2430

Ruby - Splititng Array after getting data from Mysql

I have the following code to fetch the data from MySQL database into my rails controller

@main = $connection.execute("SELECT * FROM builds WHERE platform_type IS NOT NULL")

This returns a mysql2 type object which behaves like an array i guess.

I want to split this into 2 arrays, first one where platform_type is 'TOTAL' and everything else in the other array.

Upvotes: 0

Views: 197

Answers (2)

Aleksei Matiushkin
Aleksei Matiushkin

Reputation: 121000

It actually returns a Mysql2::Result object. Of course you can do

totals = []
others = []
main.each { |r|
  (r['platform_type'] == 'TOTAL' ? totals : others) << r
}

but why not use a rails way with smth like:

Builds.where("platform_type = ?", 'TOTAL')
Builds.where("platform_type NOT IN ?", [nil, 'TOTAL'])

Upvotes: 2

Simon Woodside
Simon Woodside

Reputation: 7314

Try array.select. Something like

total = @main.select { |build| build.platform_type == 'TOTAL' }
not_total = @main.reject { |build| build.platform_type == 'TOTAL' }

http://matthewcarriere.com/2008/06/23/using-select-reject-collect-inject-and-detect/

Even better, use Enumerable.partition as per this answer: Ruby Select and Reject in one method

Upvotes: -1

Related Questions