use of org.ballerinalang.nativeimpl.io.channels.base.CharacterChannel in project ballerina by ballerina-lang.
the class AsyncReadWriteTest method writeRecords.
@Test(description = "Test which will write records to a channel using async io framework")
public void writeRecords() throws IOException, ExecutionException, InterruptedException {
// Number of characters in this file would be 6
ByteChannel byteChannel = TestUtil.openForWriting(currentDirectoryPath + "records.csv");
Channel channel = new MockByteChannel(byteChannel);
CharacterChannel characterChannel = new CharacterChannel(channel, StandardCharsets.UTF_8.name());
DelimitedRecordChannel recordChannel = new DelimitedRecordChannel(characterChannel, "\n", ",");
String[] recordOne = { "Foo", "Bar", "911" };
BStringArray recordOneArr = new BStringArray(recordOne);
DelimitedRecordWriteEvent recordWriteEvent = new DelimitedRecordWriteEvent(recordChannel, recordOneArr);
Future<EventResult> future = eventManager.publish(recordWriteEvent);
future.get();
String[] recordTwo = { "Jim", "Com", "119" };
BStringArray recordTwoArr = new BStringArray(recordTwo);
recordWriteEvent = new DelimitedRecordWriteEvent(recordChannel, recordTwoArr);
future = eventManager.publish(recordWriteEvent);
future.get();
recordChannel.close();
}
use of org.ballerinalang.nativeimpl.io.channels.base.CharacterChannel in project ballerina by ballerina-lang.
the class RecordInputOutputTest method processRecordSequence.
@Test(description = "Processors records in sequence with hasNext()")
public void processRecordSequence() throws IOException, URISyntaxException {
int expectedFieldCount = 3;
boolean hasNext = false;
// Number of characters in this file would be 6
ByteChannel byteChannel = TestUtil.openForReading("datafiles/io/records/sample.csv");
Channel channel = new MockByteChannel(byteChannel);
CharacterChannel characterChannel = new CharacterChannel(channel, StandardCharsets.UTF_8.name());
DelimitedRecordChannel recordChannel = new DelimitedRecordChannel(characterChannel, "\n", ",");
hasNext = recordChannel.hasNext();
String[] readRecord = recordChannel.read();
Assert.assertEquals(readRecord.length, expectedFieldCount);
Assert.assertTrue(hasNext);
hasNext = recordChannel.hasNext();
readRecord = recordChannel.read();
Assert.assertEquals(readRecord.length, expectedFieldCount);
Assert.assertTrue(hasNext);
hasNext = recordChannel.hasNext();
readRecord = recordChannel.read();
Assert.assertEquals(readRecord.length, expectedFieldCount);
Assert.assertTrue(hasNext);
hasNext = recordChannel.hasNext();
readRecord = recordChannel.read();
Assert.assertEquals(readRecord.length, 0);
Assert.assertFalse(hasNext);
recordChannel.close();
}
use of org.ballerinalang.nativeimpl.io.channels.base.CharacterChannel in project ballerina by ballerina-lang.
the class RecordInputOutputTest method readLongRecord.
@Test(description = "Read lengthy records")
public void readLongRecord() throws IOException, URISyntaxException {
int expectedFieldCount = 18;
// Number of characters in this file would be 6
ByteChannel byteChannel = TestUtil.openForReading("datafiles/io/records/sample4.csv");
Channel channel = new MockByteChannel(byteChannel);
CharacterChannel characterChannel = new CharacterChannel(channel, StandardCharsets.UTF_8.name());
DelimitedRecordChannel recordChannel = new DelimitedRecordChannel(characterChannel, "\n", ",");
String[] readRecord = recordChannel.read();
Assert.assertEquals(readRecord.length, expectedFieldCount);
readRecord = recordChannel.read();
Assert.assertEquals(readRecord.length, expectedFieldCount);
}
use of org.ballerinalang.nativeimpl.io.channels.base.CharacterChannel in project ballerina by ballerina-lang.
the class RecordInputOutputTest method readRecords.
@Test(description = "Reads records from file")
public void readRecords() throws IOException, URISyntaxException {
int expectedFieldCount = 3;
// Number of characters in this file would be 6
ByteChannel byteChannel = TestUtil.openForReading("datafiles/io/records/sample.csv");
Channel channel = new MockByteChannel(byteChannel);
CharacterChannel characterChannel = new CharacterChannel(channel, StandardCharsets.UTF_8.name());
DelimitedRecordChannel recordChannel = new DelimitedRecordChannel(characterChannel, "\n", ",");
String[] readRecord = recordChannel.read();
Assert.assertEquals(readRecord.length, expectedFieldCount);
readRecord = recordChannel.read();
Assert.assertEquals(readRecord.length, expectedFieldCount);
readRecord = recordChannel.read();
Assert.assertEquals(readRecord.length, expectedFieldCount);
readRecord = recordChannel.read();
Assert.assertEquals(readRecord.length, 0);
recordChannel.close();
}
use of org.ballerinalang.nativeimpl.io.channels.base.CharacterChannel in project ballerina by ballerina-lang.
the class CloseCharacterChannel method execute.
/**
* <p>
* Closes a character channel.
* </p>
* <p>
* {@inheritDoc}
*/
@Override
public void execute(Context context, CallableUnitCallback callback) {
BStruct channel = (BStruct) context.getRefArgument(CHARACTER_CHANNEL_INDEX);
CharacterChannel charChannel = (CharacterChannel) channel.getNativeData(IOConstants.CHARACTER_CHANNEL_NAME);
EventContext eventContext = new EventContext(context, callback);
IOUtils.close(charChannel, eventContext, CloseCharacterChannel::closeResponse);
}
Aggregations