1

我已经实现了一个 Spring Boot 应用程序，以将所有从文件中检索到的数据放入 JSON 数组中。由于文件太多，它会生成多个 JSON 数组。我已经修改了我的代码以将文件名存储在一个 MongoDB 集合中。它工作正常。

以下是此代码。

@SpringBootApplication
@CrossOrigin(origins = "*", maxAge = 3600)
@RestController
@RequestMapping("/api/auth/logFile")
public class SpringBootSecurityJwtMongodbApplication {

    public static void main(String[] args) {
        SpringApplication.run(SpringBootSecurityJwtMongodbApplication.class, args);
    }


    @Bean
    public ApplicationRunner runner(FTPConfiguration.GateFile gateFile) {
        return args -> {
            List<File> files = gateFile.mget(".");
            for (File file : files) {
                JSONArray arr = new JSONArray();
                System.out.println("Result:" + file.getAbsolutePath());
                run(file, arr);
            }
        };
    }

    void run(File file, JSONArray arr) throws IOException {
        SimpleDateFormat formatter = new SimpleDateFormat("hh:mm:ss");
        Pcap pcap = Pcap.openStream(file);
        JSONObject obj = new JSONObject();
        String fileName = file.getName();
        pcap.loop(
                packet -> {
                    String Time = null;
                    String Source = null;
                    String Destination = null;
                    String dataProtocol = null;
                    Long Length = null;

                    if (packet.hasProtocol(Protocol.TCP)) {
                        TCPPacket packet1 = (TCPPacket) packet.getPacket(Protocol.TCP);
                        Time = formatter.format(new Date(packet1.getArrivalTime() / 1000));
                        Source = packet1.getSourceIP();
                        Destination = packet1.getDestinationIP();
                        dataProtocol = packet1.getProtocol().toString();
                        Length = packet1.getTotalLength();

                    } else if (packet.hasProtocol(Protocol.UDP)) {
                        UDPPacket packet1 = (UDPPacket) packet.getPacket(Protocol.UDP);
                        Time = formatter.format(new Date(packet1.getArrivalTime() / 1000));
                        Source = packet1.getSourceIP();
                        Destination = packet1.getDestinationIP();
                        dataProtocol = packet1.getProtocol().toString();
                        Length = packet1.getTotalLength();

                    } else {
                        System.out.println("Not found protocol. | " + packet.getProtocol());
                    }

                    obj.put("Time", Time);
                    obj.put("Source", Source);
                    obj.put("Destination", Destination);
                    obj.put("Protocol", dataProtocol);
                    obj.put("Length", Length);
                    arr.add(obj);
                    return packet.getNextPacket() != null;
                }
        );
        System.out.println(arr);
        System.out.println(fileName);
        //save file name into LogFile Collection
        Calendar calendar = Calendar.getInstance();
        Date now = calendar.getTime();
        logfileRepo.save(new LogFile("", fileName, now)); //(id,file name, date)
    }

    @Autowired
    private LogFileRepository logfileRepo; 

}

现在我想修改代码来为每个 JSON 数组自动创建集合。这些集合名称应该是每个文件名的对象 ID。谁能帮我这个？

4

0 回答 0