新增数据库更新，新增mybatiss plus 插件 ，天眼查接口新增
anthonywj
2 years ago
parent
4671fd9bc2
commit
5a00ea8b6a
@ -0,0 +1,28 @@
|
||||
package com.glxp.udidl.admin.config;
|
||||
|
||||
import lombok.Getter;
|
||||
|
||||
/**
|
||||
* MybatisPlus自定义SQL方法枚举
|
||||
*/
|
||||
@Getter
|
||||
public enum CustomerSqlMethod {
|
||||
/**
|
||||
* 插入
|
||||
*/
|
||||
INSERT_IGNORE_ONE("insertIgnore", "插入一条数据（选择字段插入），如果中已经存在相同的记录，则忽略当前新数据", "<script>\nINSERT IGNORE INTO %s %s VALUES %s\n</script>"),
|
||||
/**
|
||||
* 替换
|
||||
*/
|
||||
REPLACE_ONE("replace", "替换一条数据（选择字段插入），存在则替换，不存在则插入", "<script>\nREPLACE INTO %s %s VALUES %s\n</script>");
|
||||
|
||||
private final String method;
|
||||
private final String desc;
|
||||
private final String sql;
|
||||
|
||||
CustomerSqlMethod(String method, String desc, String sql) {
|
||||
this.method = method;
|
||||
this.desc = desc;
|
||||
this.sql = sql;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,81 @@
|
||||
package com.glxp.udidl.admin.config;
|
||||
|
||||
import cn.hutool.core.io.IORuntimeException;
|
||||
import cn.hutool.core.io.IoUtil;
|
||||
import cn.hutool.core.lang.UUID;
|
||||
import com.glxp.udidl.admin.dao.sys.DbVersionDao;
|
||||
import com.glxp.udidl.admin.entity.sys.DbVersionEntity;
|
||||
import com.glxp.udidl.admin.entity.sys.SchemaData;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.boot.ApplicationArguments;
|
||||
import org.springframework.boot.ApplicationRunner;
|
||||
import org.springframework.core.annotation.Order;
|
||||
import org.springframework.stereotype.Component;
|
||||
|
||||
import javax.annotation.Resource;
|
||||
import java.io.InputStream;
|
||||
import java.util.ArrayList;
|
||||
import java.util.Date;
|
||||
import java.util.List;
|
||||
|
||||
@Order(1)
|
||||
@Component
|
||||
@Slf4j
|
||||
public class HdSchemaExecutor implements ApplicationRunner {
|
||||
|
||||
@Resource
|
||||
DbVersionDao hdCommonDao;
|
||||
|
||||
private List<SchemaData> schema = new ArrayList<>();
|
||||
|
||||
@Override
|
||||
public void run(ApplicationArguments args) throws Exception {
|
||||
//初始版本列表
|
||||
buildSchemas();
|
||||
//定义sql文件路径
|
||||
String basePath = "schemas/";
|
||||
//非版本控制，初始化脚本
|
||||
ClassLoader loader = this.getClass().getClassLoader();
|
||||
//通过流的方式获取项目路径下的文件
|
||||
InputStream inputStream = loader.getResourceAsStream(basePath + "init.sql");
|
||||
//获取文件内容
|
||||
String sql = IoUtil.readUtf8(inputStream);
|
||||
try {
|
||||
//判断版本表是否存在
|
||||
int count = hdCommonDao.selectTableExist("hd_version");
|
||||
if (count == 0) {
|
||||
hdCommonDao.updateSql(sql);
|
||||
}
|
||||
for (SchemaData schemaData : schema) {
|
||||
//查询版本记录是否存在
|
||||
count = hdCommonDao.selectVersion(schemaData.getVersion());
|
||||
if (count == 0) {
|
||||
log.info("--------------执行数据脚本，版本：" + schemaData.getVersion());
|
||||
//获取对应sql脚本
|
||||
inputStream = loader.getResourceAsStream(basePath + schemaData.getFileName());
|
||||
sql = IoUtil.readUtf8(inputStream);
|
||||
hdCommonDao.updateSql(sql);
|
||||
DbVersionEntity entity = new DbVersionEntity();
|
||||
entity.setId(UUID.randomUUID().toString());
|
||||
entity.setVersion(schemaData.getVersion());
|
||||
entity.setCreated(new Date());
|
||||
entity.setRemark(schemaData.getFileName());
|
||||
//写入版本记录
|
||||
hdCommonDao.insertVersion(entity);
|
||||
}
|
||||
}
|
||||
|
||||
} catch (IORuntimeException e) {
|
||||
e.printStackTrace();
|
||||
} finally {
|
||||
//关闭流
|
||||
inputStream.close();
|
||||
}
|
||||
}
|
||||
|
||||
public void buildSchemas() {
|
||||
schema.add(new SchemaData("v2.1", "schema_v2.1.sql"));
|
||||
// schema.add(new SchemaData("v2.2", "schema_v2.2.sql"));
|
||||
// schema.add(new SchemaData("v2.3", "schema_v2.3.sql"));
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,76 @@
|
||||
package com.glxp.udidl.admin.config;
|
||||
|
||||
import com.baomidou.mybatisplus.annotation.IdType;
|
||||
import com.baomidou.mybatisplus.core.enums.SqlMethod;
|
||||
import com.baomidou.mybatisplus.core.injector.AbstractMethod;
|
||||
import com.baomidou.mybatisplus.core.metadata.TableFieldInfo;
|
||||
import com.baomidou.mybatisplus.core.metadata.TableInfo;
|
||||
import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
|
||||
import com.baomidou.mybatisplus.core.toolkit.sql.SqlScriptUtils;
|
||||
import lombok.Setter;
|
||||
import lombok.experimental.Accessors;
|
||||
import org.apache.ibatis.executor.keygen.Jdbc3KeyGenerator;
|
||||
import org.apache.ibatis.executor.keygen.KeyGenerator;
|
||||
import org.apache.ibatis.executor.keygen.NoKeyGenerator;
|
||||
import org.apache.ibatis.mapping.MappedStatement;
|
||||
import org.apache.ibatis.mapping.SqlSource;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.function.Predicate;
|
||||
|
||||
/**
|
||||
* 批量新增数据,自选字段 insert ignore
|
||||
*/
|
||||
public class InsertIgnoreBatch extends AbstractMethod {
|
||||
/**
|
||||
* mapper 对应的方法名
|
||||
*/
|
||||
private static final String MAPPER_METHOD = "insertIgnoreBatch";
|
||||
|
||||
/**
|
||||
* 字段筛选条件
|
||||
*/
|
||||
@Setter
|
||||
@Accessors(chain = true)
|
||||
private Predicate<TableFieldInfo> predicate;
|
||||
|
||||
@Override
|
||||
public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
|
||||
KeyGenerator keyGenerator = new NoKeyGenerator();
|
||||
com.glxp.udidl.admin.config.CustomerSqlMethod sqlMethod = com.glxp.udidl.admin.config.CustomerSqlMethod.INSERT_IGNORE_ONE;
|
||||
List<TableFieldInfo> fieldList = tableInfo.getFieldList();
|
||||
String insertSqlColumn = tableInfo.getKeyInsertSqlColumn(false, true) +
|
||||
this.filterTableFieldInfo(fieldList, predicate, TableFieldInfo::getInsertSqlColumn, EMPTY);
|
||||
String columnScript = LEFT_BRACKET + insertSqlColumn.substring(0, insertSqlColumn.length() - 1) + RIGHT_BRACKET;
|
||||
String insertSqlProperty = tableInfo.getKeyInsertSqlProperty(false, ENTITY_DOT, true) +
|
||||
this.filterTableFieldInfo(fieldList, predicate, i -> i.getInsertSqlProperty(ENTITY_DOT), EMPTY);
|
||||
insertSqlProperty = LEFT_BRACKET + insertSqlProperty.substring(0, insertSqlProperty.length() - 1) + RIGHT_BRACKET;
|
||||
String valuesScript = SqlScriptUtils.convertForeach(insertSqlProperty, "list", null, ENTITY, COMMA);
|
||||
String keyProperty = null;
|
||||
String keyColumn = null;
|
||||
// 表包含主键处理逻辑,如果不包含主键当普通字段处理
|
||||
if (tableInfo.havePK()) {
|
||||
if (tableInfo.getIdType() == IdType.AUTO) {
|
||||
/* 自增主键 */
|
||||
keyGenerator = new Jdbc3KeyGenerator();
|
||||
keyProperty = tableInfo.getKeyProperty();
|
||||
keyColumn = tableInfo.getKeyColumn();
|
||||
} else {
|
||||
if (null != tableInfo.getKeySequence()) {
|
||||
keyGenerator = TableInfoHelper.genKeyGenerator(MAPPER_METHOD, tableInfo, builderAssistant);
|
||||
keyProperty = tableInfo.getKeyProperty();
|
||||
keyColumn = tableInfo.getKeyColumn();
|
||||
}
|
||||
}
|
||||
}
|
||||
String sql = String.format(sqlMethod.getSql(), tableInfo.getTableName(), columnScript, valuesScript);
|
||||
SqlSource sqlSource = languageDriver.createSqlSource(configuration, sql, modelClass);
|
||||
return this.addInsertMappedStatement(mapperClass, modelClass, MAPPER_METHOD, sqlSource, keyGenerator, keyProperty, keyColumn);
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getMethod(SqlMethod sqlMethod) {
|
||||
// 自定义 mapper 方法名
|
||||
return MAPPER_METHOD;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,24 @@
|
||||
package com.glxp.udidl.admin.dao.sys;
|
||||
|
||||
import com.glxp.udidl.admin.entity.sys.DbVersionEntity;
|
||||
import org.apache.ibatis.annotations.Mapper;
|
||||
import org.apache.ibatis.annotations.Param;
|
||||
import org.apache.ibatis.annotations.Update;
|
||||
|
||||
@Mapper
|
||||
public interface DbVersionDao {
|
||||
|
||||
int selectVersion(@Param("version") String version);
|
||||
|
||||
//查询版本表是否存在
|
||||
int selectTableExist(@Param("tableName") String tableName);
|
||||
|
||||
//新增版本
|
||||
int insertVersion(DbVersionEntity entity);
|
||||
|
||||
//执行sql
|
||||
@Update("${sql}")
|
||||
void updateSql(@Param("sql") String sql);
|
||||
|
||||
|
||||
}
|
||||
@ -1,9 +1,10 @@
|
||||
package com.glxp.udidl.admin.dao.tyapi;
|
||||
|
||||
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
|
||||
import com.glxp.udidl.admin.dao.BaseMapperPlus;
|
||||
import com.glxp.udidl.admin.entity.tyapi.TySupplierEntity;
|
||||
import org.apache.ibatis.annotations.Mapper;
|
||||
|
||||
@Mapper
|
||||
public interface TySupplierMapper extends BaseMapper<TySupplierEntity> {
|
||||
public interface TySupplierMapper extends BaseMapperPlus<TySupplierMapper, TySupplierEntity, TySupplierEntity> {
|
||||
}
|
||||
|
||||
@ -0,0 +1,15 @@
|
||||
package com.glxp.udidl.admin.entity.sys;
|
||||
|
||||
import lombok.Data;
|
||||
|
||||
import java.util.Date;
|
||||
|
||||
@Data
|
||||
public class DbVersionEntity {
|
||||
|
||||
private String id;
|
||||
private String version;
|
||||
private String remark;
|
||||
private Date created;
|
||||
|
||||
}
|
||||
@ -0,0 +1,22 @@
|
||||
package com.glxp.udidl.admin.entity.sys;
|
||||
|
||||
import lombok.Data;
|
||||
|
||||
|
||||
@Data
|
||||
public class SchemaData {
|
||||
/**
|
||||
* 版本号
|
||||
*/
|
||||
public String version;
|
||||
|
||||
/**
|
||||
* 文件名
|
||||
*/
|
||||
public String fileName;
|
||||
|
||||
public SchemaData(String version, String fileName) {
|
||||
this.version = version;
|
||||
this.fileName = fileName;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,131 @@
|
||||
package com.glxp.udidl.admin.res.tyapi;
|
||||
|
||||
import com.fasterxml.jackson.annotation.JsonProperty;
|
||||
import lombok.Data;
|
||||
import lombok.NoArgsConstructor;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
@NoArgsConstructor
|
||||
@Data
|
||||
public class CompanyBaseResponse {
|
||||
|
||||
@JsonProperty("result")
|
||||
private ResultDTO result;
|
||||
@JsonProperty("reason")
|
||||
private String reason;
|
||||
@JsonProperty("error_code")
|
||||
private Integer errorCode;
|
||||
|
||||
@NoArgsConstructor
|
||||
@Data
|
||||
public static class ResultDTO {
|
||||
@JsonProperty("historyNames")
|
||||
private String historyNames;
|
||||
@JsonProperty("cancelDate")
|
||||
private Object cancelDate;
|
||||
@JsonProperty("regStatus")
|
||||
private String regStatus;
|
||||
@JsonProperty("regCapital")
|
||||
private String regCapital;
|
||||
@JsonProperty("city")
|
||||
private String city;
|
||||
@JsonProperty("staffNumRange")
|
||||
private String staffNumRange;
|
||||
@JsonProperty("bondNum")
|
||||
private String bondNum;
|
||||
@JsonProperty("historyNameList")
|
||||
private List<String> historyNameList;
|
||||
@JsonProperty("industry")
|
||||
private String industry;
|
||||
@JsonProperty("bondName")
|
||||
private String bondName;
|
||||
@JsonProperty("revokeDate")
|
||||
private Object revokeDate;
|
||||
@JsonProperty("type")
|
||||
private Integer type;
|
||||
@JsonProperty("updateTimes")
|
||||
private Long updateTimes;
|
||||
@JsonProperty("legalPersonName")
|
||||
private String legalPersonName;
|
||||
@JsonProperty("revokeReason")
|
||||
private String revokeReason;
|
||||
@JsonProperty("regNumber")
|
||||
private String regNumber;
|
||||
@JsonProperty("creditCode")
|
||||
private String creditCode;
|
||||
@JsonProperty("property3")
|
||||
private String property3;
|
||||
@JsonProperty("usedBondName")
|
||||
private String usedBondName;
|
||||
@JsonProperty("approvedTime")
|
||||
private Long approvedTime;
|
||||
@JsonProperty("fromTime")
|
||||
private Long fromTime;
|
||||
@JsonProperty("socialStaffNum")
|
||||
private Integer socialStaffNum;
|
||||
@JsonProperty("actualCapitalCurrency")
|
||||
private String actualCapitalCurrency;
|
||||
@JsonProperty("alias")
|
||||
private String alias;
|
||||
@JsonProperty("companyOrgType")
|
||||
private String companyOrgType;
|
||||
@JsonProperty("id")
|
||||
private Integer id;
|
||||
@JsonProperty("cancelReason")
|
||||
private String cancelReason;
|
||||
@JsonProperty("orgNumber")
|
||||
private String orgNumber;
|
||||
@JsonProperty("email")
|
||||
private String email;
|
||||
@JsonProperty("toTime")
|
||||
private Object toTime;
|
||||
@JsonProperty("actualCapital")
|
||||
private String actualCapital;
|
||||
@JsonProperty("estiblishTime")
|
||||
private Long estiblishTime;
|
||||
@JsonProperty("regInstitute")
|
||||
private String regInstitute;
|
||||
@JsonProperty("businessScope")
|
||||
private String businessScope;
|
||||
@JsonProperty("taxNumber")
|
||||
private String taxNumber;
|
||||
@JsonProperty("regLocation")
|
||||
private String regLocation;
|
||||
@JsonProperty("regCapitalCurrency")
|
||||
private String regCapitalCurrency;
|
||||
@JsonProperty("tags")
|
||||
private String tags;
|
||||
@JsonProperty("websiteList")
|
||||
private String websiteList;
|
||||
@JsonProperty("phoneNumber")
|
||||
private String phoneNumber;
|
||||
@JsonProperty("district")
|
||||
private String district;
|
||||
@JsonProperty("bondType")
|
||||
private String bondType;
|
||||
@JsonProperty("name")
|
||||
private String name;
|
||||
@JsonProperty("percentileScore")
|
||||
private Integer percentileScore;
|
||||
@JsonProperty("industryAll")
|
||||
private IndustryAllDTO industryAll;
|
||||
@JsonProperty("isMicroEnt")
|
||||
private Integer isMicroEnt;
|
||||
@JsonProperty("base")
|
||||
private String base;
|
||||
|
||||
@NoArgsConstructor
|
||||
@Data
|
||||
public static class IndustryAllDTO {
|
||||
@JsonProperty("categoryMiddle")
|
||||
private String categoryMiddle;
|
||||
@JsonProperty("categoryBig")
|
||||
private String categoryBig;
|
||||
@JsonProperty("category")
|
||||
private String category;
|
||||
@JsonProperty("categorySmall")
|
||||
private String categorySmall;
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,47 @@
|
||||
package com.glxp.udidl.admin.res.tyapi;
|
||||
|
||||
import com.fasterxml.jackson.annotation.JsonProperty;
|
||||
import lombok.Data;
|
||||
import lombok.NoArgsConstructor;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
@NoArgsConstructor
|
||||
@Data
|
||||
public class ContactResponse {
|
||||
|
||||
@JsonProperty("result")
|
||||
private ResultDTO result;
|
||||
@JsonProperty("reason")
|
||||
private String reason;
|
||||
@JsonProperty("error_code")
|
||||
private Integer errorCode;
|
||||
|
||||
@NoArgsConstructor
|
||||
@Data
|
||||
public static class ResultDTO {
|
||||
@JsonProperty("phoneNumber")
|
||||
private String phoneNumber;
|
||||
@JsonProperty("allCalls")
|
||||
private List<AllCallsDTO> allCalls;
|
||||
@JsonProperty("regLocation")
|
||||
private String regLocation;
|
||||
@JsonProperty("id")
|
||||
private Integer id;
|
||||
@JsonProperty("email")
|
||||
private String email;
|
||||
@JsonProperty("websiteList")
|
||||
private String websiteList;
|
||||
|
||||
@NoArgsConstructor
|
||||
@Data
|
||||
public static class AllCallsDTO {
|
||||
@JsonProperty("phoneNumber")
|
||||
private String phoneNumber;
|
||||
@JsonProperty("source")
|
||||
private String source;
|
||||
@JsonProperty("tag")
|
||||
private String tag;
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,58 @@
|
||||
package com.glxp.udidl.admin.res.tyapi;
|
||||
|
||||
import com.fasterxml.jackson.annotation.JsonProperty;
|
||||
import lombok.Data;
|
||||
import lombok.NoArgsConstructor;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
@NoArgsConstructor
|
||||
@Data
|
||||
public class OpenSearchResponse {
|
||||
@JsonProperty("result")
|
||||
private ResultDTO result;
|
||||
@JsonProperty("reason")
|
||||
private String reason;
|
||||
@JsonProperty("error_code")
|
||||
private Integer errorCode;
|
||||
|
||||
@NoArgsConstructor
|
||||
@Data
|
||||
public static class ResultDTO {
|
||||
@JsonProperty("total")
|
||||
private Integer total;
|
||||
@JsonProperty("items")
|
||||
private List<ItemsDTO> items;
|
||||
|
||||
@NoArgsConstructor
|
||||
@Data
|
||||
public static class ItemsDTO {
|
||||
@JsonProperty("regStatus")
|
||||
private String regStatus;
|
||||
@JsonProperty("estiblishTime")
|
||||
private String estiblishTime;
|
||||
@JsonProperty("regCapital")
|
||||
private String regCapital;
|
||||
@JsonProperty("companyType")
|
||||
private Integer companyType;
|
||||
@JsonProperty("matchType")
|
||||
private String matchType;
|
||||
@JsonProperty("type")
|
||||
private Integer type;
|
||||
@JsonProperty("legalPersonName")
|
||||
private String legalPersonName;
|
||||
@JsonProperty("regNumber")
|
||||
private String regNumber;
|
||||
@JsonProperty("creditCode")
|
||||
private String creditCode;
|
||||
@JsonProperty("name")
|
||||
private String name;
|
||||
@JsonProperty("id")
|
||||
private Integer id;
|
||||
@JsonProperty("orgNumber")
|
||||
private String orgNumber;
|
||||
@JsonProperty("base")
|
||||
private String base;
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,68 @@
|
||||
package com.glxp.udidl.admin.res.tyapi;
|
||||
|
||||
import com.fasterxml.jackson.annotation.JsonProperty;
|
||||
import lombok.Data;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
|
||||
@Data
|
||||
public class SupplierResponse {
|
||||
|
||||
@JsonProperty("result")
|
||||
private ResultDTO result;
|
||||
@JsonProperty("reason")
|
||||
private String reason;
|
||||
@JsonProperty("error_code")
|
||||
private Integer errorCode;
|
||||
|
||||
|
||||
@Data
|
||||
public static class ResultDTO {
|
||||
@JsonProperty("suppliesYear")
|
||||
private List<SuppliesYearDTO> suppliesYear;
|
||||
@JsonProperty("pageBean")
|
||||
private PageBeanDTO pageBean;
|
||||
|
||||
|
||||
@Data
|
||||
public static class PageBeanDTO {
|
||||
@JsonProperty("result")
|
||||
private List<ResultDTO.PageBeanDTO.SupplierDTO> result;
|
||||
@JsonProperty("total")
|
||||
private Integer total;
|
||||
|
||||
|
||||
@Data
|
||||
public static class SupplierDTO {
|
||||
@JsonProperty("supplier_graphId")
|
||||
private Long supplier_graphId;
|
||||
@JsonProperty("announcement_date")
|
||||
private Long announcement_date;
|
||||
@JsonProperty("amt")
|
||||
private String amt;
|
||||
@JsonProperty("logo")
|
||||
private String logo;
|
||||
@JsonProperty("alias")
|
||||
private String alias;
|
||||
@JsonProperty("supplier_name")
|
||||
private String supplier_name;
|
||||
@JsonProperty("relationship")
|
||||
private String relationship;
|
||||
@JsonProperty("dataSource")
|
||||
private String dataSource;
|
||||
@JsonProperty("ratio")
|
||||
private String ratio;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@Data
|
||||
public static class SuppliesYearDTO {
|
||||
@JsonProperty("title")
|
||||
private String title;
|
||||
@JsonProperty("value")
|
||||
private String value;
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,12 @@
|
||||
package com.glxp.udidl.admin.service;
|
||||
|
||||
import com.baomidou.mybatisplus.extension.service.IService;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
public interface CustomService<T> extends IService<T> {
|
||||
int insertIgnore(T entity);
|
||||
boolean insertIgnoreBatch(List<T> entityList);
|
||||
int replace(T entity);
|
||||
boolean replaceBatch(List<T> entityList);
|
||||
}
|
||||
@ -0,0 +1,29 @@
|
||||
package com.glxp.udidl.admin.service;
|
||||
|
||||
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
|
||||
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
||||
import com.glxp.udidl.admin.dao.BaseMapperPlus;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
public class CustomServiceImpl<M extends BaseMapper<T>, T> extends ServiceImpl<BaseMapperPlus<M, T, T>, T> implements CustomService<T> {
|
||||
@Override
|
||||
public int insertIgnore(T entity) {
|
||||
return baseMapper.insertIgnore(entity);
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean insertIgnoreBatch(List<T> entityList) {
|
||||
return baseMapper.insertIgnoreBatchs(entityList);
|
||||
}
|
||||
|
||||
@Override
|
||||
public int replace(T entity) {
|
||||
return baseMapper.replace(entity);
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean replaceBatch(List<T> entityList) {
|
||||
return baseMapper.replaceBatchs(entityList);
|
||||
}
|
||||
}
|
||||
@ -1,10 +1,48 @@
|
||||
package com.glxp.udidl.admin.service.tyapi;
|
||||
|
||||
import cn.hutool.core.bean.BeanUtil;
|
||||
import cn.hutool.core.collection.CollUtil;
|
||||
import com.glxp.udidl.admin.res.tyapi.SupplierResponse;
|
||||
import org.springframework.stereotype.Service;
|
||||
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
||||
import com.glxp.udidl.admin.entity.tyapi.TySupplierEntity;
|
||||
import com.glxp.udidl.admin.dao.tyapi.TySupplierMapper;
|
||||
|
||||
import javax.annotation.Resource;
|
||||
import java.util.Date;
|
||||
import java.util.List;
|
||||
|
||||
@Service
|
||||
public class TySupplierService extends ServiceImpl<TySupplierMapper, TySupplierEntity> {
|
||||
@Resource
|
||||
TyDlHttpClient tyDlHttpClient;
|
||||
@Resource
|
||||
TySupplierMapper tySupplierMapper;
|
||||
|
||||
public void download(String hospital) {
|
||||
|
||||
Integer pageNum = 1;
|
||||
Integer pageSize = 20;
|
||||
while (true) {
|
||||
SupplierResponse supplierResponse = tyDlHttpClient.getSupplier(hospital, pageNum, pageSize);
|
||||
if (supplierResponse.getErrorCode() == 0) {
|
||||
List<SupplierResponse.ResultDTO.PageBeanDTO.SupplierDTO> supplierDTOS = supplierResponse.getResult().getPageBean().getResult();
|
||||
List<TySupplierEntity> tySupplierEntities = BeanUtil.copyToList(supplierDTOS, TySupplierEntity.class);
|
||||
if (CollUtil.isNotEmpty(supplierDTOS)) {
|
||||
for (TySupplierEntity tySupplierEntity : tySupplierEntities) {
|
||||
tySupplierEntity.setHospName(hospital);
|
||||
tySupplierEntity.setUpdateTime(new Date());
|
||||
}
|
||||
}
|
||||
tySupplierMapper.insertBatch(tySupplierEntities);
|
||||
if (CollUtil.isNotEmpty(supplierDTOS) && supplierDTOS.size() == 20) {
|
||||
pageNum = pageNum + 1;
|
||||
} else {
|
||||
return;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
||||
@ -0,0 +1,184 @@
|
||||
package com.glxp.udidl.admin.util;
|
||||
|
||||
import cn.hutool.core.collection.CollUtil;
|
||||
import cn.hutool.core.lang.SimpleCache;
|
||||
import cn.hutool.core.map.MapUtil;
|
||||
import cn.hutool.core.util.ObjectUtil;
|
||||
import cn.hutool.core.util.ReflectUtil;
|
||||
import cn.hutool.core.util.StrUtil;
|
||||
import lombok.AccessLevel;
|
||||
import lombok.NoArgsConstructor;
|
||||
import org.springframework.cglib.beans.BeanCopier;
|
||||
import org.springframework.cglib.beans.BeanMap;
|
||||
import org.springframework.cglib.core.Converter;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* bean深拷贝工具(基于 cglib 性能优异)
|
||||
* <p>
|
||||
* 重点 cglib 不支持 拷贝到链式对象
|
||||
* 例如: 源对象 拷贝到 目标(链式对象)
|
||||
* 请区分好`浅拷贝`和`深拷贝`再做使用
|
||||
*
|
||||
* @author Lion Li
|
||||
*/
|
||||
@NoArgsConstructor(access = AccessLevel.PRIVATE)
|
||||
public class BeanCopyUtils {
|
||||
|
||||
/**
|
||||
* 单对象基于class创建拷贝
|
||||
*
|
||||
* @param source 数据来源实体
|
||||
* @param desc 描述对象 转换后的对象
|
||||
* @return desc
|
||||
*/
|
||||
public static <T, V> V copy(T source, Class<V> desc) {
|
||||
if (ObjectUtil.isNull(source)) {
|
||||
return null;
|
||||
}
|
||||
if (ObjectUtil.isNull(desc)) {
|
||||
return null;
|
||||
}
|
||||
final V target = ReflectUtil.newInstanceIfPossible(desc);
|
||||
return copy(source, target);
|
||||
}
|
||||
|
||||
/**
|
||||
* 单对象基于对象创建拷贝
|
||||
*
|
||||
* @param source 数据来源实体
|
||||
* @param desc 转换后的对象
|
||||
* @return desc
|
||||
*/
|
||||
public static <T, V> V copy(T source, V desc) {
|
||||
if (ObjectUtil.isNull(source)) {
|
||||
return null;
|
||||
}
|
||||
if (ObjectUtil.isNull(desc)) {
|
||||
return null;
|
||||
}
|
||||
BeanCopier beanCopier = BeanCopierCache.INSTANCE.get(source.getClass(), desc.getClass(), null);
|
||||
beanCopier.copy(source, desc, null);
|
||||
return desc;
|
||||
}
|
||||
|
||||
/**
|
||||
* 列表对象基于class创建拷贝
|
||||
*
|
||||
* @param sourceList 数据来源实体列表
|
||||
* @param desc 描述对象 转换后的对象
|
||||
* @return desc
|
||||
*/
|
||||
public static <T, V> List<V> copyList(List<T> sourceList, Class<V> desc) {
|
||||
if (ObjectUtil.isNull(sourceList)) {
|
||||
return null;
|
||||
}
|
||||
if (CollUtil.isEmpty(sourceList)) {
|
||||
return CollUtil.newArrayList();
|
||||
}
|
||||
return StreamUtils.toList(sourceList, source -> {
|
||||
V target = ReflectUtil.newInstanceIfPossible(desc);
|
||||
copy(source, target);
|
||||
return target;
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* bean拷贝到map
|
||||
*
|
||||
* @param bean 数据来源实体
|
||||
* @return map对象
|
||||
*/
|
||||
@SuppressWarnings("unchecked")
|
||||
public static <T> Map<String, Object> copyToMap(T bean) {
|
||||
if (ObjectUtil.isNull(bean)) {
|
||||
return null;
|
||||
}
|
||||
return BeanMap.create(bean);
|
||||
}
|
||||
|
||||
/**
|
||||
* map拷贝到bean
|
||||
*
|
||||
* @param map 数据来源
|
||||
* @param beanClass bean类
|
||||
* @return bean对象
|
||||
*/
|
||||
public static <T> T mapToBean(Map<String, Object> map, Class<T> beanClass) {
|
||||
if (MapUtil.isEmpty(map)) {
|
||||
return null;
|
||||
}
|
||||
if (ObjectUtil.isNull(beanClass)) {
|
||||
return null;
|
||||
}
|
||||
T bean = ReflectUtil.newInstanceIfPossible(beanClass);
|
||||
return mapToBean(map, bean);
|
||||
}
|
||||
|
||||
/**
|
||||
* map拷贝到bean
|
||||
*
|
||||
* @param map 数据来源
|
||||
* @param bean bean对象
|
||||
* @return bean对象
|
||||
*/
|
||||
public static <T> T mapToBean(Map<String, Object> map, T bean) {
|
||||
if (MapUtil.isEmpty(map)) {
|
||||
return null;
|
||||
}
|
||||
if (ObjectUtil.isNull(bean)) {
|
||||
return null;
|
||||
}
|
||||
BeanMap.create(bean).putAll(map);
|
||||
return bean;
|
||||
}
|
||||
|
||||
/**
|
||||
* BeanCopier属性缓存<br>
|
||||
* 缓存用于防止多次反射造成的性能问题
|
||||
*
|
||||
* @author Looly
|
||||
* @since 5.4.1
|
||||
*/
|
||||
public enum BeanCopierCache {
|
||||
/**
|
||||
* BeanCopier属性缓存单例
|
||||
*/
|
||||
INSTANCE;
|
||||
|
||||
private final SimpleCache<String, BeanCopier> cache = new SimpleCache<>();
|
||||
|
||||
/**
|
||||
* 获得类与转换器生成的key在{@link BeanCopier}的Map中对应的元素
|
||||
*
|
||||
* @param srcClass 源Bean的类
|
||||
* @param targetClass 目标Bean的类
|
||||
* @param converter 转换器
|
||||
* @return Map中对应的BeanCopier
|
||||
*/
|
||||
public BeanCopier get(Class<?> srcClass, Class<?> targetClass, Converter converter) {
|
||||
final String key = genKey(srcClass, targetClass, converter);
|
||||
return cache.get(key, () -> BeanCopier.create(srcClass, targetClass, converter != null));
|
||||
}
|
||||
|
||||
/**
|
||||
* 获得类与转换器生成的key
|
||||
*
|
||||
* @param srcClass 源Bean的类
|
||||
* @param targetClass 目标Bean的类
|
||||
* @param converter 转换器
|
||||
* @return 属性名和Map映射的key
|
||||
*/
|
||||
private String genKey(Class<?> srcClass, Class<?> targetClass, Converter converter) {
|
||||
final StringBuilder key = StrUtil.builder()
|
||||
.append(srcClass.getName()).append('#').append(targetClass.getName());
|
||||
if(null != converter){
|
||||
key.append('#').append(converter.getClass().getName());
|
||||
}
|
||||
return key.toString();
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,21 @@
|
||||
<?xml version="1.0" encoding="UTF-8" ?>
|
||||
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd" >
|
||||
<mapper namespace="com.glxp.udidl.admin.dao.sys.DbVersionDao">
|
||||
<select id="selectVersion" resultType="int">
|
||||
selecT count(1)
|
||||
from sys_db_version
|
||||
where version = #{version}
|
||||
</select>
|
||||
|
||||
<select id="selectTableExist" resultType="int">
|
||||
select count(*) count
|
||||
from information_schema.TABLES
|
||||
where TABLE_NAME = #{tableName}
|
||||
and table_schema = (select database())
|
||||
</select>
|
||||
|
||||
<insert id="insertVersion">
|
||||
insert into sys_db_version(id, version, remark, created)
|
||||
values (uuid(), #{version}, #{remark}, #{created})
|
||||
</insert>
|
||||
</mapper>
|
||||
@ -0,0 +1,105 @@
|
||||
SET NAMES utf8mb4;
|
||||
SET FOREIGN_KEY_CHECKS = 0;
|
||||
DROP TABLE IF EXISTS `sys_db_version`;
|
||||
CREATE TABLE `sys_db_version`
|
||||
(
|
||||
`id` varchar(64) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NOT NULL,
|
||||
`version` varchar(64) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NULL DEFAULT NULL COMMENT '版本号',
|
||||
`created` datetime(0) NULL DEFAULT NULL COMMENT '创建时间',
|
||||
`remark` varchar(500) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NULL DEFAULT NULL COMMENT '备注',
|
||||
PRIMARY KEY (`id`) USING BTREE
|
||||
) ENGINE = InnoDB
|
||||
CHARACTER SET = utf8mb4
|
||||
COLLATE = utf8mb4_0900_ai_ci COMMENT = '数据版本'
|
||||
ROW_FORMAT = Dynamic;
|
||||
|
||||
SET FOREIGN_KEY_CHECKS = 1;
|
||||
|
||||
|
||||
/* 创建函数Pro_Temp_ColumnWork操作表字段 */
|
||||
DROP PROCEDURE IF EXISTS Pro_Temp_ColumnWork;
|
||||
CREATE PROCEDURE `Pro_Temp_ColumnWork`(TableName VARCHAR(50), ColumnName VARCHAR(50), SqlStr VARCHAR(4000), CType INT)
|
||||
BEGIN
|
||||
DECLARE
|
||||
Rows1 INT;
|
||||
|
||||
SET Rows1 = 0;
|
||||
SELECT COUNT(*)
|
||||
INTO Rows1
|
||||
FROM INFORMATION_SCHEMA.COLUMNS
|
||||
WHERE table_schema = DATABASE()
|
||||
AND upper(table_name) = TableName
|
||||
AND upper(column_name) = ColumnName;
|
||||
IF
|
||||
(CType = 1 AND Rows1 <= 0) THEN
|
||||
|
||||
SET SqlStr := CONCAT('ALTER TABLE ', TableName, ' ADD COLUMN ', ColumnName, ' ', SqlStr);
|
||||
|
||||
ELSEIF (CType = 2 AND Rows1 > 0) THEN
|
||||
|
||||
SET SqlStr := CONCAT('ALTER TABLE ', TableName, ' MODIFY ', ColumnName, ' ', SqlStr);
|
||||
|
||||
ELSEIF (CType = 3 AND Rows1 > 0) THEN
|
||||
|
||||
SET SqlStr := CONCAT('ALTER TABLE ', TableName, ' DROP COLUMN ', ColumnName);
|
||||
ELSE
|
||||
SET SqlStr := '';
|
||||
|
||||
END IF;
|
||||
IF
|
||||
(SqlStr <> '') THEN
|
||||
|
||||
SET @SQL1 = SqlStr;
|
||||
PREPARE stmt1
|
||||
FROM
|
||||
@SQL1;
|
||||
EXECUTE stmt1;
|
||||
|
||||
END IF;
|
||||
|
||||
END;
|
||||
/** 函数创建结束 **/
|
||||
|
||||
/*创建定义普通索引函数*/
|
||||
DROP PROCEDURE IF EXISTS Modify_index;
|
||||
|
||||
CREATE PROCEDURE Modify_index(
|
||||
TableName VARCHAR(50),
|
||||
ColumnNames VARCHAR(500),
|
||||
idx_name VARCHAR(50),
|
||||
idx_type VARCHAR(50))
|
||||
BEGIN
|
||||
DECLARE
|
||||
Rows1 int;
|
||||
DECLARE
|
||||
SqlStr VARCHAR(4000);
|
||||
DECLARE
|
||||
target_database VARCHAR(100);
|
||||
SELECT DATABASE
|
||||
()
|
||||
INTO target_database;
|
||||
|
||||
SET Rows1 = 0;
|
||||
SELECT COUNT(*)
|
||||
INTO Rows1
|
||||
FROM information_schema.statistics
|
||||
WHERE table_schema = DATABASE()
|
||||
AND upper(table_name) = upper(TableName)
|
||||
AND upper(index_name) = upper(idx_name);
|
||||
IF Rows1 <= 0 THEN
|
||||
SET SqlStr :=
|
||||
CONCAT('alter table ', TableName, ' ADD INDEX ', idx_name, '(', ColumnNames, ') USING ', idx_type);
|
||||
END IF;
|
||||
IF
|
||||
(SqlStr <> '') THEN
|
||||
|
||||
SET @SQL1 = SqlStr;
|
||||
PREPARE stmt1
|
||||
FROM
|
||||
@SQL1;
|
||||
EXECUTE stmt1;
|
||||
|
||||
END IF;
|
||||
|
||||
END;
|
||||
/*创建定义普通索引函数结束*/
|
||||
Loading…
Reference in New Issue