switched to latest yb smart driver. Also added loadBalance=true

Author: Amit Chauhan

README.md

@@ -10,9 +10,8 @@ YB Workload Simulator is a Java application that simulates workloads against Yug
 * [Code setup and Installation](#code-setup-and-installation)
    * [Build the jar file](#build-the-jar-file)
       * [Additional parameters for YCQL workloads](#additional-parameters-for-ycql-workloads)
-* [How to build your own workload](#how-to-build-your-own-workload)
 * [Start a read and write workload](#start-a-read-and-write-workload)
-* [Create your own workload .java file](#create-your-own-workload-java-file)
+* [Create your own workload](#create-your-own-workload)
 
 ## Download the jar
 
@@ -79,36 +78,12 @@ By default, workload simulator uses "Yugabyte Smart Driver". You can run the App
 -Dspring.profiles.active=pgdriver
 ```
 
-### Run the application on a YugabyteDB Managed cluster
+### Run the application on a YugabyteDB Managed cluster (Aeon)
+YugabyteDB smart driver doesn't connect to YugabyteDB Aeon cluster if you are running your app from your laptop. You will need to run the app from a machine which has VPC peered with Yugabyte Aeon VPC. 
 
-1. To connect the application to your cluster, ensure that you have downloaded the cluster SSL certificate and your computer is added to the IP allow list. Refer to [Before you begin](https://docs.yugabyte.com//preview/develop/build-apps/cloud-add-ip/).
+#### Workaround:
+If you need to run app against YugabyteDB Aeon on laptop, you can use the postgres driver. Please see "Run the application using postgres driver"
 
-1. To start the application against a running YugabyteDB Managed cluster, use the following command:
-
-    ```sh
-    java -Dnode=<host name> \
-         -Ddbname=<dbname> \
-         -Ddbuser=<dbuser> \
-         -Ddbpassword=<dbpassword> \
-         -Dssl=true \
-         -Dsslmode=verify-full \
-         -Dsslrootcert=<path-to-cluster-certificate> \
-         -jar ./yb-workload-sim-0.0.2.jar
-    ```
-
-    Replace the following:
-
-    * `host name` - the host name of your YugabyteDB cluster. For YugabyteDB Managed, select your cluster on the Clusters page, and click Settings. The host is displayed under Connection Parameters.
-
-    * `dbname` - the name of the database you are connecting to (the default is yugabyte).
-
-    * `dbuser` and `dbpassword` - the username and password for the YugabyteDB database. Use the credentials in the credentials file you downloaded when you created your cluster.
-
-    * `path-to-cluster-certificate` with the path to the cluster certificate on your computer.
-
-1. To view the application UI, navigate to `http://localhost:8080`.
-
-    Additional parameters for YugabyteDB Managed to start/stop nodes and scale cluster can be provided in UI by clicking settings gear option on right hand side.
 
 ## Code setup and Installation
 
@@ -134,31 +109,12 @@ A jar file gets created at : <yb-workload-sim>/target/yb-workload-simulator.jar.
 -Dspring.data.cassandra.sslcertpath=<path-to-root-cert> [ex. /Users/username/root.crt]  
 ```
 
-## How to build your own workload
-
-1. Download the latest archive from the [releases](https://github.com/YugabyteDB-Samples/yb-workload-simulator/releases) page and unzip the file.
-
-1. From the root of your project run the following maven command:
-
-    ```sh
-    ./mvnw spring-boot:run -Dspring-boot.run.profiles=dev
-    ```
-
-    A jar file gets created at `<yb-workload-sim>/target/yb-simu-base-app.jar`
-
-    You can also get the jar file in VS Code by right clicking the "WorkloadSimulationApplication.java" file and selecting Run Java.
-
-    ![image](https://user-images.githubusercontent.com/78859174/196289685-74854a5a-1cb5-4b50-81b9-08534bab9a25.png)
-
-    Verify the name of the jar file in the target directory, and use the name of that file wherever you see `yb-workload-sim-0.0.2.jar`.
-
-    ![image](https://user-images.githubusercontent.com/78859174/196288218-13d499ee-a401-4b25-b95c-6e42c64a9824.png)
 
 ## Start a read and write workload
 
 To view the application UI, navigate to <http://localhost:8080>
 
-1. In the application UI, click the hamburger icon at the top of the page beside **Active Workloads for Generic**.
+1. In the application UI, click the hamburger icon at the top of the page beside **Active Workloads**.
 
 1. Select the **Usable Operations** tab.
 
@@ -172,42 +128,19 @@ To view the application UI, navigate to <http://localhost:8080>
 
 The Latency and Throughput charts show the workload running on the cluster.
 
-## Create your own workload .java file
-
-To create your workload file, do the following:
-
-1. Copy the `GenericWorkload.java` file into a file with a new name. The file in this example is named "InstructionsWorkload.java".
-
-    ![image](https://user-images.githubusercontent.com/78859174/196455986-2a4df344-26dc-4fbc-a153-e3b66a71cb6e.png)
-
-    There are 3 "FIX ME" sections to change the name of the class after you copy and paste the "GenericWorkload.java" file.
-
-    ![image](https://user-images.githubusercontent.com/78859174/196456157-fd832363-496b-4b25-ba38-1d9826013517.png)
-
-    ![image](https://user-images.githubusercontent.com/78859174/196456219-19325cf8-5ac7-400f-830f-ff1ba155b8b5.png)
-
-    ![image](https://user-images.githubusercontent.com/78859174/196456330-4d10f7cd-8931-4f19-b664-5a58b702cbe0.png)
-
-1. To be able to call the new workload, you have to modify the `WorkloadConfig.java` file in [src/main/java/com/yugabyte/simulation/config/](src/main/java/com/yugabyte/simulation/config/).
-
-    ![image](https://user-images.githubusercontent.com/78859174/196456659-71160757-e5f5-44fc-9b0a-c87897e27292.png)
-
-    ![image](https://user-images.githubusercontent.com/78859174/196456775-84b931c4-fccf-4abc-aaeb-d3541ad8327d.png)
-
-1. Run the "WorkloadSimulationApplication.java" as a java file.
-
-    ![image](https://user-images.githubusercontent.com/78859174/196457069-5f47b875-51a2-48cc-ae48-cf8991cc93ea.png)
-
-    ![image](https://user-images.githubusercontent.com/78859174/196457185-dfe2ee07-f1ba-4abf-9f59-718fdeb024df.png)
-
-    A new "jar" file gets created in your target directory.
-
-    ![image](https://user-images.githubusercontent.com/78859174/196457412-60f5adc1-aec1-4a14-b8ea-f63fd64a7acb.png)
-
-1. You should be able to run the code with the following command (code from line 136 above):
-
-    ![image](https://user-images.githubusercontent.com/78859174/196458425-88c0951f-c17e-41bb-98ec-d1fba7983128.png)
-
-1. Navigate to <http://localhost:8080> and you will see the IP address of the machine you are using:
-
-    ![image](https://user-images.githubusercontent.com/78859174/196458637-30acf3d6-7fed-49f1-a64e-f38447acc975.png)
+## Create your own workload
+It is very easy to bring in your Data Model and run simulations against it.
+1. Navigate to following directory: src/main/java/com/yugabyte/simulation/service
+2. Clone one of existing workloads (ex. QuikShipWorkload) and create your workload java file (example: MyAwesomeWorkload.java ).
+3. In this new java file, add your DDL and simulation you wish to run.
+4. Add a new entry for this new workload in WorkloadConfig.java file
+```sh
+   @Bean(name="MyAwesomeWorkload")
+   public WorkloadSimulation myAwesomeWorkload(){
+   return new MyAwesomeWorkload();
+   }
+   ```
+5. When running the workload override the default workload
+```sh
+-Dspring.workload=MyAwesomeWorkload
+```

pom.xml

@@ -49,7 +49,7 @@
 		<dependency>
 			<groupId>com.yugabyte</groupId>
 			<artifactId>jdbc-yugabytedb</artifactId>
-			<version>42.3.5-yb-4</version>
+			<version>42.7.3-yb-4</version>
 		</dependency>
 		<dependency>
 			<groupId>com.zaxxer</groupId>

src/main/resources/application-ybdriver.yaml

@@ -19,6 +19,7 @@ spring:
         portNumber: ${port:5433}
         databaseName: ${dbname:yugabyte}
         topologyKeys: "aws.us-west-2.*"
+        loadBalance: true
         #additionalEndpoints: 127.0.0.2:5433,127.0.0.3:5433
         #ssl: ${ssl:false}
         #sslmode: ${sslmode:disable}