-
Notifications
You must be signed in to change notification settings - Fork 3.3k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add Avro to JSON data format conversion
- Loading branch information
1 parent
05e65a7
commit 2d12a16
Showing
6 changed files
with
155 additions
and
1 deletion.
There are no files selected for viewing
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -59,7 +59,8 @@ | |
"From Braille", | ||
"Parse TLV", | ||
"CSV to JSON", | ||
"JSON to CSV" | ||
"JSON to CSV", | ||
"Avro to JSON" | ||
] | ||
}, | ||
{ | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,79 @@ | ||
/** | ||
* @author jarrodconnolly [[email protected]] | ||
* @copyright Crown Copyright 2019 | ||
* @license Apache-2.0 | ||
*/ | ||
|
||
import Operation from "../Operation.mjs"; | ||
import OperationError from "../errors/OperationError.mjs"; | ||
import avro from "avsc"; | ||
|
||
/** | ||
* Avro to JSON operation | ||
*/ | ||
class AvroToJSON extends Operation { | ||
|
||
/** | ||
* AvroToJSON constructor | ||
*/ | ||
constructor() { | ||
super(); | ||
|
||
this.name = "Avro to JSON"; | ||
this.module = "Avro"; | ||
this.description = "Converts Avro encoded data into JSON."; | ||
this.infoURL = "https://avro.apache.org/docs/current/spec.html"; | ||
this.inputType = "ArrayBuffer"; | ||
this.outputType = "JSON"; | ||
this.args = [{ | ||
name: "Force Valid JSON", | ||
type: "boolean", | ||
value: true | ||
}]; | ||
} | ||
|
||
/** | ||
* @param {ArrayBuffer} input | ||
* @param {Object[]} args | ||
* @returns {JSON} | ||
*/ | ||
run(input, args) { | ||
const self = this; | ||
if (input.byteLength <= 0) { | ||
throw new OperationError("Please provide an input."); | ||
} | ||
|
||
const forceJSON = args[0]; | ||
|
||
return new Promise((resolve, reject) => { | ||
const result = []; | ||
const inpArray = new Uint8Array(input); | ||
const decoder = new avro.streams.BlockDecoder(); | ||
|
||
decoder | ||
.on("data", function (obj) { | ||
result.push(obj); | ||
}) | ||
.on("error", function () { | ||
reject(new OperationError("Error parsing Avro file.")); | ||
}) | ||
.on("end", function () { | ||
if (forceJSON) { | ||
self.presentType = "JSON"; | ||
self.outputType = "JSON"; | ||
resolve(result.length === 1 ? result[0] : result); | ||
} else { | ||
self.presentType = "string"; | ||
self.outputType = "string"; | ||
const data = result.reduce((result, current) => result + JSON.stringify(current) + "\n", ""); | ||
resolve(data); | ||
} | ||
}); | ||
|
||
decoder.write(inpArray); | ||
decoder.end(); | ||
}); | ||
} | ||
} | ||
|
||
export default AvroToJSON; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,67 @@ | ||
/** | ||
* | ||
* Avro to JSON tests. | ||
* | ||
* @author jarrodconnolly [[email protected]] | ||
* @copyright Crown Copyright 2019 | ||
* @license Apache-2.0 | ||
*/ | ||
|
||
import TestRegister from "../../lib/TestRegister"; | ||
|
||
TestRegister.addTests([ | ||
{ | ||
name: "Avro to JSON: no input (force JSON true)", | ||
input: "", | ||
expectedOutput: "Please provide an input.", | ||
recipeConfig: [ | ||
{ | ||
op: "Avro to JSON", | ||
args: [true] | ||
} | ||
], | ||
}, | ||
{ | ||
name: "Avro to JSON: no input (force JSON false)", | ||
input: "", | ||
expectedOutput: "Please provide an input.", | ||
recipeConfig: [ | ||
{ | ||
op: "Avro to JSON", | ||
args: [false] | ||
} | ||
], | ||
}, | ||
{ | ||
name: "Avro to JSON: small (force JSON true)", | ||
input: "\x4f\x62\x6a\x01\x04\x16\x61\x76\x72\x6f\x2e\x73\x63\x68\x65\x6d\x61\x96\x01\x7b\x22\x74\x79\x70\x65\x22\x3a\x22\x72\x65" + | ||
"\x63\x6f\x72\x64\x22\x2c\x22\x6e\x61\x6d\x65\x22\x3a\x22\x73\x6d\x61\x6c\x6c\x22\x2c\x22\x66\x69\x65\x6c\x64\x73\x22\x3a" + | ||
"\x5b\x7b\x22\x6e\x61\x6d\x65\x22\x3a\x22\x6e\x61\x6d\x65\x22\x2c\x22\x74\x79\x70\x65\x22\x3a\x22\x73\x74\x72\x69\x6e\x67" + | ||
"\x22\x7d\x5d\x7d\x14\x61\x76\x72\x6f\x2e\x63\x6f\x64\x65\x63\x08\x6e\x75\x6c\x6c\x00\x4e\x02\x47\x63\x2e\x37\x02\xe5\xb7" + | ||
"\x5c\xda\xb9\xa6\x2f\x15\x41\x02\x0e\x0c\x6d\x79\x6e\x61\x6d\x65\x4e\x02\x47\x63\x2e\x37\x02\xe5\xb7\x5c\xda\xb9\xa6\x2f" + | ||
"\x15\x41", | ||
expectedOutput: "{\n \"name\": \"myname\"\n}", | ||
recipeConfig: [ | ||
{ | ||
op: "Avro to JSON", | ||
args: [true] | ||
} | ||
], | ||
}, | ||
{ | ||
name: "Avro to JSON: small (force JSON false)", | ||
input: "\x4f\x62\x6a\x01\x04\x16\x61\x76\x72\x6f\x2e\x73\x63\x68\x65\x6d\x61\x96\x01\x7b\x22\x74\x79\x70\x65\x22\x3a\x22\x72\x65" + | ||
"\x63\x6f\x72\x64\x22\x2c\x22\x6e\x61\x6d\x65\x22\x3a\x22\x73\x6d\x61\x6c\x6c\x22\x2c\x22\x66\x69\x65\x6c\x64\x73\x22\x3a" + | ||
"\x5b\x7b\x22\x6e\x61\x6d\x65\x22\x3a\x22\x6e\x61\x6d\x65\x22\x2c\x22\x74\x79\x70\x65\x22\x3a\x22\x73\x74\x72\x69\x6e\x67" + | ||
"\x22\x7d\x5d\x7d\x14\x61\x76\x72\x6f\x2e\x63\x6f\x64\x65\x63\x08\x6e\x75\x6c\x6c\x00\x4e\x02\x47\x63\x2e\x37\x02\xe5\xb7" + | ||
"\x5c\xda\xb9\xa6\x2f\x15\x41\x02\x0e\x0c\x6d\x79\x6e\x61\x6d\x65\x4e\x02\x47\x63\x2e\x37\x02\xe5\xb7\x5c\xda\xb9\xa6\x2f" + | ||
"\x15\x41", | ||
expectedOutput: "{\"name\":\"myname\"}\n", | ||
recipeConfig: [ | ||
{ | ||
op: "Avro to JSON", | ||
args: [false] | ||
} | ||
], | ||
} | ||
]); |