diff --git a/mybatis-flex-solon-plugin/src/main/java/com/mybatisflex/solon/integration/XPluginImpl.java b/mybatis-flex-solon-plugin/src/main/java/com/mybatisflex/solon/integration/XPluginImpl.java index fd8f4aa4..4647f3ea 100644 --- a/mybatis-flex-solon-plugin/src/main/java/com/mybatisflex/solon/integration/XPluginImpl.java +++ b/mybatis-flex-solon-plugin/src/main/java/com/mybatisflex/solon/integration/XPluginImpl.java @@ -87,7 +87,11 @@ public class XPluginImpl implements Plugin { String dsName = entry.getKey(); DataSource ds = new DataSourceBuilder(entry.getValue()).build(); BeanWrap bw = context.wrap(dsName, ds, dsName.equals(flexProperties.getDefaultDatasourceKey())); - loadDs(context, bw); + context.putWrap(dsName, bw); + if (bw.typed()) { + context.putWrap(DataSource.class, bw); + } + context.wrapPublish(bw); } } }