Updated to include change streams

Author: Josh Smith

spring-data/src/main/java/com/mongodb/ms0/example/springdata/ChangeStreamConfig.java

@@ -0,0 +1,22 @@
+package com.mongodb.ms0.example.springdata;
+
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.data.mongodb.core.MongoTemplate;
+import org.springframework.data.mongodb.core.messaging.DefaultMessageListenerContainer;
+import org.springframework.data.mongodb.core.messaging.MessageListenerContainer;
+
+@Configuration
+public class ChangeStreamConfig {
+
+
+    @Autowired
+    MongoTemplate template;
+
+
+
+
+
+
+}

spring-data/src/main/java/com/mongodb/ms0/example/springdata/SpringDataApplication.java

@@ -1,13 +1,53 @@
 package com.mongodb.ms0.example.springdata;
 
+import com.mongodb.client.model.changestream.ChangeStreamDocument;
+import com.mongodb.ms0.example.springdata.models.Customer;
+import com.mongodb.ms0.example.springdata.services.ChangeStreamListener;
+import org.bson.Document;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.context.ConfigurableApplicationContext;
+import org.springframework.context.annotation.Bean;
+import org.springframework.data.mongodb.core.MongoTemplate;
+import org.springframework.data.mongodb.core.messaging.*;
+
+import static org.springframework.data.mongodb.core.aggregation.Aggregation.newAggregation;
 
 @SpringBootApplication
 public class SpringDataApplication {
 
+    @Bean
+    MessageListenerContainer messageListenerContainer(MongoTemplate template) {
+        return new DefaultMessageListenerContainer(template);
+    }
+
+
+
+
     public static void main(String[] args) {
-        SpringApplication.run(SpringDataApplication.class, args);
+        ConfigurableApplicationContext context = SpringApplication.run(SpringDataApplication.class, args);
+
+        MongoTemplate template = context.getBean(MongoTemplate.class);
+        MessageListenerContainer container = context.getBean(MessageListenerContainer.class);
+        ChangeStreamListener listener = context.getBean(ChangeStreamListener.class);
+
+        ChangeStreamRequest<Customer> request = ChangeStreamRequest.builder()
+                .collection("customers")
+                .publishTo(listener)
+                .database("providerPref")
+                .filter()
+                .build();
+
+        container.register(request, Customer.class);
+        container.start();
+
+
+
+
+
+
+
     }
 
 }

spring-data/src/main/java/com/mongodb/ms0/example/springdata/controllers/CustomerController.java

@@ -3,6 +3,7 @@
 import com.mongodb.ms0.example.springdata.models.Customer;
 import com.mongodb.ms0.example.springdata.services.CustomerService;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.data.domain.Page;
 import org.springframework.http.MediaType;
 import org.springframework.web.bind.annotation.*;
 
@@ -24,6 +25,11 @@ public Customer getCustomerById(@PathVariable("id") String id) {
         return customer.orElse(null);
     }
 
+    @GetMapping
+    public Page<Customer> getAllCustomers() {
+        return service.getAllCustomers(1);
+    }
+
     @PostMapping
     public Customer createCustomer(@RequestBody Customer customer) {
         return service.createCustomer(customer);
@@ -36,4 +42,10 @@ public List<Customer> customerSearch(@RequestBody Map<String, String> values) {
     }
 
 
+    @PatchMapping(value="{id}")
+    public Customer updateCustomer(@PathVariable("id") String id, @RequestBody Customer customer){
+        return service.updateCustomer(id, customer);
+    }
+
+
 }

spring-data/src/main/java/com/mongodb/ms0/example/springdata/models/CustomerAudit.java

@@ -0,0 +1,46 @@
+package com.mongodb.ms0.example.springdata.models;
+
+import org.bson.types.ObjectId;
+import org.springframework.data.mongodb.core.mapping.Document;
+
+import java.util.Date;
+import java.util.HashMap;
+import java.util.Set;
+
+@Document("customer_history")
+public class CustomerAudit {
+
+    private ObjectId id;
+    private HashMap<String, Object> fields;
+    private Date changedDate;
+
+    public CustomerAudit(){
+        this.changedDate = new Date();
+        this.fields = new HashMap<>();
+    }
+
+
+    public HashMap<String, Object> getFields() {
+        return fields;
+    }
+
+    public void setFields(HashMap<String, Object> fields) {
+        this.fields = fields;
+    }
+
+    public Date getChangedDate() {
+        return changedDate;
+    }
+
+    public void setChangedDate(Date changedDate) {
+        this.changedDate = changedDate;
+    }
+
+    public ObjectId getId() {
+        return id;
+    }
+
+    public void setId(ObjectId id) {
+        this.id = id;
+    }
+}

spring-data/src/main/java/com/mongodb/ms0/example/springdata/repository/CustomerRepository.java

@@ -4,6 +4,8 @@
 import org.springframework.data.mongodb.repository.Aggregation;
 import org.springframework.data.mongodb.repository.MongoRepository;
 import org.springframework.data.mongodb.repository.Query;
+import org.springframework.data.mongodb.repository.Update;
+import org.springframework.data.repository.query.Param;
 import org.springframework.stereotype.Repository;
 
 import java.util.List;
@@ -21,4 +23,8 @@ public interface CustomerRepository extends MongoRepository<Customer, String> {
     @Query("{ 'id' : ?0 }")
     Optional<Customer> findById(String id);
 
+    @Query("{ 'id':  ?2 }")
+    @Update("{'$set': {'firstName': ?0, 'lastName': ?1}}")
+    long updateCustomer(String firstName,  String lastName,  String id);
+
 }

spring-data/src/main/java/com/mongodb/ms0/example/springdata/services/ChangeStreamListener.java

@@ -0,0 +1,40 @@
+package com.mongodb.ms0.example.springdata.services;
+
+import com.mongodb.client.model.changestream.ChangeStreamDocument;
+import com.mongodb.ms0.example.springdata.models.Customer;
+import com.mongodb.ms0.example.springdata.models.CustomerAudit;
+import org.bson.Document;
+import org.springframework.data.mongodb.core.MongoTemplate;
+import org.springframework.data.mongodb.core.messaging.Message;
+import org.springframework.data.mongodb.core.messaging.MessageListener;
+import org.springframework.stereotype.Component;
+
+
+
+@Component
+public class ChangeStreamListener implements MessageListener<ChangeStreamDocument<Document>, Customer> {
+
+
+    private MongoTemplate template;
+
+
+    public ChangeStreamListener(MongoTemplate template){
+        super();
+        this.template = template;
+    }
+
+
+
+    @Override
+    public void onMessage(Message<ChangeStreamDocument<Document>, Customer> message) {
+        System.out.println(message.getBody().getFirstName());
+        CustomerAudit audit = new CustomerAudit();
+        if (message.getRaw().getUpdateDescription() != null) {
+            for (String key : message.getRaw().getUpdateDescription().getUpdatedFields().keySet()) {
+                audit.getFields().put(key, message.getRaw().getUpdateDescription().getUpdatedFields().get(key).asString());
+            }
+            template.insert(audit);
+        }
+
+    }
+}

spring-data/src/main/java/com/mongodb/ms0/example/springdata/services/CustomerService.java

@@ -1,9 +1,19 @@
 package com.mongodb.ms0.example.springdata.services;
 
 import com.mongodb.ms0.example.springdata.models.Customer;
+import com.mongodb.ms0.example.springdata.models.CustomerAudit;
 import com.mongodb.ms0.example.springdata.repository.CustomerRepository;
+import org.bson.types.ObjectId;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.data.domain.Page;
+import org.springframework.data.domain.PageRequest;
+import org.springframework.data.domain.Pageable;
+import org.springframework.data.mongodb.core.MongoTemplate;
+import org.springframework.data.mongodb.core.query.Criteria;
+import org.springframework.data.mongodb.core.query.Query;
+import org.springframework.data.mongodb.core.query.Update;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 
 import java.util.List;
 import java.util.Optional;
@@ -14,6 +24,9 @@ public class CustomerService {
     @Autowired
     private CustomerRepository repository;
 
+    @Autowired
+    private MongoTemplate template;
+
 
     public Optional<Customer> getCustomerById(String id) {
         return repository.findById(id);
@@ -27,4 +40,44 @@ public Customer createCustomer(Customer customer) {
     public List<Customer> customerSearch(String name) {
         return repository.customerSearch(name);
     }
+
+    @Transactional
+    public Customer updateCustomer(String id, Customer customer) {
+
+        Query query = new Query();
+        query.addCriteria(Criteria.where("_id").is(id));
+        Update update = new Update();
+        update.set("firstName", customer.getFirstName());
+        update.set("lastName", customer.getLastName());
+
+        CustomerAudit audit = new CustomerAudit();
+        //audit.setId(new ObjectId("65280bf1e618e05b7f569a28"));
+        audit.getFields().put("firstName", customer.getFirstName());
+        audit.getFields().put("lastName", customer.getLastName());
+
+        template.save(audit);
+
+
+        template.updateFirst(query, update, Customer.class);
+        return template.findById(id, Customer.class);
+
+
+
+        /*
+        long updated =  repository.updateCustomer(customer.getFirstName(), customer.getLastName(), id);
+        if (updated > 0) {
+            return customer;
+        } else {
+            return null;
+        }
+        */
+
+
+
+    }
+
+    public Page<Customer> getAllCustomers(int page){
+        Pageable pageable = PageRequest.of(0,10);
+        return repository.findAll(pageable);
+    }
 }

spring-data/src/main/resources/application.properties

@@ -5,8 +5,9 @@
 #spring.data.mongodb.uri=mongodb+srv://<username>:<password>@cluster0.g8fm0.mongodb.net/mygrocerylist?retryWrites=true&w=majority
 
 #Local MongoDB - no authentication
-spring.data.mongodb.uri=mongodb://localhost:27017/ms0
+#spring.data.mongodb.uri=mongodb://localhost:27017/ms0
+spring.data.mongodb.uri=mongodb+srv://pov_user:[email protected]/providerPref?retryWrites=true&w=majority
 
 
-logging.level.org.springframework.data.mongodb.core.MongoTemplate=DEBUG
-logging.level.org.mongodb.driver=DEBUG
+logging.level.org.springframework.data.mongodb.core=DEBUG
+logging.level.org.mongodb.driver=INFO