Package ai.specformula.spring

Class TrackingDataSource

java.lang.Object

ai.specformula.spring.TrackingDataSource

All Implemented Interfaces:

Wrapper, CommonDataSource, DataSource

public class TrackingDataSource
extends Object
implements DataSource

JDBC DataSource 追蹤代理，本身即為 DataSource。

透過 JDK 動態代理攔截 getConnection() 回傳的 Connection / Statement， 解析 SQL 中的 INSERT INTO 並記錄被寫入的 table。 清理時只清理 dirty tables，大幅減少 disposeWorld 時間。

由 TestcontainerFactory 建立時自動包裝； SpecFormulaBridge.setDataSource(String, DataSource) 會以 instanceof TrackingDataSource 判斷是否啟用 dirty table tracking。

Method Summary

Modifier and Type	Method	Description
void	clearDirtyTables()	清除 dirty table 記錄（每個 scenario 結束後呼叫）。
Connection	getConnection()
Connection	getConnection(String username, String password)
Set<String>	getDirtyTables()	取得被 INSERT 過的 table 名稱集合。
int	getLoginTimeout()
PrintWriter	getLogWriter()
DataSource	getOriginal()	取得原始 DataSource。
Logger	getParentLogger()
boolean	isWrapperFor(Class<?> iface)
void	setLoginTimeout(int seconds)
void	setLogWriter(PrintWriter out)
<T> T	unwrap(Class<T> iface)
static TrackingDataSource	wrap(DataSource ds)	包裝 DataSource，回傳 TrackingDataSource 實例。

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface javax.sql.CommonDataSource

createShardingKeyBuilder

Methods inherited from interface javax.sql.DataSource

createConnectionBuilder

Method Details

wrap

public static TrackingDataSource wrap(DataSource ds)

包裝 DataSource，回傳 TrackingDataSource 實例。

getOriginal

public DataSource getOriginal()

取得原始 DataSource。

getDirtyTables

public Set<String> getDirtyTables()

取得被 INSERT 過的 table 名稱集合。

clearDirtyTables

public void clearDirtyTables()

清除 dirty table 記錄（每個 scenario 結束後呼叫）。

getConnection

public Connection getConnection()
                         throws SQLException

Specified by:

getConnection in interface DataSource

Throws:

SQLException

getConnection

public Connection getConnection(String username,
 String password)
                         throws SQLException

Specified by:

getConnection in interface DataSource

Throws:

SQLException

getLogWriter

public PrintWriter getLogWriter()
                         throws SQLException

Specified by:

getLogWriter in interface CommonDataSource

Specified by:

getLogWriter in interface DataSource

Throws:

SQLException

setLogWriter

public void setLogWriter(PrintWriter out)
                  throws SQLException

Specified by:

setLogWriter in interface CommonDataSource

Specified by:

setLogWriter in interface DataSource

Throws:

SQLException

setLoginTimeout

public void setLoginTimeout(int seconds)
                     throws SQLException

Specified by:

setLoginTimeout in interface CommonDataSource

Specified by:

setLoginTimeout in interface DataSource

Throws:

SQLException

getLoginTimeout

public int getLoginTimeout()
                    throws SQLException

Specified by:

getLoginTimeout in interface CommonDataSource

Specified by:

getLoginTimeout in interface DataSource

Throws:

SQLException

getParentLogger

public Logger getParentLogger()
                       throws SQLFeatureNotSupportedException

Specified by:

getParentLogger in interface CommonDataSource

Throws:

SQLFeatureNotSupportedException

unwrap

public <T> T unwrap(Class<T> iface)
             throws SQLException

Specified by:

unwrap in interface Wrapper

Throws:

SQLException

isWrapperFor

public boolean isWrapperFor(Class<?> iface)
                     throws SQLException

Specified by:

isWrapperFor in interface Wrapper

Throws:

SQLException