Class: AvroTurf

Inherits:
Object
  • Object
show all
Defined in:
lib/avro_turf.rb,
lib/avro_turf/version.rb,
lib/avro_turf/messaging.rb,
lib/avro_turf/mutable_schema_store.rb,
lib/avro_turf/schema_to_avro_patch.rb

Defined Under Namespace

Modules: AvroGemPatch Classes: CachedConfluentSchemaRegistry, ConfluentSchemaRegistry, DiskCache, Error, InMemoryCache, Messaging, MutableSchemaStore, SchemaError, SchemaNotFoundError, SchemaStore

Constant Summary collapse

DEFAULT_SCHEMAS_PATH =
"./schemas"
VERSION =
"1.4.1"
SchemaRegistry =

AvroTurf::SchemaRegistry is deprecated and will be removed in a future release. Use AvroTurf::ConfluentSchemaRegistry instead.

AvroTurf::ConfluentSchemaRegistry
CachedSchemaRegistry =

AvroTurf::CachedSchemaRegistry is deprecated and will be removed in a future release. Use AvroTurf::CachedConfluentSchemaRegistry instead.

AvroTurf::CachedConfluentSchemaRegistry

Instance Method Summary collapse

Constructor Details

#initialize(schemas_path: nil, schema_store: nil, namespace: nil, codec: nil) ⇒ AvroTurf

Create a new AvroTurf instance with the specified configuration.

schemas_path - The String path to the root directory containing Avro schemas (default: “./schemas”). schema_store - A schema store object that responds to #find(schema_name, namespace). namespace - The String namespace that should be used to qualify schema names (optional). codec - The String name of a codec that should be used to compress messages (optional).

Currently, the only valid codec name is `deflate`.


32
33
34
35
36
37
# File 'lib/avro_turf.rb', line 32

def initialize(schemas_path: nil, schema_store: nil, namespace: nil, codec: nil)
  @namespace = namespace
  @schema_store = schema_store || 
    SchemaStore.new(path: schemas_path || DEFAULT_SCHEMAS_PATH)
  @codec = codec
end

Instance Method Details

#decode(encoded_data, schema_name: nil, namespace: @namespace) ⇒ Object

Decodes Avro data.

encoded_data - A String containing Avro-encoded data. schema_name - The String name of the schema that should be used to read

the data. If nil, the writer schema will be used.

namespace - The namespace of the Avro schema used to decode the data.

Returns whatever is encoded in the data.


88
89
90
91
# File 'lib/avro_turf.rb', line 88

def decode(encoded_data, schema_name: nil, namespace: @namespace)
  stream = StringIO.new(encoded_data)
  decode_stream(stream, schema_name: schema_name, namespace: namespace)
end

#decode_stream(stream, schema_name: nil, namespace: @namespace) ⇒ Object

Decodes Avro data from an IO stream.

stream - An IO object containing Avro data. schema_name - The String name of the schema that should be used to read

the data. If nil, the writer schema will be used.

namespace - The namespace of the Avro schema used to decode the data.

Returns whatever is encoded in the stream.


101
102
103
104
105
106
# File 'lib/avro_turf.rb', line 101

def decode_stream(stream, schema_name: nil, namespace: @namespace)
  schema = schema_name && @schema_store.find(schema_name, namespace)
  reader = Avro::IO::DatumReader.new(nil, schema)
  dr = Avro::DataFile::Reader.new(stream, reader)
  dr.first
end

#encode(data, schema_name: nil, namespace: @namespace, validate: false) ⇒ Object

Encodes data to Avro using the specified schema.

data - The data that should be encoded. schema_name - The name of a schema in the `schemas_path`. validate - The boolean for performing complete data validation before

encoding it, Avro::SchemaValidator::ValidationError with
a descriptive message will be raised in case of invalid message.

Returns a String containing the encoded data.


48
49
50
51
52
53
54
# File 'lib/avro_turf.rb', line 48

def encode(data, schema_name: nil, namespace: @namespace, validate: false)
  stream = StringIO.new

  encode_to_stream(data, stream: stream, schema_name: schema_name, namespace: namespace, validate: validate)

  stream.string
end

#encode_to_stream(data, schema_name: nil, stream: nil, namespace: @namespace, validate: false) ⇒ Object

Encodes data to Avro using the specified schema and writes it to the specified stream.

data - The data that should be encoded. schema_name - The name of a schema in the `schemas_path`. stream - An IO object that the encoded data should be written to (optional). validate - The boolean for performing complete data validation before

encoding it, Avro::SchemaValidator::ValidationError with
a descriptive message will be raised in case of invalid message.

Returns nothing.


67
68
69
70
71
72
73
74
75
76
77
78
# File 'lib/avro_turf.rb', line 67

def encode_to_stream(data, schema_name: nil, stream: nil, namespace: @namespace, validate: false)
  schema = @schema_store.find(schema_name, namespace)
  writer = Avro::IO::DatumWriter.new(schema)

  if validate
    Avro::SchemaValidator.validate!(schema, data, recursive: true, encoded: false, fail_on_extra_fields: true)
  end

  dw = Avro::DataFile::Writer.new(stream, writer, schema, @codec)
  dw << data.as_avro
  dw.close
end

#load_schemas!Object

Loads all schema definition files in the `schemas_dir`.


123
124
125
# File 'lib/avro_turf.rb', line 123

def load_schemas!
  @schema_store.load_schemas!
end

#valid?(data, schema_name: nil, namespace: @namespace) ⇒ Boolean

Validates data against an Avro schema.

data - The data that should be validated. schema - The String name of the schema that should be used to validate

the data.

namespace - The namespace of the Avro schema (optional).

Returns true if the data is valid, false otherwise.

Returns:

  • (Boolean)

116
117
118
119
120
# File 'lib/avro_turf.rb', line 116

def valid?(data, schema_name: nil, namespace: @namespace)
  schema = schema_name && @schema_store.find(schema_name, namespace)

  Avro::Schema.validate(schema, data.as_avro)
end