1
0
mirror of https://github.com/LaCasemate/fab-manager.git synced 2024-12-13 23:48:55 +01:00
fab-manager/app/services/statistics_export_service.rb

85 lines
3.2 KiB
Ruby
Raw Normal View History

# frozen_string_literal: true
require 'abstract_controller'
require 'action_controller'
require 'action_view'
require 'active_record'
# require any helpers
2022-12-01 13:03:19 +01:00
require './app/helpers/excel_helper'
2022-12-01 13:03:19 +01:00
# Export statistics (from elasticsearch) to an excel file
class StatisticsExportService
def export_global(export)
# query all stats with range arguments
query = MultiJson.load(export.query)
@results = Elasticsearch::Model.client.search(index: 'stats', scroll: '30s', body: query)
scroll_id = @results['_scroll_id']
while @results['hits']['hits'].size != @results['hits']['total']
scroll_res = Elasticsearch::Model.client.scroll(scroll: '30s', scroll_id: scroll_id)
@results['hits']['hits'].concat(scroll_res['hits']['hits'])
scroll_id = scroll_res['_scroll_id']
end
ids = @results['hits']['hits'].map { |u| u['_source']['userId'] }
@users = User.includes(:profile).where(id: ids)
@indices = StatisticIndex.all.includes(:statistic_fields, statistic_types: [:statistic_sub_types])
ActionController::Base.prepend_view_path './app/views/'
# place data in view_assigns
2022-12-01 13:03:19 +01:00
view_assigns = { results: @results, users: @users, indices: @indices }
av = ActionView::Base.new(ActionController::Base.view_paths, view_assigns)
av.class_eval do
# include any needed helpers (for the view)
2022-12-01 13:03:19 +01:00
include ExcelHelper
end
content = av.render template: 'exports/statistics_global.xlsx.axlsx'
# write content to file
2022-12-01 13:03:19 +01:00
File.binwrite(export.file, content)
end
2022-12-01 13:03:19 +01:00
# rubocop:disable Style/DocumentDynamicEvalDefinition
2023-02-20 15:37:51 +01:00
%w[account event machine project subscription training space order].each do |path|
class_eval %{
def export_#{path}(export)
query = MultiJson.load(export.query)
type_key = export.key
@results = Elasticsearch::Model.client.search({index: 'stats', type: '#{path}', scroll: '30s', body: query})
scroll_id = @results['_scroll_id']
while @results['hits']['hits'].size != @results['hits']['total']
scroll_res = Elasticsearch::Model.client.scroll(scroll: '30s', scroll_id: scroll_id)
@results['hits']['hits'].concat(scroll_res['hits']['hits'])
scroll_id = scroll_res['_scroll_id']
end
ids = @results['hits']['hits'].map { |u| u['_source']['userId'] }
@users = User.includes(:profile).where(:id => ids)
@index = StatisticIndex.find_by(es_type_key: '#{path}')
@type = StatisticType.find_by(key: type_key, statistic_index_id: @index.id)
@subtypes = @type.statistic_sub_types
@fields = @index.statistic_fields
ActionController::Base.prepend_view_path './app/views/'
# place data in view_assigns
view_assigns = {results: @results, users: @users, index: @index, type: @type, subtypes: @subtypes, fields: @fields}
av = ActionView::Base.new(ActionController::Base.view_paths, view_assigns)
av.class_eval do
# include any needed helpers (for the view)
2022-12-01 13:03:19 +01:00
include ExcelHelper
end
content = av.render template: 'exports/statistics_current.xlsx.axlsx'
# write content to file
2022-12-01 13:03:19 +01:00
File.binwrite(export.file, content)
end
}, __FILE__, __LINE__ - 35
end
2022-12-01 13:03:19 +01:00
# rubocop:enable Style/DocumentDynamicEvalDefinition
end