从零构建图数据库：Kuzu实战指南（附Go语言代码示例）

# 安装Go语言Kuzu驱动
go get github.com/kuzudb/go-kuzu

package main

import (
    "fmt"
    "os"
    "github.com/kuzudb/go-kuzu"
)

func main() {
    // 创建磁盘数据库（切换为""即内存模式）
    dbPath := "social_graph_db"
    os.RemoveAll(dbPath) // 清空旧数据
    
    systemConfig := kuzu.DefaultSystemConfig()
    systemConfig.BufferPoolSize = 1024 * 1024 * 1024 // 1GB缓存
    db, _ := kuzu.OpenDatabase(dbPath, systemConfig)
    defer db.Close()
    
    conn, _ := kuzu.OpenConnection(db)
    defer conn.Close()

    // 创建节点与关系表
    schemaQueries := []string{
        `CREATE NODE TABLE User(name STRING, age INT64, PRIMARY KEY (name))`,
        `CREATE NODE TABLE City(name STRING, population INT64, PRIMARY KEY (name))`,
        `CREATE REL TABLE Follows(FROM User TO User, since INT64)`,
        `CREATE REL TABLE LivesIn(FROM User TO City)`,
    }
    
    for _, query := range schemaQueries {
        result, _ := conn.Query(query)
        defer result.Close()
    }

    // 批量导入数据集
    dataQueries := []string{
        `COPY User FROM "user_data.csv"`,
        `COPY City FROM "city_data.csv"`,
        `COPY Follows FROM "follows_data.csv"`,
        `COPY LivesIn FROM "lives_in_data.csv"`,
    }
    
    for _, query := range dataQueries {
        result, _ := conn.Query(query)
        defer result.Close()
    }

    // 查询用户关注关系
    query := `
        MATCH (a:User)-[e:Follows]->(b:User) 
        RETURN a.name AS follower, e.since AS followYear, b.name AS followed
        ORDER BY followYear DESC
    `
    
    result, _ := conn.Query(query)
    defer result.Close()
    
    fmt.Println("┌───────────┬─────────────┬──────────┐")
    fmt.Println("│ Follower  │ Follow Year │ Followed │")
    fmt.Println("├───────────┼─────────────┼──────────┤")
    for result.HasNext() {
        tuple, _ := result.Next()
        slice, _ := tuple.GetAsSlice()
        fmt.Printf("│ %-9s │ %-11d │ %-8s │\n", slice[0], slice[1], slice[2])
    }
    fmt.Println("└───────────┴─────────────┴──────────┘")
}

MATCH (a:User)-[:Follows*2]->(b:User)
RETURN a.name AS origin, b.name AS target

MATCH (u:User)-[:LivesIn]->(c:City)
WHERE c.population > 1000000
RETURN u.name AS resident, c.name AS metropolis

MATCH path = (a:User)-[:Follows*..3]->(b:User)
WHERE a.name = 'Adam'
RETURN nodes(path) AS connectionPath