@Override public Customer save(Customer customer) { operations.save(customer); return customer; }
@Override public NamedMongoScript register(NamedMongoScript script) { Assert.notNull(script, "Script must not be null!"); mongoOperations.save(script, SCRIPT_COLLECTION_NAME); return script; }
@Override public <S extends T> S save(S entity) { Assert.notNull(entity, "Entity must not be null!"); if (entityInformation.isNew(entity)) { return mongoOperations.insert(entity, entityInformation.getCollectionName()); } return mongoOperations.save(entity, entityInformation.getCollectionName()); }
@Test public void testWriteNoTransactionWithCollection() throws Exception { List<Object> items = new ArrayList<Object>() {{ add(new Object()); add(new Object()); }}; writer.setCollection("collection"); writer.write(items); verify(template).save(items.get(0), "collection"); verify(template).save(items.get(1), "collection"); }
@Test public void testWriteTransactionNoCollection() throws Exception { final List<Object> items = new ArrayList<Object>() {{ add(new Object()); add(new Object()); }}; new TransactionTemplate(transactionManager).execute((TransactionCallback<Void>) status -> { try { writer.write(items); } catch (Exception e) { fail("An exception was thrown while writing: " + e.getMessage()); } return null; }); verify(template).save(items.get(0)); verify(template).save(items.get(1)); }
@Test public void testBasicWrite() throws Exception { MongoItemWriter<String> writer = new MongoItemWriterBuilder<String>().template(this.template).build(); writer.write(this.items); verify(this.template).save(this.items.get(0)); verify(this.template).save(this.items.get(1)); verify(this.template, never()).remove(this.items.get(0)); verify(this.template, never()).remove(this.items.get(1)); }
@Test public void testDelete() throws Exception { MongoItemWriter<String> writer = new MongoItemWriterBuilder<String>().template(this.template) .delete(true) .build(); writer.write(this.items); verify(this.template).remove(this.items.get(0)); verify(this.template).remove(this.items.get(1)); verify(this.template, never()).save(this.items.get(0)); verify(this.template, never()).save(this.items.get(1)); }
@Test public void testWriteToCollection() throws Exception { MongoItemWriter<String> writer = new MongoItemWriterBuilder<String>().collection("collection") .template(this.template) .build(); writer.write(this.items); verify(this.template).save(this.items.get(0), "collection"); verify(this.template).save(this.items.get(1), "collection"); verify(this.template, never()).remove(this.items.get(0), "collection"); verify(this.template, never()).remove(this.items.get(1), "collection"); }
@Test public void testWriteTransactionWithCollection() throws Exception { final List<Object> items = new ArrayList<Object>() {{ add(new Object()); add(new Object()); }}; writer.setCollection("collection"); new TransactionTemplate(transactionManager).execute((TransactionCallback<Void>) status -> { try { writer.write(items); } catch (Exception e) { fail("An exception was thrown while writing: " + e.getMessage()); } return null; }); verify(template).save(items.get(0), "collection"); verify(template).save(items.get(1), "collection"); }
}).when(mongoOperations).save(any(String.class)); writers.add(i, new MongoItemWriter<>()); writers.get(i).setTemplate(mongoOperations);
@Override public NamedMongoScript register(NamedMongoScript script) { Assert.notNull(script, "Script must not be null!"); mongoOperations.save(script, SCRIPT_COLLECTION_NAME); return script; }
@Override protected void handleMessageInternal(Message<?> message) throws Exception { Assert.isTrue(this.initialized, "This class is not yet initialized. Invoke its afterPropertiesSet() method"); String collectionName = this.collectionNameExpression.getValue(this.evaluationContext, message, String.class); Assert.notNull(collectionName, "'collectionNameExpression' must not evaluate to null"); Object payload = message.getPayload(); this.mongoTemplate.save(payload, collectionName); } }
@Override public <S extends T> S save(S entity) { Assert.notNull(entity, "Entity must not be null!"); if (entityInformation.isNew(entity)) { return mongoOperations.insert(entity, entityInformation.getCollectionName()); } return mongoOperations.save(entity, entityInformation.getCollectionName()); }
@Override public void addRequest(String token, ConfirmableActionRequest request) { Preconditions.checkArgument(request != null); Preconditions.checkArgument(StringUtils.isNotBlank(token)); Preconditions.checkArgument(token.equals(request.getToken())); mongoTemplate.save( new ConfirmationRequestRecord(request) ); }
@Override @Transactional(readOnly = false) public RegisteredService save(final RegisteredService svc) { if (svc.getId() == -1) { ((AbstractRegisteredService) svc).setId(svc.hashCode()); } this.mongoTemplate.save(svc, this.collectionName); log.debug("Saved registered service: {}", svc); return this.findServiceById(svc.getId()); }
@Override public void removeGrantedTenant(final String providerId, final String tenantId) { final Provider provider = findAndThrowErrorIfNotExist(new Provider(providerId)); provider.getTenantsAuth().remove(tenantId); provider.getTenantsListVisible().remove(tenantId); getMongoOps().save(provider); }
@Test public void testCount() { mongoOperations.save(new InventoryDocument("Apple Watch", 20.0)); mongoOperations.save(new InventoryDocument("Apple TV", 20.0)); mongoOperations.save(new InventoryDocument("Philips TV", 1.0)); Integer count = inventoryMongoRepository.count(); assertThat(count).isEqualTo(3); }
public Session putData(String identifier, KeyValuePair keyValuePair) { // create a session if none exist. buildSession(identifier); Query q = new Query(Criteria.where("identifier").is(identifier)); Session sessionResult = this.source.findOne(q, Session.class); sessionResult.addKeyValuePair(keyValuePair); this.source.save(sessionResult); return sessionResult; }