Initial commit

lib/tasks/distribution.rake

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+require 'csv'
+
+namespace :distribution do
+  # Detailed instruction https://github.com/rubykube/peatio/blob/master/docs/tasks/distribution.md
+  # Required fields for distribution:
+  # - uid
+  # - currency_id
+  # - amount
+  #
+  # Usage:
+  # For distribution process: -> bundle exec rake distribution:process['file_name.csv']
+
+  desc 'Distribution process'
+  task :process, [:config_load_path] => [:environment] do |_, args|
+    csv_table = File.read(Rails.root.join(args[:config_load_path]))
+    count = 0
+    errors_count = 0
+    CSV.parse(csv_table, headers: true, quote_empty: false).each do |row|
+      row = row.to_h.compact.symbolize_keys!
+      uid = row[:uid]
+      currency_id = row[:currency_id]
+      amount = row[:amount].to_d
+      member = Member.find_by_uid!(uid)
+      currency = Currency.find(currency_id)
+      account = member.get_account(currency)
+
+      ActiveRecord::Base.transaction do
+        Operations::Asset.credit!(currency: currency, amount: amount, reference_type: 'Distribution')
+        Operations::Liability.credit!(kind: :main, currency: currency, member_id: member.id, amount: amount, reference_type: 'Distribution')
+        account.update!(balance: account.balance + amount)
+      end
+    rescue StandardError => e
+      message = { error: e.message, uid: row[:uid] }
+      ::Rails.logger.error message
+      errors_count += 1
+    end
+    Kernel.puts "Distributions processed #{count}"
+    Kernel.puts "Errored #{errors_count}"
+  end
+end