Search in sources :

Example 1 with HugeBufferConsumer

use of com.questdb.net.ha.comsumer.HugeBufferConsumer in project questdb by bluestreak01.

the class JournalServerAgentTest method setUp.

@Before
public void setUp() throws Exception {
    channel = new MockByteChannel();
    quoteWriter = getFactory().writer(Quote.class);
    tradeWriter = getFactory().writer(Trade.class);
    ServerConfig config = new ServerConfig() {

        {
            setHeartbeatFrequency(100);
            setEnableMultiCast(false);
        }
    };
    server = new JournalServer(config, getFactory());
    server.publish(quoteWriter);
    agent = new JournalServerAgent(server, new InetSocketAddress(NetworkConfig.DEFAULT_DATA_PORT), null);
    hugeBufferConsumer = new HugeBufferConsumer(temp.newFile());
}
Also used : Quote(com.questdb.model.Quote) Trade(com.questdb.model.Trade) ServerConfig(com.questdb.net.ha.config.ServerConfig) InetSocketAddress(java.net.InetSocketAddress) HugeBufferConsumer(com.questdb.net.ha.comsumer.HugeBufferConsumer)

Example 2 with HugeBufferConsumer

use of com.questdb.net.ha.comsumer.HugeBufferConsumer in project questdb by bluestreak01.

the class MetadataReplicationTest method testReplication.

@Test
public void testReplication() {
    try (JournalWriter w = getFactory().writer(Quote.class)) {
        MockByteChannel channel = new MockByteChannel();
        HugeBufferProducer p = new HugeBufferProducer(new File(w.getLocation(), JournalConfiguration.FILE_NAME));
        HugeBufferConsumer c = new HugeBufferConsumer(new File(w.getLocation(), "_remote"));
        p.write(channel);
        c.read(channel);
        try (JournalWriter w2 = getFactory().writer(new JournalMetadata<>(c.getHb(), "xyz"))) {
            Assert.assertTrue(w.getMetadata().isCompatible(w2.getMetadata(), false));
        }
        p.free();
        c.free();
    } catch (Exception e) {
        e.printStackTrace();
    }
}
Also used : JournalWriter(com.questdb.store.JournalWriter) HugeBufferProducer(com.questdb.net.ha.producer.HugeBufferProducer) HugeBufferConsumer(com.questdb.net.ha.comsumer.HugeBufferConsumer) File(java.io.File) AbstractTest(com.questdb.test.tools.AbstractTest) Test(org.junit.Test)

Example 3 with HugeBufferConsumer

use of com.questdb.net.ha.comsumer.HugeBufferConsumer in project questdb by bluestreak01.

the class JournalClient method subscribeOne.

private void subscribeOne(int index, SubscriptionHolder holder, String name, boolean newSubscription) {
    if (newSubscription) {
        SubscriptionHolder sub = new SubscriptionHolder();
        sub.local = holder.local;
        sub.remote = holder.remote;
        sub.listener = holder.listener;
        sub.writer = holder.writer;
        subscriptions.add(sub);
    }
    JournalWriter<?> writer = writers.getQuiet(index);
    try {
        commandProducer.write(channel, Command.ADD_KEY_CMD);
        setKeyRequestProducer.write(channel, new IndexedJournalKey(index, holder.remote));
        checkAck();
        // todo: do we really have to use file here?
        JournalMetadata<?> metadata;
        File file = Files.makeTempFile();
        try {
            try (HugeBufferConsumer h = new HugeBufferConsumer(file)) {
                h.read(channel);
                metadata = new JournalMetadata(h.getHb(), name);
            } catch (JournalException e) {
                throw new JournalNetworkException(e);
            }
        } finally {
            Files.delete(file);
        }
        boolean validate = true;
        if (writer == null) {
            if (holder.writer == null) {
                try {
                    writer = factory.writer(metadata);
                } catch (JournalException e) {
                    LOG.error().$("Failed to create writer: ").$(e).$();
                    unsubscribe(index, null, holder, JournalEvents.EVT_JNL_INCOMPATIBLE);
                    return;
                }
                writersToClose.add(writer);
                validate = false;
            } else {
                writer = holder.writer;
            }
            writer.disableCommitOnClose();
            statusSentList.extendAndSet(index, 0);
            deltaConsumers.extendAndSet(index, new JournalDeltaConsumer(writer));
            writers.extendAndSet(index, writer);
            writer.setJournalListener(holder.listener);
        } else {
            statusSentList.setQuick(index, 0);
        }
        if (validate && !metadata.isCompatible(writer.getMetadata(), false)) {
            LOG.error().$("Journal ").$(holder.local.getName()).$(" is not compatible with ").$(holder.remote.getName()).$("(remote)").$();
            unsubscribe(index, writer, holder, JournalEvents.EVT_JNL_INCOMPATIBLE);
            return;
        }
        commandProducer.write(channel, Command.DELTA_REQUEST_CMD);
        journalClientStateProducer.write(channel, new IndexedJournal(index, writer));
        checkAck();
        statusSentList.setQuick(index, 1);
        if (holder.listener != null) {
            holder.listener.onEvent(JournalEvents.EVT_JNL_SUBSCRIBED);
        }
        LOG.info().$("Subscribed ").$(name).$(" to ").$(holder.remote.getName()).$("(remote)").$();
    } catch (JournalNetworkException e) {
        LOG.error().$("Failed to subscribe ").$(name).$(" to ").$(holder.remote.getName()).$("(remote)").$();
        unsubscribe(index, writer, holder, JournalEvents.EVT_JNL_SERVER_ERROR);
    }
}
Also used : JournalMetadata(com.questdb.store.factory.configuration.JournalMetadata) IncompatibleJournalException(com.questdb.ex.IncompatibleJournalException) JournalException(com.questdb.std.ex.JournalException) JournalDeltaConsumer(com.questdb.net.ha.comsumer.JournalDeltaConsumer) IndexedJournalKey(com.questdb.net.ha.model.IndexedJournalKey) JournalNetworkException(com.questdb.std.ex.JournalNetworkException) HugeBufferConsumer(com.questdb.net.ha.comsumer.HugeBufferConsumer) IndexedJournal(com.questdb.net.ha.model.IndexedJournal) File(java.io.File)

Aggregations

HugeBufferConsumer (com.questdb.net.ha.comsumer.HugeBufferConsumer)3 File (java.io.File)2 IncompatibleJournalException (com.questdb.ex.IncompatibleJournalException)1 Quote (com.questdb.model.Quote)1 Trade (com.questdb.model.Trade)1 JournalDeltaConsumer (com.questdb.net.ha.comsumer.JournalDeltaConsumer)1 ServerConfig (com.questdb.net.ha.config.ServerConfig)1 IndexedJournal (com.questdb.net.ha.model.IndexedJournal)1 IndexedJournalKey (com.questdb.net.ha.model.IndexedJournalKey)1 HugeBufferProducer (com.questdb.net.ha.producer.HugeBufferProducer)1 JournalException (com.questdb.std.ex.JournalException)1 JournalNetworkException (com.questdb.std.ex.JournalNetworkException)1 JournalWriter (com.questdb.store.JournalWriter)1 JournalMetadata (com.questdb.store.factory.configuration.JournalMetadata)1 AbstractTest (com.questdb.test.tools.AbstractTest)1 InetSocketAddress (java.net.InetSocketAddress)1 Test (org.junit.Test)1