0
点赞
收藏
分享

微信扫一扫

Java读写MongoDB数据库


最近写了一个小程序,将一对txt文件中的内容,读出来,写入到MongoDB数据库中,然后查询,代码如下:

package com.test;

import com.mongodb.Block;
import com.mongodb.DBObject;
import com.mongodb.MongoClient;
import com.mongodb.client.FindIterable;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;
import com.mongodb.util.JSON;
import org.bson.BSONObject;
import org.bson.Document;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Set;


public class App {
    public static void main(String[] args) {
        connectMongoDB();
    }

   

    private static void connectMongoDB() {
        try {
            // 连接到 mongodb 服务
            MongoClient mongoClient = new MongoClient("localhost", 27017);

            // 连接到数据库
            MongoDatabase mongoDatabase = mongoClient.getDatabase("Car");
            System.out.println("Connect to database successfully");
            MongoCollection<DBObject> collection = mongoDatabase.getCollection("cars", DBObject.class);
//导入数据
//            List<String> carsFiles = getFiles("C:\\Car");
//            for (String Car : carsFiles) {
//                List<DBObject> carInfos=readFile(Car);
//                collection.insertMany(carInfos);
//            }
            //读取数据
            FindIterable<DBObject> iter = collection.find();
            iter.forEach(new Block<DBObject>() {
                @Override
                public void apply(DBObject dbObject) {
                    System.out.println(dbObject.toString());
                }
            });


        } catch (Exception e) {
            System.err.println(e.getClass().getName() + ": " + e.getMessage());
        }
    }

    /**
     * 获取文件列表
     *
     * @param parentPath
     * @return
     */
    private static List<String> getFiles(String parentPath) {
        List<String> list = new ArrayList<>();
        File file = new File(parentPath);
        if (file.exists()) {
            File[] files = file.listFiles();
            if (null == files || files.length == 0) {
                System.out.println("文件夹是空的!");
            } else {
                for (File file2 : files) {
                    if (file2.isDirectory()) {
                        System.out.println("文件夹:" + file2.getAbsolutePath());
                    } else {
                        list.add(file2.getAbsolutePath());
                    }
                }
            }
        } else {
            System.out.println("文件不存在!");
        }
        return list;
    }

    /**
     * 读取单个文件
     *
     * @param path
     * @return
     */
    private static List<DBObject> readFile(String path) {
        List<DBObject> dbObjects = new ArrayList<>();
        try {
            BufferedReader br = new BufferedReader(new FileReader(path));//构造一个BufferedReader类来读取文件
            String s = null;
            while ((s = br.readLine()) != null) {//使用readLine方法,一次读一行

                String[] sourceStrArray = s.split(",");
                String json = "{" +
                        "'name':'" + sourceStrArray[0] + "'," +
                        "'time':'" + sourceStrArray[1] + "'," +
                        "'lng':'" + sourceStrArray[2] + "'," +
                        "'lat':'" + sourceStrArray[3] + "'" +
                        "}";
                DBObject bson = (DBObject) JSON.parse(json);
                dbObjects.add(bson);
            }
            br.close();
        } catch (Exception e) {
            e.printStackTrace();

        }
        return dbObjects;
    }

    private static List<Car> readFile1(String path) {
        List<Car> cars = new ArrayList<>();
        try {
            BufferedReader br = new BufferedReader(new FileReader(path));//构造一个BufferedReader类来读取文件
            String s = null;
            while ((s = br.readLine()) != null) {//使用readLine方法,一次读一行

                String[] sourceStrArray = s.split(",");
                Car car = new Car();
                car.setName(sourceStrArray[0]);
                car.setTime(sourceStrArray[1]);
                car.setLng(sourceStrArray[2]);
                car.setLat(sourceStrArray[3]);
                cars.add(car);
            }
            br.close();
        } catch (Exception e) {
            e.printStackTrace();

        }
        return cars;
    }
}

 

举报

相关推荐

0 条评论