Examples with RecordCursor com.questdb.common.RecordCursor used on opensource projects

Example 11 with RecordCursor

use of com.questdb.common.RecordCursor in project questdb by bluestreak01.

the class AbstractOptimiserTest method assertString.

protected void assertString(String query, int columnIndex) throws ParserException {
    try (RecordSource rs = compiler.compile(FACTORY_CONTAINER.getFactory(), query)) {
        RecordCursor cursor = rs.prepareCursor(FACTORY_CONTAINER.getFactory());
        try {
            while (cursor.hasNext()) {
                Record r = cursor.next();
                int len = r.getStrLen(columnIndex);
                CharSequence s = r.getFlyweightStr(columnIndex);
                if (s != null) {
                    CharSequence csB = r.getFlyweightStrB(columnIndex);
                    TestUtils.assertEquals(s, csB);
                    Assert.assertEquals(len, s.length());
                    Assert.assertFalse(s == csB);
                } else {
                    Assert.assertEquals(-1, len);
                    Assert.assertNull(r.getFlyweightStr(columnIndex));
                    Assert.assertNull(r.getFlyweightStrB(columnIndex));
                }
            }
        } finally {
            cursor.releaseCursor();
        }
    }
}

Example 12 with RecordCursor

use of com.questdb.common.RecordCursor in project questdb by bluestreak01.

the class AbstractOptimiserTest method assertSymbol.

public static void assertSymbol(String query, int columnIndex) throws ParserException {
    try (RecordSource src = compiler.compile(FACTORY_CONTAINER.getFactory(), query)) {
        RecordCursor cursor = src.prepareCursor(FACTORY_CONTAINER.getFactory());
        try {
            SymbolTable tab = cursor.getStorageFacade().getSymbolTable(columnIndex);
            while (cursor.hasNext()) {
                Record r = cursor.next();
                TestUtils.assertEquals(r.getSym(columnIndex), tab.value(r.getInt(columnIndex)));
            }
        } finally {
            cursor.releaseCursor();
        }
    }
}

Example 13 with RecordCursor

use of com.questdb.common.RecordCursor in project questdb by bluestreak01.

the class AsOfPartitionedJoinRecordSourceTest method testStrings.

@Test
public void testStrings() throws Exception {
    try (AsOfPartitionedJoinRecordSource source = new AsOfPartitionedJoinRecordSource(compileSource("y"), 0, new NoRowIdRecordSource().of(compileSource("x")), 0, keys, keys, 512, 512, 512, cc)) {
        StringSink testSink = new StringSink();
        int idx = source.getMetadata().getColumnIndex("trader");
        RecordCursor cursor = source.prepareCursor(FACTORY_CONTAINER.getFactory());
        try {
            for (Record r : cursor) {
                testSink.clear();
                r.getStr(idx, testSink);
                if (r.getFlyweightStr(idx) == null) {
                    Assert.assertTrue(testSink.length() == 0);
                } else {
                    TestUtils.assertEquals(r.getFlyweightStr(idx), testSink);
                }
                TestUtils.assertEquals(r.getFlyweightStr(idx), r.getFlyweightStr(idx));
            }
        } finally {
            cursor.releaseCursor();
        }
    }
}

Example 14 with RecordCursor

use of com.questdb.common.RecordCursor in project questdb by bluestreak01.

the class DDLTests method testCast.

public void testCast(int from, int to) throws Exception {
    int n = 100;
    try (JournalWriter w1 = compiler.createWriter(getFactory(), "create table y (a " + ColumnType.nameOf(from) + ") record hint 100")) {
        Rnd rnd = new Rnd();
        for (int i = 0; i < n; i++) {
            JournalEntryWriter ew = w1.entryWriter();
            switch(from) {
                case ColumnType.INT:
                    ew.putInt(0, rnd.nextInt());
                    break;
                case ColumnType.LONG:
                    ew.putLong(0, rnd.nextLong());
                    break;
                case ColumnType.DATE:
                    ew.putDate(0, rnd.nextLong());
                    break;
                case ColumnType.BYTE:
                    ew.put(0, rnd.nextByte());
                    break;
                case ColumnType.SHORT:
                    ew.putShort(0, rnd.nextShort());
                    break;
                case ColumnType.FLOAT:
                    ew.putFloat(0, rnd.nextFloat());
                    break;
                case ColumnType.DOUBLE:
                    ew.putDouble(0, rnd.nextDouble());
                    break;
                case ColumnType.SYMBOL:
                    ew.putSym(0, rnd.nextChars(10));
                    break;
                case ColumnType.STRING:
                    ew.putStr(0, rnd.nextChars(10));
                    break;
                default:
                    break;
            }
            ew.append();
        }
        w1.commit();
    }
    exec("create table x as (y), cast(a as " + ColumnType.nameOf(to) + ") record hint 100");
    try (RecordSource rs = compiler.compile(getFactory(), "x")) {
        Rnd rnd = new Rnd();
        Assert.assertEquals(to, rs.getMetadata().getColumnQuick(0).getType());
        RecordCursor cursor = rs.prepareCursor(getFactory());
        try {
            while (cursor.hasNext()) {
                switch(from) {
                    case ColumnType.INT:
                        switch(to) {
                            case ColumnType.SHORT:
                                Assert.assertEquals((short) rnd.nextInt(), cursor.next().getShort(0));
                                break;
                            case ColumnType.LONG:
                                Assert.assertEquals((long) rnd.nextInt(), cursor.next().getLong(0));
                                break;
                            case ColumnType.BYTE:
                                Assert.assertEquals((byte) rnd.nextInt(), cursor.next().getByte(0));
                                break;
                            case ColumnType.FLOAT:
                                Assert.assertEquals((float) rnd.nextInt(), cursor.next().getFloat(0), 0.000000001f);
                                break;
                            case ColumnType.DOUBLE:
                                Assert.assertEquals((double) rnd.nextInt(), cursor.next().getDouble(0), 0.000000001);
                                break;
                            case ColumnType.DATE:
                                Assert.assertEquals((long) rnd.nextInt(), cursor.next().getDate(0));
                                break;
                            case ColumnType.INT:
                                Assert.assertEquals(rnd.nextInt(), cursor.next().getInt(0));
                                break;
                            default:
                                break;
                        }
                        break;
                    case ColumnType.LONG:
                    case ColumnType.DATE:
                        switch(to) {
                            case ColumnType.SHORT:
                                Assert.assertEquals((short) rnd.nextLong(), cursor.next().getShort(0));
                                break;
                            case ColumnType.LONG:
                                Assert.assertEquals(rnd.nextLong(), cursor.next().getLong(0));
                                break;
                            case ColumnType.BYTE:
                                Assert.assertEquals((byte) rnd.nextLong(), cursor.next().getByte(0));
                                break;
                            case ColumnType.FLOAT:
                                Assert.assertEquals((float) rnd.nextLong(), cursor.next().getFloat(0), 0.000000001f);
                                break;
                            case ColumnType.DOUBLE:
                                Assert.assertEquals((double) rnd.nextLong(), cursor.next().getDouble(0), 0.000000001);
                                break;
                            case ColumnType.DATE:
                                Assert.assertEquals(rnd.nextLong(), cursor.next().getDate(0));
                                break;
                            case ColumnType.INT:
                                Assert.assertEquals((int) rnd.nextLong(), cursor.next().getInt(0));
                                break;
                            default:
                                break;
                        }
                        break;
                    case ColumnType.BYTE:
                        switch(to) {
                            case ColumnType.SHORT:
                                Assert.assertEquals((short) rnd.nextByte(), cursor.next().getShort(0));
                                break;
                            case ColumnType.LONG:
                                Assert.assertEquals((long) rnd.nextByte(), cursor.next().getLong(0));
                                break;
                            case ColumnType.BYTE:
                                Assert.assertEquals(rnd.nextByte(), cursor.next().getByte(0));
                                break;
                            case ColumnType.FLOAT:
                                Assert.assertEquals((float) rnd.nextByte(), cursor.next().getFloat(0), 0.000000001f);
                                break;
                            case ColumnType.DOUBLE:
                                Assert.assertEquals((double) rnd.nextByte(), cursor.next().getDouble(0), 0.000000001);
                                break;
                            case ColumnType.DATE:
                                Assert.assertEquals((long) rnd.nextByte(), cursor.next().getDate(0));
                                break;
                            case ColumnType.INT:
                                Assert.assertEquals((int) rnd.nextByte(), cursor.next().getInt(0));
                                break;
                            default:
                                break;
                        }
                        break;
                    case ColumnType.SHORT:
                        switch(to) {
                            case ColumnType.SHORT:
                                Assert.assertEquals(rnd.nextShort(), cursor.next().getShort(0));
                                break;
                            case ColumnType.LONG:
                                Assert.assertEquals((long) rnd.nextShort(), cursor.next().getLong(0));
                                break;
                            case ColumnType.BYTE:
                                Assert.assertEquals((byte) rnd.nextShort(), cursor.next().getByte(0));
                                break;
                            case ColumnType.FLOAT:
                                Assert.assertEquals((float) rnd.nextShort(), cursor.next().getFloat(0), 0.000000001f);
                                break;
                            case ColumnType.DOUBLE:
                                Assert.assertEquals((double) rnd.nextShort(), cursor.next().getDouble(0), 0.000000001);
                                break;
                            case ColumnType.DATE:
                                Assert.assertEquals((long) rnd.nextShort(), cursor.next().getDate(0));
                                break;
                            case ColumnType.INT:
                                Assert.assertEquals((int) rnd.nextShort(), cursor.next().getInt(0));
                                break;
                            default:
                                break;
                        }
                        break;
                    case ColumnType.FLOAT:
                        switch(to) {
                            case ColumnType.SHORT:
                                Assert.assertEquals((short) rnd.nextFloat(), cursor.next().getShort(0));
                                break;
                            case ColumnType.LONG:
                                Assert.assertEquals((long) rnd.nextFloat(), cursor.next().getLong(0));
                                break;
                            case ColumnType.BYTE:
                                Assert.assertEquals((byte) rnd.nextFloat(), cursor.next().getByte(0));
                                break;
                            case ColumnType.FLOAT:
                                Assert.assertEquals(rnd.nextFloat(), cursor.next().getFloat(0), 0.000000001f);
                                break;
                            case ColumnType.DOUBLE:
                                Assert.assertEquals((double) rnd.nextFloat(), cursor.next().getDouble(0), 0.000000001);
                                break;
                            case ColumnType.DATE:
                                Assert.assertEquals((long) rnd.nextFloat(), cursor.next().getDate(0));
                                break;
                            case ColumnType.INT:
                                Assert.assertEquals((int) rnd.nextFloat(), cursor.next().getInt(0));
                                break;
                            default:
                                break;
                        }
                        break;
                    case ColumnType.DOUBLE:
                        switch(to) {
                            case ColumnType.SHORT:
                                Assert.assertEquals((short) rnd.nextDouble(), cursor.next().getShort(0));
                                break;
                            case ColumnType.LONG:
                                Assert.assertEquals((long) rnd.nextDouble(), cursor.next().getLong(0));
                                break;
                            case ColumnType.BYTE:
                                Assert.assertEquals((byte) rnd.nextDouble(), cursor.next().getByte(0));
                                break;
                            case ColumnType.FLOAT:
                                Assert.assertEquals((float) rnd.nextDouble(), cursor.next().getFloat(0), 0.000000001f);
                                break;
                            case ColumnType.DOUBLE:
                                Assert.assertEquals(rnd.nextDouble(), cursor.next().getDouble(0), 0.000000001);
                                break;
                            case ColumnType.DATE:
                                Assert.assertEquals((long) rnd.nextDouble(), cursor.next().getDate(0));
                                break;
                            case ColumnType.INT:
                                Assert.assertEquals((int) rnd.nextDouble(), cursor.next().getInt(0));
                                break;
                            default:
                                break;
                        }
                        break;
                    case ColumnType.STRING:
                        switch(to) {
                            case ColumnType.SYMBOL:
                                TestUtils.assertEquals(rnd.nextChars(10), cursor.next().getSym(0));
                                break;
                            default:
                                TestUtils.assertEquals(rnd.nextChars(10), cursor.next().getFlyweightStr(0));
                                break;
                        }
                        break;
                    case ColumnType.SYMBOL:
                        switch(to) {
                            case ColumnType.STRING:
                                TestUtils.assertEquals(rnd.nextChars(10), cursor.next().getFlyweightStr(0));
                                break;
                            default:
                                TestUtils.assertEquals(rnd.nextChars(10), cursor.next().getSym(0));
                                break;
                        }
                        break;
                    default:
                        break;
                }
            }
        } finally {
            cursor.releaseCursor();
        }
    }
}

Example 15 with RecordCursor

use of com.questdb.common.RecordCursor in project questdb by bluestreak01.

the class MergingRowSourceTest method testHeapMerge.

@Test
public void testHeapMerge() throws JournalException, NumericException {
    try (JournalWriter<Quote> w = getFactory().writer(Quote.class)) {
        TestUtils.generateQuoteData(w, 100000, DateFormatUtils.parseDateTime("2014-02-11T00:00:00.000Z"), 10);
        RowSource srcA = new KvIndexSymLookupRowSource("sym", "BP.L", true);
        RowSource srcB = new KvIndexSymLookupRowSource("sym", "WTB.L", true);
        RecordSource rs = new JournalRecordSource(new JournalPartitionSource(w.getMetadata(), true), new HeapMergingRowSource(srcA, srcB));
        long last = 0;
        RecordCursor c = rs.prepareCursor(getFactory());
        try {
            int ts = rs.getMetadata().getColumnIndex("timestamp");
            while (c.hasNext()) {
                long r = c.next().getDate(ts);
                Assert.assertTrue(r > last);
                last = r;
            }
        } finally {
            c.releaseCursor();
        }
    }
}