Form构建

Form节点

先读取form10k数据，构建form节点

    private final FormRepository formRepository;
    
    @SneakyThrows
    @PostMapping("node")
    public void createNodes() {
        var fileDir = new File("F:\\workspace\\code\\learn\\sec-edgar-notebooks\\data\\sample\\form10k");
        File[] files = fileDir.listFiles();
        for (File file : files) {
            if (!file.getName().contains(".json")) continue;
            var form10K = new ObjectMapper().readValue(file, Form10K.class);
            var fullText = "About " +
                           String.join(",", form10K.getNames()) +
                           "..." +
                           form10K.getItem1() +
                           "\n" +
                           form10K.getItem1a() +
                           "\n" +
                           form10K.getItem7() +
                           "\n" +
                           form10K.getItem7a();
            var formId = file.getName().replace(".json", "");
            var form = Form.builder().id(formId)
                    .fullText(fullText)
                    .cusip6(form10K.getCusip6())
                    .source(form10K.getSource())
                    .build();
            formRepository.save(form);
        }
    }

Form与Chunk的关联

PartOf关系

根据formId关联Chunk节点和Form节点，然后使用merge创建PART_OF关系。

match (c:Chunk), (f:Form) where c.formId=f.id
merge (c)-[r:PART_OF]->(f)
return count(r);

@PostMapping("relationship/section")
    public void createSectionRelationship() {
        neo4jClient.query("""
                        match (c:Chunk),(f:Form) where c.chunkSeqId=0 and f.id = c.formId
                        merge (f)-[r:SECTION {item:c.item}] -> (c)
                        return count(r)
                        """)
                .run();
    }

Section关系

form关联Chunk链接的头节点，c.chunkSeqId=0代表该节点是开头。{item:c.item}是给SECTION关系添加属性。之前说过一个form有多个item因此需要该属性做区分。

match (c:Chunk),(f:Form) where c.chunkSeqId=0 and f.id = c.formId
merge (f)-[r:SECTION {item:c.item}] -> (c)
return count(r)

@PostMapping("relationship/section")
    public void createSectionRelationship() {
        neo4jClient.query("""
                        match (c:Chunk),(f:Form) where c.chunkSeqId=0 and f.id = c.formId
                        merge (f)-[r:SECTION {item:c.item}] -> (c)
                        return count(r)
                        """)
                .run();
    }