/** * @param topVer Topology version. * @return Future. */ public GridFutureAdapter addDataStreamerFuture(AffinityTopologyVersion topVer) { final DataStreamerFuture fut = new DataStreamerFuture(topVer); boolean add = dataStreamerFuts.add(fut); assert add; return fut; }
@Override public boolean apply(UUID uuid, CacheEvent evt) { evts.add(evt); latch.countDown(); return true; } };
/** {@inheritDoc} */ @Override public Lock obtainLock(Directory dir, String lockName) throws IOException { if (locks.add(lockName)) return new LockImpl(lockName); else throw new LockObtainFailedException("lock instance already obtained: (dir=" + dir + ", lockName=" + lockName + ")"); }
@Override public boolean apply(Event evt) { CacheEvent evt0 = (CacheEvent)evt; if (evt0.closureClassName() != null) evts.add(evt0); return true; } }, EVT_CACHE_OBJECT_READ);
/** {@inheritDoc} */ @Override public void onUpdated(Iterable<CacheEntryEvent<?, ?>> evts) throws CacheEntryListenerException { for (CacheEntryEvent<?, ?> e : evts) { Integer key = (Integer)e.getKey(); keys.add(key); assert this.evts.put(key, e) == null; } }
/** {@inheritDoc} */ @Override public int write(ByteBuffer srcBuf, long position) throws IOException { ChannelOpFuture fut = holder.get(); fut.reset(); asyncFuts.add(fut); ch.write(srcBuf, position, null, fut); try { return fut.getUninterruptibly(); } catch (IgniteCheckedException e) { throw new IOException(e); } finally { asyncFuts.remove(fut); } }
/** {@inheritDoc} */ @SuppressWarnings({"SimplifiableIfStatement"}) @Override public boolean add(E e) { A.notNull(e, "e"); removeStale(); if (!contains(e)) return store.add(fact.apply(e)); return false; }
/** {@inheritDoc} */ @SuppressWarnings({"SimplifiableIfStatement"}) @Override public boolean add(E e) { A.notNull(e, "e"); removeStale(); if (!contains(e)) return store.add(fact.apply(e)); return false; }
/** {@inheritDoc} */ @Override public void onMessage(UUID nodeId, Message msg, IgniteRunnable msgC) { msgC.run(); assertTrue(msg instanceof GridTestMessage); cntr.incrementAndGet(); GridTestMessage msg0 = (GridTestMessage)msg; assertEquals(nodeId, msg0.getSourceNodeId()); assertTrue(msgIds.add(msg0.getMsgId())); latch.countDown(); }
@Override public boolean apply(Event evt) { CacheEvent evt0 = (CacheEvent)evt; if (evt0.closureClassName() != null) { System.out.println("ADDED: [nodeId=" + evt0.node() + ", evt=" + evt0 + ']'); evts.add(evt0); } return true; } }, EVT_CACHE_OBJECT_READ);
@Override public Void call() throws Exception { while (!stopped.get()) { Ignite node = grid(ThreadLocalRandom.current().nextInt(1, 5)); IgniteInternalFuture fut; int fieldNum = ThreadLocalRandom.current().nextInt(0, dynColCnt.get() + 1); boolean removed = fields.remove(fieldNum); if (removed) fut = dropCols(node, QueryUtils.DFLT_SCHEMA, "newCol" + fieldNum); else { fieldNum = dynColCnt.getAndIncrement(); fut = addCols(node, QueryUtils.DFLT_SCHEMA, c("newCol" + fieldNum, Integer.class.getName())); } try { fut.get(); if (!removed) fields.add(fieldNum); } catch (SchemaOperationException e) { // No-op. } catch (Exception e) { fail("Unexpected exception: " + e); } } return null; } }, 1);
@Override public Void call() throws Exception { while (!stopped.get()) { Ignite node = grid(ThreadLocalRandom.current().nextInt(1, 5)); IgniteInternalFuture fut; int fieldNum = ThreadLocalRandom.current().nextInt(0, dynColCnt.get() + 1); boolean removed = fields.remove(fieldNum); if (removed) fut = dropCols(node, QueryUtils.DFLT_SCHEMA, "newCol" + fieldNum); else { fieldNum = dynColCnt.getAndIncrement(); fut = addCols(node, QueryUtils.DFLT_SCHEMA, c("newCol" + fieldNum, Integer.class.getName())); } try { fut.get(); if (!removed) fields.add(fieldNum); } catch (SchemaOperationException e) { // No-op. } catch (Exception e) { fail("Unexpected exception: " + e); } } return null; } }, 1);
/** {@inheritDoc} */ @Override public void onUpdated(Iterable<CacheEntryEvent<?, ?>> evts) { try { for (CacheEntryEvent<?, ?> evt : evts) { CountDownLatch latch = this.latch; log.info("Received cache event [evt=" + evt + ", left=" + (latch != null ? latch.getCount() : null) + ']'); this.evts.put(evt.getKey(), evt); keys.add((Integer)evt.getKey()); if (allEvts != null) allEvts.add(evt); assertTrue(latch != null); assertTrue(latch.getCount() > 0); latch.countDown(); if (latch.getCount() == 0) { this.latch = null; keys.clear(); } } } catch (Throwable e) { err = true; log.error("Unexpected error", e); } } }
/** {@inheritDoc} */ @Override protected void writeToSocket(Socket sock, OutputStream out, TcpDiscoveryAbstractMessage msg, long timeout) throws IOException, IgniteCheckedException { boolean add = msgIds.add(msg.id()); if (checkDuplicates && !add && !(msg instanceof TcpDiscoveryMetricsUpdateMessage)) { log.error("Send duplicated message: " + msg); failed = true; } super.writeToSocket(sock, out, msg, timeout); } }
@Override public Void call() throws Exception { ThreadLocalRandom rnd = ThreadLocalRandom.current(); if (rnd.nextBoolean()) { Integer stopIdx; for (;;) { stopIdx = rnd.nextInt(initNodes - 1) + 1; if (stopNodes.add(stopIdx)) break; } log.info("Stop node: " + stopIdx); stopGrid(getTestIgniteInstanceName(stopIdx), true, false); } else { int nodeIdx = idx.incrementAndGet(); if (rnd.nextInt(5) == 0) { log.info("Start client: " + nodeIdx); client.set(true); } else log.info("Start server: " + nodeIdx); startGrid(nodeIdx); } return null; } }, threads, "test-thread");
/** {@inheritDoc} */ @Override protected BinaryObjectExImpl toBinary(BinaryMarshaller marsh, Object obj) throws Exception { byte[] arr = marsh.marshal(obj); long ptr = GridUnsafe.allocateMemory(arr.length); ptrs.add(ptr); GridUnsafe.copyHeapOffheap(arr, GridUnsafe.BYTE_ARR_OFF, ptr, arr.length); return new BinaryObjectOffheapImpl(ctx, ptr, 0, arr.length); } }
threadIds.add(Thread.currentThread().getId());
/** {@inheritDoc} */ @Override protected BinaryObjectExImpl toBinary(BinaryMarshaller marsh, Object obj) throws Exception { byte[] arr = marsh.marshal(obj); long ptr = GridUnsafe.allocateMemory(arr.length); ptrs.add(ptr); GridUnsafe.copyHeapOffheap(arr, GridUnsafe.BYTE_ARR_OFF, ptr, arr.length); return new BinaryObjectOffheapImpl(binaryContext(marsh), ptr, 0, arr.length); } }
/** {@inheritDoc} */ @Override public boolean apply(Event evt) { DiscoveryEvent evt0 = (DiscoveryEvent)evt; if (evt.type() == EVT_NODE_FAILED || evt.type() == EVT_NODE_LEFT) { if (!leftIds.add(evt0.eventNode().id())) failed("Duplicated failed node ID: " + evt0.eventNode().id()); } else { assertEquals(EVT_NODE_JOINED, evt.type()); if (!joinIds.add(evt0.eventNode().id())) failed("Duplicated joined node ID: " + evt0.eventNode().id()); } return true; }
/** {@inheritDoc} */ @SuppressWarnings({"SimplifiableIfStatement"}) @Override public boolean add(E e) { A.notNull(e, "e"); removeStale(); if (!contains(e)) return store.add(fact.apply(e)); return false; }