HHH-19682 - Add Support for GaussDB Lock Timeout (lockwait_timeout)

SweetWuXiaoMei · sebersole · commit 88f2a0687ed9 · 2025-08-06T09:28:42.000-05:00
diff --git a/hibernate-community-dialects/src/main/java/org/hibernate/community/dialect/lock/internal/GaussDBLockingSupport.java b/hibernate-community-dialects/src/main/java/org/hibernate/community/dialect/lock/internal/GaussDBLockingSupport.java
@@ -94,21 +94,32 @@ public Timeout getLockTimeout(Connection connection, SessionFactoryImplementor f
 						return Timeouts.WAIT_FOREVER;
 					}
 					if ( value.endsWith( "min" ) ) {
-						final int min = Integer.parseInt( value.substring( 0, value.length() - 3 ) );
-						return Timeout.milliseconds( min * 60 * 1000 );
+						final int minute = getTimeout( value, 3 );
+						return Timeout.milliseconds( minute * 60 * 1000);
 					}
 					else if ( value.endsWith( "s" ) ) {
-						final int second = Integer.parseInt( value.substring( 0, value.length() - 1 ) );
-						return Timeout.seconds(second);
+						final int seconds  = getTimeout( value, 1 );
+						return Timeout.seconds(seconds);
 					}
-					final int milliseconds = Integer.parseInt( value.substring( 0, value.length() - 2 ) );
+					final int milliseconds  = getTimeout( value, 2 );
 					return Timeout.milliseconds(milliseconds);
 				},
 				connection,
 				factory
 		);
 	}
 
+	private static int getTimeout(String value, int unitLength) {
+		final int number;
+		try {
+			number = Integer.parseInt( value.substring( 0, value.length() - unitLength ) );
+		}
+		catch (NumberFormatException e) {
+			throw new RuntimeException( e );
+		}
+		return number;
+	}
+
 	@Override
 	public void setLockTimeout(Timeout timeout, Connection connection, SessionFactoryImplementor factory) {
 		Helper.setLockTimeout(
