Merge pull request eugenp#14305 from Eredisse/master

BAEL-6024 - Guide to YugabyteDB
wangrupeng · Jul 7, 2023 · 7e0db60 · 7e0db60
2 parents 69a5a4b + 2c806b3
commit 7e0db60
Show file tree

Hide file tree

Showing 5 changed files with 138 additions and 95 deletions.
diff --git a/persistence-modules/spring-data-yugabytedb/pom.xml b/persistence-modules/spring-data-yugabytedb/pom.xml
@@ -1,55 +1,51 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
-		 xmlns="http://maven.apache.org/POM/4.0.0"
-		 xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
-	<modelVersion>4.0.0</modelVersion>
-	<artifactId>spring-data-yugabytedb</artifactId>
-	<version>1.0</version>
-	<name>spring-data-yugabytedb</name>
-	<packaging>jar</packaging>
+         xmlns="http://maven.apache.org/POM/4.0.0"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <modelVersion>4.0.0</modelVersion>
+    <artifactId>spring-data-yugabytedb</artifactId>
+    <version>1.0</version>
+    <name>spring-data-yugabytedb</name>
+    <packaging>jar</packaging>
 
-	<parent>
-		<groupId>com.baeldung</groupId>
-		<artifactId>parent-boot-2</artifactId>
-		<version>0.0.1-SNAPSHOT</version>
-		<relativePath>../../parent-boot-2</relativePath>
-	</parent>
+    <parent>
+        <groupId>com.baeldung</groupId>
+        <artifactId>parent-boot-2</artifactId>
+        <version>0.0.1-SNAPSHOT</version>
+        <relativePath>../../parent-boot-2</relativePath>
+    </parent>
 
-	<dependencies>
-		<dependency>
-			<groupId>org.springframework.boot</groupId>
-			<artifactId>spring-boot-starter-web</artifactId>
-		</dependency>
-		<dependency>
-			<groupId>org.projectlombok</groupId>
-			<artifactId>lombok</artifactId>
-		</dependency>
-		<dependency>
-			<groupId>org.springframework</groupId>
-			<artifactId>spring-test</artifactId>
-		</dependency>
-		<dependency>
-			<groupId>org.springframework.boot</groupId>
-			<artifactId>spring-boot-starter-test</artifactId>
-			<scope>test</scope>
-		</dependency>
-		<dependency>
-			<groupId>org.postgresql</groupId>
-			<artifactId>postgresql</artifactId>
-		</dependency>
-		<dependency>
-			<groupId>org.springframework.boot</groupId>
-			<artifactId>spring-boot-starter-data-jpa</artifactId>
-		</dependency>
-	</dependencies>
+    <dependencies>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-web</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>org.springframework</groupId>
+            <artifactId>spring-test</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-test</artifactId>
+            <scope>test</scope>
+        </dependency>
+        <dependency>
+            <groupId>org.postgresql</groupId>
+            <artifactId>postgresql</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-data-jpa</artifactId>
+        </dependency>
+    </dependencies>
 
-	<build>
-		<plugins>
-			<plugin>
-				<groupId>org.apache.maven.plugins</groupId>
-				<artifactId>maven-surefire-plugin</artifactId>
-			</plugin>
-		</plugins>
-	</build>
+    <build>
+        <plugins>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-surefire-plugin</artifactId>
+            </plugin>
+        </plugins>
+    </build>
 
 </project>
diff --git a/persistence-modules/spring-data-yugabytedb/src/main/java/com/baeldung/Main.java b/persistence-modules/spring-data-yugabytedb/src/main/java/com/baeldung/Main.java
@@ -8,25 +8,25 @@
 @SpringBootApplication
 public class Main implements CommandLineRunner {
 
-	@Autowired
-	private UserRepository userRepository;
+    @Autowired
+    private UserRepository userRepository;
 
-	public static void main(String[] args) {
-		SpringApplication.run(Main.class, args);
-	}
+    public static void main(String[] args) {
+        SpringApplication.run(Main.class, args);
+    }
 
-	@Override
-	public void run(String... args) throws InterruptedException {
+    @Override
+    public void run(String... args) throws InterruptedException {
 
-		int iterationCount = 1_000;
-		int elementsPerIteration = 100;
+        int iterationCount = 1_000;
+        int elementsPerIteration = 100;
 
-		for (int i = 0; i < iterationCount; i++) {
-			for (long j = 0; j < elementsPerIteration; j++) {
-				User user = new User();
-				userRepository.save(user);
-			}
-			Thread.sleep(1000);
-		}
-	}
+        for (int i = 0; i < iterationCount; i++) {
+            for (long j = 0; j < elementsPerIteration; j++) {
+                User user = new User();
+                userRepository.save(user);
+            }
+            Thread.sleep(1000);
+        }
+    }
 }
diff --git a/persistence-modules/spring-data-yugabytedb/src/main/java/com/baeldung/User.java b/persistence-modules/spring-data-yugabytedb/src/main/java/com/baeldung/User.java
@@ -11,34 +11,31 @@
 @Table(name = "users")
 public class User {
 
-	@Id
-	@GeneratedValue(strategy = GenerationType.IDENTITY)
-	private Long id;
-
-	@Column
-	private String name;
-
-	Long getId() {
-		return id;
-	}
-
-	void setId(Long id) {
-		this.id = id;
-	}
-
-	String getName() {
-		return name;
-	}
-
-	void setName(String name) {
-		this.name = name;
-	}
-
-	@Override
-	public String toString() {
-		return "User{" +
-				"id=" + id +
-				", name='" + name + '\'' +
-				'}';
-	}
+    @Id
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    private Long id;
+
+    @Column
+    private String name;
+
+    public Long getId() {
+        return id;
+    }
+
+    public void setId(Long id) {
+        this.id = id;
+    }
+
+    public String getName() {
+        return name;
+    }
+
+    public void setName(String name) {
+        this.name = name;
+    }
+
+    @Override
+    public String toString() {
+        return "User{" + "id=" + id + ", name='" + name + '\'' + '}';
+    }
 }
diff --git a/persistence-modules/spring-data-yugabytedb/src/main/resources/docker-compose.yml b/persistence-modules/spring-data-yugabytedb/src/main/resources/docker-compose.yml
@@ -0,0 +1,13 @@
+version: '3'
+
+services:
+  yugabytedb:
+    image: yugabytedb/yugabyte:latest
+    container_name: yugabyte
+    user: root
+    ports:
+      - '5433:5433'
+      - '7000:7000'
+      - '9000:9000'
+    command: ["bin/yugabyted", "start", "--daemon=false"]
+
diff --git a/...istence-modules/spring-data-yugabytedb/src/test/java/com/baeldung/YugabyteDBLiveTest.java b/...istence-modules/spring-data-yugabytedb/src/test/java/com/baeldung/YugabyteDBLiveTest.java
@@ -0,0 +1,37 @@
+package com.baeldung;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+import java.util.List;
+
+import org.junit.jupiter.api.Test;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.context.TestPropertySource;
+import org.springframework.test.context.junit.jupiter.SpringJUnitConfig;
+
+/*
+ To run this test we need to run the databases first.
+ A dedicated docker-compose.yml file is located under the resources directory.
+ We can run it by simple executing `docker-compose up`.
+ */
+@SpringJUnitConfig
+@SpringBootTest
+@TestPropertySource("classpath:application.properties")
+public class YugabyteDBLiveTest {
+
+	@Autowired
+	private UserRepository userRepository;
+
+	@Test
+	void givenTwoUsers_whenPersistUsingJPARepository_thenUserAreSaved() {
+	    User user1 = new User();
+	    user1.setName("Alex");
+	    User user2 = new User();
+	    user2.setName("John");
+	    userRepository.save(user1);
+	    userRepository.save(user2);
+	    List<User> allUsers = userRepository.findAll();
+	    assertEquals(2, allUsers.size());
+	}
+}