Sha256: 8a36145d826cfe3cac3da25d7872effd35e77e6b20cd0efab25b6f2c67891c74
Contents?: true
Size: 1.84 KB
Versions: 5
Compression:
Stored size: 1.84 KB
Contents
# frozen_string_literal: true
require 'rake/tasklib'
require 'avro/builder'
module Avro
module Builder
module Rake
class AvroGenerateTask < ::Rake::TaskLib
attr_accessor :name, :task_namespace, :task_desc, :root,
:load_paths, :dependencies, :filetype
def initialize(name: :generate, dependencies: [])
super()
@name = name
@task_namespace = :avro
@task_desc = 'Generate Avro schema files from Avro::Builder DSL'
@load_paths = []
@root = "#{Rails.root}/avro/dsl" if defined?(Rails)
@dependencies = dependencies
@filetype = 'avsc'
yield self if block_given?
define
end
private
# Define the rake task
def define
namespace task_namespace do
desc task_desc
task(name.to_sym => dependencies) do
raise '"root" must be specified for Avro DSL files' unless root
Avro::Builder.add_load_path(*[root, load_paths].flatten)
Dir["#{root}/**/*.rb"].each do |dsl_file|
puts "Generating Avro schema from #{dsl_file}"
output_file = dsl_file.sub('/dsl/', '/schema/').sub(/\.rb$/, ".#{filetype}")
dsl = Avro::Builder.build_dsl(filename: dsl_file)
if dsl.abstract?
if File.exist?(output_file)
puts "... Removing #{output_file} for abstract type"
FileUtils.rm(output_file)
end
else
schema = dsl.to_json
FileUtils.mkdir_p(File.dirname(output_file))
File.write(output_file, schema.end_with?("\n") ? schema : schema << "\n")
end
end
end
end
end
end
end
end
end
Version data entries
5 entries across 5 versions & 1 rubygems