Added trading logic

Author: NLxAROSA

src/main/java/com/redislabs/demos/redisbank/Utilities.java

@@ -48,4 +48,28 @@ private static final void appendDigits(StringBuilder builder, SecureRandom rando
 
     }
 
+    public static final String generatePortfolioIdFrom(String userName)   {
+        if (StringUtils.isEmpty(userName)) {
+            throw new IllegalArgumentException("Parameter userName should have a value");
+        }
+
+        try {
+            // Set the seed of the pseudo random generator to the username,
+            // so we always get the same sequence for the same username
+            SecureRandom random = SecureRandom.getInstance("SHA1PRNG");
+            random.setSeed(userName.getBytes("UTF-8"));
+
+            StringBuilder builder = new StringBuilder();
+            builder.append("NL");
+            appendDigits(builder, random, 2);
+            builder.append("RBINV");
+            appendDigits(builder, random, 5);
+
+            return builder.toString();
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+        
+    }
+
 }

src/main/java/com/redislabs/demos/redisbank/investments/StockPortfolio.java

@@ -18,5 +18,7 @@ public class StockPortfolio {
     @Id
     private Long id;
     private HashMap<Stock, Double> stocks = new HashMap<>();
+    private String portfolioId;
+    private String portfolioHolderName;
 
 }

src/main/java/com/redislabs/demos/redisbank/investments/StockPortfolioGenerator.java

@@ -0,0 +1,67 @@
+package com.redislabs.demos.redisbank.investments;
+
+import java.io.UnsupportedEncodingException;
+import java.security.NoSuchAlgorithmException;
+import java.security.SecureRandom;
+import java.text.DateFormat;
+import java.text.NumberFormat;
+import java.text.SimpleDateFormat;
+import java.util.ArrayList;
+import java.util.Date;
+import java.util.List;
+
+import com.redislabs.demos.redisbank.SerializationUtil;
+import com.redislabs.demos.redisbank.Utilities;
+
+public class StockPortfolioGenerator {
+
+    private List<Stock> stocks = new ArrayList<>();
+    private final StockPortfolio spf;
+    private final SecureRandom random;
+    private final DateFormat df = new SimpleDateFormat("yyyy.MM.dd 'at' HH:mm:ss");
+    private final NumberFormat nf = NumberFormat.getCurrencyInstance();
+    private final TradeRepository tr;
+    private final StockPortfolioRepository spr;
+
+    public StockPortfolioGenerator(TradeRepository tr, StockPortfolioRepository spr)
+            throws NoSuchAlgorithmException, UnsupportedEncodingException {
+        this.random = SecureRandom.getInstance("SHA1PRNG");
+        this.random.setSeed("lars".getBytes("UTF-8")); // Prime the RNG so it always generates the same pseudorandom set
+        this.stocks = SerializationUtil.loadObjectList(Stock.class, "/stocks.csv");
+        this.spf = generatePortfolioFor("lars");
+        this.tr = tr;
+        this.spr = spr;
+        generateTrades(this.spf);
+    }
+
+    private StockPortfolio generatePortfolioFor(String userName) {
+
+        StockPortfolio spf = new StockPortfolio();
+
+        for (Stock stock : stocks) {
+            spf.getStocks().put(stock, 100.0);
+        }
+
+        spf.setPortfolioHolderName("userName");
+        spf.setPortfolioId(Utilities.generatePortfolioIdFrom(userName));
+        spr.save(spf);
+        return spf;
+    }
+
+    private void generateTrades(StockPortfolio spf) {
+        for (Stock stock : spf.getStocks().keySet()) {
+            Double amountLeft = spf.getStocks().get(stock).doubleValue();
+            int amountToSell = this.random.nextInt(10);
+            Trade trade = new Trade();
+            trade.setAmount(amountToSell);
+            trade.setCreatedDate(df.format(new Date()));
+            trade.setFromPortfolio(Utilities.generatePortfolioIdFrom(spf.getPortfolioHolderName()));
+            trade.setFromPortfolioName(spf.getPortfolioHolderName());
+            trade.setStock(stock);
+            tr.save(trade);
+            amountLeft = amountLeft - amountToSell;
+            spf.getStocks().put(stock, amountLeft);
+        }
+        spr.save(spf);
+    }
+}

src/main/java/com/redislabs/demos/redisbank/investments/StockPortfolioRepository.java

@@ -0,0 +1,7 @@
+package com.redislabs.demos.redisbank.investments;
+
+import org.springframework.data.repository.CrudRepository;
+
+public interface StockPortfolioRepository extends CrudRepository<StockPortfolio, String> {
+
+}

src/main/java/com/redislabs/demos/redisbank/investments/Trade.java

@@ -0,0 +1,27 @@
+package com.redislabs.demos.redisbank.investments;
+
+import org.springframework.data.annotation.Id;
+import org.springframework.data.redis.core.RedisHash;
+
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+@Data
+@NoArgsConstructor
+@AllArgsConstructor
+@RedisHash
+public class Trade {
+    
+    @Id
+    private Long id;
+    private Stock stock;
+    private int amount;
+    private String fromPortfolio;
+    private String fromPortfolioName;
+    private String toPortfolio;
+    private String toPortfolioName;
+    private String createdDate;
+    private String settledDate;
+    private String settledBy;
+}

src/main/java/com/redislabs/demos/redisbank/investments/TradeController.java

@@ -0,0 +1,31 @@
+package com.redislabs.demos.redisbank.investments;
+
+import java.text.DateFormat;
+import java.text.SimpleDateFormat;
+import java.util.Date;
+import java.util.Optional;
+
+public class TradeController {
+
+    private final DateFormat df = new SimpleDateFormat("yyyy.MM.dd 'at' HH:mm:ss");
+    private final TradeRepository tr;
+
+    public TradeController(TradeRepository tr) {
+        this.tr = tr;
+    }
+
+    public void settleTrade(String tradeId)    {
+        Optional<Trade> tradeMaybe = tr.findById(tradeId);
+        
+        if (tradeMaybe.isPresent())  {
+            Trade trade = tradeMaybe.get();
+            trade.setSettledBy("Trading desk");
+            trade.setSettledDate(df.format(new Date()));
+            tr.save(trade);
+        }else{
+            // Trade not found
+        }
+
+    }
+
+}

src/main/java/com/redislabs/demos/redisbank/investments/TradeRepository.java

@@ -0,0 +1,7 @@
+package com.redislabs.demos.redisbank.investments;
+
+import org.springframework.data.repository.CrudRepository;
+
+public interface TradeRepository extends CrudRepository<Trade, String>{
+    
+}

src/main/resources/stocks.csv

@@ -0,0 +1,9 @@
+id,name
+1,BMW
+2,Philips
+3,Redis Labs
+4,VMware
+5,Pivotal
+6,Dell
+7,Aegon
+8,ASML