[Spring cloud 一步步实现广告系统] 14. 全量索引代码实现

  • 时间:
  • 浏览:2
  • 来源:幸运快3_快3平台官网_幸运快3平台官网

上一节当.我 都儿实现了索引基本操作的类以及索引缓存工具类,本小节当.我 都儿之后刚结速实现加载全量索引数据,在加载全量索引数据之后,当.我 都儿需用先将数据库中的表数据导出到一份文件中。Let's code.

1.首先定义另一4个常量类,用来存储导出文件存储的目录和文件名称

之后当.我 都儿导出的文件需用在搜索服务中使用到,之后,当.我 都儿将文件名 & 目录以及导出对象的信息编写在mscx-ad-commom项目中。

public class FileConstant {
    public static final String DATA_ROOT_DIR = "/Users/xxx/Documents/promotion/data/mysql/";

    //各个表数据的存储文件名
    public static final String AD_PLAN = "ad_plan.data";
    public static final String AD_UNIT = "ad_unit.data";
    public static final String AD_CREATIVE = "ad_creative.data";
    public static final String AD_CREATIVE_RELARION_UNIT = "ad_creative_relation_unit.data";
    public static final String AD_UNIT_HOBBY = "ad_unit_hobby.data";
    public static final String AD_UNIT_DISTRICT = "ad_unit_district.data";
    public static final String AD_UNIT_KEYWORD = "ad_unit_keyword.data";
}

2.定义索引对象导出的字段信息,依然用Ad_Plan为例。

/**
 * AdPlanTable for 需用导出的表字段信息 => 是搜索索引字段一一对应
 *
 * @author <a href="mailto:magicianisaac@gmail.com">Isaac.Zhang | 若初</a>
 */
@Data
@AllArgsConstructor
@NoArgsConstructor
public class AdPlanTable {
    private Long planId;
    private Long userId;
    private Integer planStatus;
    private Date startDate;
    private Date endDate;
}

3.导出文件服务实现

同样,最好的实现妙招 却说我将导出服务作为另一4个子工程来独立运行,我这里直接实现在了mscx-ad-db项目中

  • 定义另一4个空接口,为了符合当.我 都儿的编码规范
/**
 * IExportDataService for 导出数据库广告索引初始化数据
 *
 * @author <a href="mailto:magicianisaac@gmail.com">Isaac.Zhang | 若初</a>
 */
public interface IExportDataService {
}
  • 实现service
@Slf4j
@Service
public class ExportDataServiceImpl implements IExportDataService {

    @Autowired
    private AdPlanRepository planRepository;

    /**
     * 导出 {@code AdPlan} from DB to File
     *
     * @param fileName 文件名称
     */
    public void exportAdPlanTable(String fileName) {
        List<AdPlan> planList = planRepository.findAllByPlanStatus(CommonStatus.VALID.getStatus());
        if (CollectionUtils.isEmpty(planList)) {
            return;
        }

        List<AdPlanTable> planTables = new ArrayList<>();
        planList.forEach(item -> planTables.add(
                new AdPlanTable(
                        item.getPlanId(),
                        item.getUserId(),
                        item.getPlanStatus(),
                        item.getStartDate(),
                        item.getEndDate()
                )
        ));

        //将数据写入文件
        Path path = Paths.get(fileName);
        try (BufferedWriter writer = Files.newBufferedWriter(path)) {
            for (AdPlanTable adPlanTable : planTables) {
                writer.write(JSON.toJSONString(adPlanTable));
                writer.newLine();
            }
            writer.close();
        } catch (IOException e) {
            e.printStackTrace();
            log.error("export AdPlanTable Exception!");
        }
    }
}
  • 实现Controller,提供操作入口
@Slf4j
@Controller
@RequestMapping("/export")
public class ExportDataController {
    private final ExportDataServiceImpl exportDataService;

    @Autowired
    public ExportDataController(ExportDataServiceImpl exportDataService) {
        this.exportDataService = exportDataService;
    }

    @GetMapping("/export-plan")
    public CommonResponse exportAdPlans() {

        exportDataService.exportAdPlanTable(String.format("%s%s", FileConstant.DATA_ROOT_DIR, FileConstant.AD_PLAN));
        return new CommonResponse();
    }
}
  • 结果文件内容如下,每一行都代表了另一4个推广计划
{"endDate":15614382000000,"planId":10,"planStatus":1,"startDate":15614382000000,"userId":10}
{"endDate":15614382000000,"planId":11,"planStatus":1,"startDate":15614382000000,"userId":10}
根据文件内容构建索引

当.我 都儿在之后编写索引服务的之后,创建了其他索引需用使用的实体对象类,比如构建推广计划索引的之后,需用使用到的实体对象com.sxzhongf.ad.index.adplan.AdPlanIndexObject,另另一4个 呢,当.我 都儿在上一节实现索引导出的之后,实体对象又是common 包中的com.sxzhongf.ad.common.export.table.AdPlanTable,读取出来文件中的数据并能反序列化为JSON.parseObject(p, AdPlanTable.class),当.我 都儿需用将另一4个对象做相互映射并能创建索引信息。

1.首先当.我 都儿定义另一4个操作类型枚举,代表当.我 都儿每一次的操作类型(也需用对应到后期binlog监听的操作类型

public enum OperationTypeEnum {
    ADD,
    UPDATE,
    DELETE,
    OTHER;

    public static OperationTypeEnum convert(EventType type) {
        switch (type) {
            case EXT_WRITE_ROWS:
                return ADD;
            case EXT_UPDATE_ROWS:
                return UPDATE;
            case EXT_DELETE_ROWS:
                return DELETE;
            default:
                return OTHER;
        }
    }
}

2.之后全量索引的加载和增量索引加载的本质是一样的,全量索引我我随便说说却说我有一种特殊的增量索引,为了代码的可复用,当.我 都儿创建统一的类来操作索引。

/**
 * AdLevelDataHandler for 通用避免索引类
 * 1. 索引之间地处层级划分,也却说我相互之间拥有依赖关系的划分
 * 2. 加载全量索引我我随便说说是增量索引 "上加"的有一种特殊实现
 *
 * @author <a href="mailto:magicianisaac@gmail.com">Isaac.Zhang | 若初</a>
 */
@Slf4j
public class AdLevelDataHandler {

    /**
     * 实现广告推广计划的第二层级索引实现。
     * (第一级为用户层级,之后用户层级不参与索引,全都从level 2之后刚结速)
     * 第二层级的索引是表示 不依赖于其他索引,之后可被其他索引所依赖
     */
    public static void handleLevel2Index(AdPlanTable adPlanTable, OperationTypeEnum type) {
        // 对象转换
        AdPlanIndexObject planIndexObject = new AdPlanIndexObject(
                adPlanTable.getPlanId(),
                adPlanTable.getUserId(),
                adPlanTable.getPlanStatus(),
                adPlanTable.getStartDate(),
                adPlanTable.getEndDate()
        );

        //调用通用妙招

避免,使用IndexDataTableUtils#of来获取索引的实现类bean
        handleBinlogEvent(
                    // 在前一节当.我

都儿实现了另一4个索引工具类,来获取注入的bean对象
                IndexDataTableUtils.of(AdPlanIndexAwareImpl.class),
                planIndexObject.getPlanId(),
                planIndexObject,
                type
        );
    }

    /**
     * 避免全量索引和增量索引的通用避免妙招

     * K,V代表索引的键和值
     *
     * @param index 索引实现代理类父级
     * @param key   键
     * @param value 值
     * @param type  操作类型
     */
    private static <K, V> void handleBinlogEvent(IIndexAware<K, V> index, K key, V value, OperationTypeEnum type) {
        switch (type) {
            case ADD:
                index.add(key, value);
                break;
            case UPDATE:
                index.update(key, value);
                break;
            case DELETE:
                index.delete(key, value);
                break;
            default:
                break;
        }
    }
}

3.读取文件实现全量索引加载。

之后当.我 都儿文件加载之后需用依赖另另一4个 组件,也却说我当.我 都儿的索引工具类,需用添上加@DependsOn("indexDataTableUtils"),全量索引在系统启动的之后就需用加载,当.我 都儿需用上加@PostConstruct来实现初始化加载,被@PostConstruct修饰的妙招 会在服务器加载Servlet的之后运行,之后只会被服务器调用一次。

@Component
@DependsOn("indexDataTableUtils")
public class IndexFileLoader {

    /**
     * 服务启动时,执行全量索引加载
     */
    @PostConstruct
    public void init() {
        //加载 推广计划
        List<String> adPlanStrings = loadExportedData(String.format("%s%s",
                FileConstant.DATA_ROOT_DIR, FileConstant.AD_PLAN
        ));
        adPlanStrings.forEach(p -> AdLevelDataHandler.handleLevel2Index(
                JSON.parseObject(p, AdPlanTable.class), OperationTypeEnum.ADD
        ));
    }

    /**
     * <h3>读取全量索引加载需用的文件</h3>
     *
     * @param fileName 文件名称
     * @return 文件行数据
     */
    private List<String> loadExportedData(String fileName) {
        try (BufferedReader reader = Files.newBufferedReader(Paths.get(fileName))) {
            return reader.lines().collect(Collectors.toList());
        } catch (IOException e) {
            throw new RuntimeException(e.getMessage());
        }
    }
}

Tips

在实现初始化加载全量索引的过程中,一定要保证数据加载的顺序问题报告 报告 ,之后不同的数据有之后地处着相互依赖的关联关系,一旦顺序写错,会造成进程报错问题报告 报告 。