diff --git a/agt-module-infra/agt-module-infra-server/src/main/java/org/agt/module/infra/framework/file/core/client/db/DBFileClient.java b/agt-module-infra/agt-module-infra-server/src/main/java/org/agt/module/infra/framework/file/core/client/db/DBFileClient.java index fa31360..f93eb90 100644 --- a/agt-module-infra/agt-module-infra-server/src/main/java/org/agt/module/infra/framework/file/core/client/db/DBFileClient.java +++ b/agt-module-infra/agt-module-infra-server/src/main/java/org/agt/module/infra/framework/file/core/client/db/DBFileClient.java @@ -1,6 +1,7 @@ package org.agt.module.infra.framework.file.core.client.db; import cn.hutool.core.collection.CollUtil; +import cn.hutool.core.util.StrUtil; import cn.hutool.extra.spring.SpringUtil; import org.agt.module.infra.dal.dataobject.file.FileContentDO; import org.agt.module.infra.dal.mysql.file.FileContentMapper; @@ -18,6 +19,13 @@ public class DBFileClient extends AbstractFileClient { private FileContentMapper fileContentMapper; + /** + * 邮件的开关,默认为 true + */ + private String domain; + + private String port; + public DBFileClient(Long id, DBFileClientConfig config) { super(id, config); } @@ -25,6 +33,8 @@ public class DBFileClient extends AbstractFileClient { @Override protected void doInit() { fileContentMapper = SpringUtil.getBean(FileContentMapper.class); + domain = SpringUtil.getProperty("agt.domain", "http://localhost"); + port = SpringUtil.getProperty("server.port", "48080"); } @Override @@ -33,7 +43,11 @@ public class DBFileClient extends AbstractFileClient { .setPath(path).setContent(content); fileContentMapper.insert(contentDO); // 拼接返回路径 - return super.formatFileUrl(config.getDomain(), path); + String fileDomain = config.getDomain(); + if (StrUtil.isBlank(fileDomain)) { + fileDomain = domain + ":" + port; + } + return super.formatFileUrl(fileDomain, path); } @Override