about summary refs log tree commit diff
diff options
context:
space:
mode:
authorEugen Rochko <eugen@zeonfederated.com>2016-10-15 12:58:34 +0200
committerEugen Rochko <eugen@zeonfederated.com>2016-10-15 13:00:21 +0200
commitedefd87adfadd6c1d7f8087459406464815a5fd0 (patch)
treeb0338b28350f3508856812e2d515b8a1cbb5d61f
parent70ab6624f5f397ddd05136db2fa37c902f0867eb (diff)
Fix multiple results return from FollowSuggestion.get
-rw-r--r--app/models/follow_suggestion.rb4
1 files changed, 2 insertions, 2 deletions
diff --git a/app/models/follow_suggestion.rb b/app/models/follow_suggestion.rb
index f4515700a..c14dba672 100644
--- a/app/models/follow_suggestion.rb
+++ b/app/models/follow_suggestion.rb
@@ -1,8 +1,8 @@
 class FollowSuggestion
-  def self.get(for_account_id)
+  def self.get(for_account_id, limit = 6)
     neo = Neography::Rest.new
     account_ids = neo.execute_query('START a=node:account_index(Account={id}) MATCH (a)-[:follows]->(b)-[:follows]->(c) WHERE a <> c AND NOT (a)-[:follows]->(c) RETURN DISTINCT c.account_id', id: for_account_id)
-    Account.where(id: account_ids['data'].first) unless account_ids.empty?
+    Account.where(id: account_ids['data'].map(&:first)).limit(limit) unless account_ids.empty?
   rescue Neography::NeographyError, Excon::Error::Socket => e
     Rails.logger.error e
     []