Orientdb--寫入數據

最近在調研圖數據庫，由於使用的是斯坦福數據（pyorientdb 不支持3.7協議，因此換成kotlin進行寫入數據）。

腳本內容

class ImportListener : ApplicationListener<ApplicationReadyEvent> {

    private val log = LoggerFactory.getLogger(ImportListener::class.java)

    override fun onApplicationEvent(event: ApplicationReadyEvent) {
    // 連接orientdb
        val orientDB = OrientDB("remote:localhost", OrientDBConfig.defaultConfig())
        val db = orientDB.open("pgap_test", "root", "haizhi1234")
        importVer(db)
        importEdge(db)
        db.close()
        orientDB.close()
    }

// 導入邊
    private fun importEdge(db: ODatabaseSession) {
        if (db.getClass("stf_edge") == null)
            db.createEdgeClass("stf_edge")

        readFileFromResources("/data/soc-pokec-relationships.txt") {
            val addEdge = findVertexById(it[0], db)
            addEdge?.addEdge(findVertexById(it[1], db))
            addEdge?.save<OVertexDocument>()
            log.info("----edge is ---${it[0]} to ${it[1]}")
        }
    }

	// 導入邊
    private fun importVer(db: ODatabaseSession) {
        var stfVe = db.getClass("stf")
        if (stfVe == null)
            stfVe = db.createVertexClass("stf")
        if (stfVe.getProperty("name") == null) {
            stfVe.createProperty("name", OType.STRING)
            stfVe.createIndex("stf_name_index", OClass.INDEX_TYPE.NOTUNIQUE, "name")
        }
        if (stfVe.getProperty("id") == null) {
            stfVe.createProperty("id", OType.STRING)
            stfVe.createIndex("stf_id_index", OClass.INDEX_TYPE.UNIQUE, "id")
        }
        if (stfVe.getProperty("createTime") == null) {
            stfVe.createProperty("createTime", OType.STRING)
        }
        if (stfVe.getProperty("endTime") == null) {
            stfVe.createProperty("endTime", OType.STRING)
        }
        if (stfVe.getProperty("type") == null) {
            stfVe.createProperty("type", OType.STRING)
        }

        readFileFromResources("/data/soc-pokec-profiles.txt") {
            val vAddress = db.newVertex("stf")
            // 創建頂點
            vAddress.setProperty("id", it[0])
            vAddress.setProperty("name", it[4])
            vAddress.setProperty("createTime", it[5])
            vAddress.setProperty("endTime", it[6])
            vAddress.setProperty("type", it[7])
            vAddress.save<OVertex>()
            log.info("----v is ---${it[0]}")
        }
    }

// 讀取文件
    private fun readFileFromResources(path: String, apply: (List<String>) -> Unit) {
//        val resourceAsStream = Thread.currentThread().contextClassLoader.getResourceAsStream(path)
        val file = File(path)
        // 讀取
        val bufferedReader = BufferedReader(InputStreamReader( FileInputStream(file)))
        while (true) {
            val readLine = bufferedReader.readLine()
            if (StringUtils.isEmpty(readLine))
                break
            val split = readLine.split('\t')
            apply.invoke(split)
        }
    }

    fun findVertexById(id: String, db: ODatabaseSession): OVertex? {
        val query = "select * from `stf` where id = ?"
        val findFirst = db.query(query, id).vertexStream().findFirst()
        if (!findFirst.isPresent) {
            return null
        }
        return findFirst.get()
    }
}

引用

官方java操作

git demo地址

demo地址