diff --git a/README.md b/README.md index 35b6e6a..08037db 100644 --- a/README.md +++ b/README.md @@ -1,3 +1,277 @@ # knowledge-graph-agent -🤖 以知识图谱为核心的学习辅助对话智能体，集成 层级记忆、自省反馈、CoT、邮件自动化等技术 \ No newline at end of file +🤖 以知识图谱为核心的学习辅助对话智能体，集成 层级记忆、自省反馈、CoT、邮件自动化等技术 + +--- + +## MVP 开发任务清单 + +### 技术栈 + +**后端** + +- 语言: Go 1.26.1 +- 框架: Gin 1.12.0 +- 数据库: Neo4j 5.x +- AI 服务: 硅基流动 OpenAI Completion + +**前端** + +- 框架: React 19.2.4 + TypeScript 5.9.3 +- 图谱可视化: AntV G6 5.1.0 +- 状态管理: Zustand 5.0.12 +- 样式: Tailwind CSS 4.2.2 +- HTTP 客户端: Axios 1.14.0 + +--- + +## 阶段一： Neo4j 数据库集成 + +### 1.1 环境搭建 + +- [x] 连接 Neo4j 5.x 服务器（http://47.121.181.112/） + - 使用服务器提供的数据库实例 + - 验证服务运行状态 + - 配置基础认证（用户名/密码） + - 记录连接信息（bolt://47.121.181.112:7687） +- [ ] 选择并安装 Go Neo4j 客户端库 + - 评估候选库（如 github.com/neo4j/neo4j-go-driver） + - 添加到 go.mod + - 运行 `go mod tidy` + +- [ ] 配置数据库连接字符串 + - 新增配置文件或环境变量：`NEO4J_URI`、`NEO4J_USERNAME`、`NEO4J_PASSWORD` + - 在 `main.go` 中读取并验证连接 + - 实现简单的健康检查接口（GET /db/health） + +### 1.2 数据模型设计与迁移 + +- [ ] 设计 Neo4j 节点与边的属性模型 + - 节点标签（例如：Concept、Tool、Application） + - 节点属性：id、label、type、properties（JSON）、x、y、style（JSON） + - 关系类型：RELATIONSHIP、DEPENDENCY等 + - 关系属性：id、label、type、properties、style（JSON） + +- [ ] 编写数据迁移脚本 `scripts/migrate_to_neo4j.go` + - 读取现有 `data.json` + - 创建 Neo4j 节点（批量 CREATE） + - 创建 Neo4j 关系（批量 CREATE） + - 添加唯一约束：`CREATE CONSTRAINT FOR (n:Node) REQUIRE n.id IS UNIQUE` + - 输出迁移统计（节点数、边数） + +- [ ] 执行迁移并验证数据完整性 + - 运行迁移脚本 + - 在 Neo4j Browser 中执行 Cypher 验证： + - `MATCH (n) RETURN count(n) AS totalNodes` + - `MATCH ()-[r]->() RETURN count(r) AS totalEdges` + - 对比原始 JSON 的统计 + +### 1.3 Neo4j 数据服务层 + +- [ ] 创建 `services/neo4j_service.go` + - 实现 `NewNeo4jService(uri, username, password)` + - 实现基础 CRUD 接口： + - `GetAllNodes() ([]Node, error)` + - `GetAllEdges() ([]Edge, error)` + - `GetNodeByID(id string) (Node, error)` + - `GetNeighbors(nodeID string) (nodes []Node, edges []Edge, error)` + - `SearchNodes(query string) ([]Node, error)` + +- [ ] 在 `services/data_service.go` 中切换数据源 + - 移除 JSON 文件加载逻辑 + - 注入 `Neo4jService` 依赖 + - 更新现有方法委托给 Neo4j 查询： + - `GetGraphData()` + - `GetNodeByID()` + - `GetNeighbors()` + - `SearchNodes()` + - `GetStats()` + +- [ ] 添加并发安全与连接池管理 + - 使用 Neo4j 驱动会话管理 + - 确保连接池合理设置 + - 添加重试机制（简单指数退避） + +### 1.4 API 验证与测试 + +- [ ] 后端单元测试 + - 为 `neo4j_service.go` 编写测试 + - Mock Neo4j 驱动（如有工具支持）或使用集成测试 + - 测试搜索与邻居查询 + +- [ ] API 契约测试 + - 使用 Postman 或 curl 验证： + - GET /api/graph + - GET /api/nodes/:id + - GET /api/nodes/:id/neighbors + - GET /api/search?q=query + +- [ ] 前端兼容性验证 + - 服务端无需修改即可兼容现有前端 + - 验证图谱渲染、搜索、节点详情、邻居加载 + +--- + +## 阶段二：图谱构建功能增强 + +### 2.1 后端 CRUD API + +- [ ] 创建节点 API + - POST /api/nodes + - 请求体：`{label, type, properties, x, y, style}` + - 实现：调用 Neo4j CREATE 节点 + - 返回：新建节点的完整信息 + +- [ ] 更新节点 API + - PUT /api/nodes/:id + - 请求体：`{label, type, properties, x, y, style}` + - 实现：调用 Neo4j SET 更新指定节点属性 + - 返回：更新后节点 + - 错误处理：节点不存在返回 404 + +- [ ] 删除节点 API + - DELETE /api/nodes/:id + - 实现：先删除相关关系，再删除节点（或使用 DETACH DELETE） + - 返回：204 No Content 或 404 + +- [ ] 创建边（关系）API + - POST /api/edges + - 请求体：`{source, target, label, type, properties, style}` + - 实现：使用 Cypher MATCH + CREATE 创建关系 + - 返回：新建边的完整信息 + - 验证 source 和 target 节点是否存在 + +- [ ] 更新边 API + - PUT /api/edges/:id + - 请求体：`{label, type, properties, style}` + - 实现：使用 Cypher 找到边并 SET 属性 + - 返回：更新后边 + +- [ ] 删除边 API + - DELETE /api/edges/:id + - 实现：Cypher DELETE 关系 + - 返回：204 No Content 或 404 + +### 2.2 批量操作 API + +- [ ] 批量创建节点 API + - POST /api/nodes/batch + - 请求体：`{nodes: [...]}` + - 实现：事务批量创建，任一失败则全部回滚 + - 返回：成功与失败清单 + +- [ ] 批量创建边 API + - POST /api/edges/batch + - 请求体：`{edges: [...]}` + - 实现：事务批量创建边 + - 返回：创建成功数量与失败清单 + +### 2.3 前端编辑界面 + +- [ ] 新增编辑状态与管理 + - 在 `graphStore.ts` 中添加 `editingNode` 与 `editingEdge` + - 添加方法：`startEditingNode(node)`、`startEditingEdge(edge)`、`cancelEditing()`、`saveNode/Edge()` + +- [ ] 节点编辑 Dialog 组件 + - 表单字段：label、type、properties（JSON 编辑器）、x、y、style（JSON 编辑器） + - 用于新建与更新节点 + - 集成到节点详情面板或右键菜单 + +- [ ] 边编辑 Dialog 组件 + - 表单字段：label、type、properties、style + - 用于新建与更新边 + - 集成到边点击交互或右键菜单 + +- [ ] 增强图谱视图交互 + - 支持从工具栏拖拽创建节点 + - 支持通过连线工具创建边（点击两个节点） + - 实时回显编辑结果 + +### 2.4 数据持久化与刷新 + +- [ ] 前端自动刷新机制 + - CRUD 操作后自动调用 `loadGraphData()` 刷新图谱 + - 优化：对于增量更新，避免全量重绘 + +- [ ] 添加撤销与重做 + - 维护操作历史栈 + - 提供 Undo/Redo 方法与按钮 + +--- + +## 阶段三：智能对话基础功能 + +### 3.1 AI 客户端封装 + +- [ ] 安装 Go HTTP 客户端库（如 resty 或使用标准库） +- [ ] 创建 `services/ai_service.go` + - 方法：`NewAIService(apiKey, baseURL)` + - 核心方法：`ChatCompletion(messages []Message) (string, error)` + - 实现：调用硅基流动 OpenAI Completion 接口 + - 添加请求超时与错误处理 + +- [ ] 请求与响应模型定义 + - `models/ai.go`：定义 `Message`、`ChatRequest`、`ChatResponse` + - 统一错误类型与包装 + +### 3.2 对话 API + +- [ ] 对话会话管理 + - 创建 `services/conversation_service.go` + - 维护会话上下文（最近 N 条消息） + - 方法：`CreateConversation()`、`AddMessage(sessionID, role, content)`、`GetHistory(sessionID)` + +- [ ] 创建对话 API + - POST /api/chat + - 请求体：`{message, sessionId?}` + - 实现： + 1. 创建或获取会话 + 2. 添加用户消息到历史 + 3. 调用 `AIService.ChatCompletion` + 4. 保存 AI 回复到历史 + 5. 返回 AI 回复 + - 支持流式返回（如平台支持） + +### 3.3 图谱查询与对话集成 + +- [ ] 知识检索策略 + - 在 `neo4j_service.go` 中添加： + - `SearchNodesByKeywords(keywords []string) ([]Node, error)` + - `GetRelevantContext(nodeID string, depth int) ([]Node, []Edge, error)` + - 支持根据消息关键词检索相关节点 + +- [ ] 上下文增强生成 + - 在对话前步骤： + - 提取消息中的关键实体或关键词 + - 检索图谱中的相关节点与关系 + - 将检索结果作为系统提示或上下文注入到对话请求 + - 输出格式：自然语言摘要或结构化 JSON（根据需求） + +- [ ] 优化 AI 提示词 + - 设计 System Prompt：介绍图谱助手角色与能力 + - 提供示例对话 + +### 3.4 前端对话界面 + +- [ ] 创建 Chat 组件 + - 消息列表（用户消息与 AI 消息） + - 输入框与发送按钮 + - 显示加载状态 + +- [ ] 集成到图谱页面 + - 在 `KnowledgeGraph.tsx` 中添加 Chat 组件 + - 配置布局（侧边或底部面板） + +- [ ] 增强交互 + - 支持快捷键发送（Enter） + - 显示会话历史切换（可选） + - 消息支持 Markdown 渲染 + +--- + +## 后续增强阶段（非 MVP） + +- 记忆与自省系统 +- 邮件自动化 +- 定时任务 +- CoT 推理增强 diff --git a/backend/.gitignore b/backend/.gitignore new file mode 100644 index 0000000..d4eeb2c --- /dev/null +++ b/backend/.gitignore @@ -0,0 +1,5 @@ +.env +node_modules/ +dist/ +build/ +*.log \ No newline at end of file diff --git a/backend/.vscode/launch.json b/backend/.vscode/launch.json new file mode 100644 index 0000000..126435f --- /dev/null +++ b/backend/.vscode/launch.json @@ -0,0 +1,22 @@ +{ + // 使用 IntelliSense 了解相关属性。 + // 悬停以查看现有属性的描述。 + // 欲了解更多信息，请访问: https://go.microsoft.com/fwlink/?linkid=830387 + "version": "0.2.0", + "configurations": [ + { + "name": "Launch Package", + "type": "go", + "request": "launch", + "mode": "auto", + "program": "${workspaceFolder}" + }, + { + "name": "Launch Package", + "type": "go", + "request": "launch", + "mode": "auto", + "program": "${fileDirname}" + } + ] +} diff --git a/backend/README.md b/backend/README.md new file mode 100644 index 0000000..6eb02f8 --- /dev/null +++ b/backend/README.md @@ -0,0 +1,277 @@ +# knowledge-graph-agent + +🤖 以知识图谱为核心的学习辅助对话智能体，集成 层级记忆、自省反馈、CoT、邮件自动化等技术 + +--- + +## MVP 开发任务清单 + +### 技术栈 + +**后端** + +- 语言: Go 1.26.1 +- 框架: Gin 1.12.0 +- 数据库: Neo4j 5.x +- AI 服务: 硅基流动 OpenAI Completion + +**前端** + +- 框架: React 19.2.4 + TypeScript 5.9.3 +- 图谱可视化: AntV G6 5.1.0 +- 状态管理: Zustand 5.0.12 +- 样式: Tailwind CSS 4.2.2 +- HTTP 客户端: Axios 1.14.0 + +--- + +## 阶段一： Neo4j 数据库集成 + +### 1.1 环境搭建 + +- [x] 连接 Neo4j 5.x 服务器（http://47.121.181.112/） + - 使用服务器提供的数据库实例 + - 验证服务运行状态 + - 配置基础认证（用户名/密码） + - 记录连接信息（bolt://47.121.181.112:7687） +- [x] 选择并安装 Go Neo4j 客户端库 + - 评估候选库（如 github.com/neo4j/neo4j-go-driver） + - 添加到 go.mod + - 运行 `go mod tidy` + +- [ ] 配置数据库连接字符串 + - 新增配置文件或环境变量：`NEO4J_URI`、`NEO4J_USERNAME`、`NEO4J_PASSWORD` + - 在 `main.go` 中读取并验证连接 + - 实现简单的健康检查接口（GET /db/health） + +### 1.2 数据模型设计与迁移 + +- [ ] 设计 Neo4j 节点与边的属性模型 + - 节点标签（例如：Concept、Tool、Application） + - 节点属性：id、label、type、properties（JSON）、x、y、style（JSON） + - 关系类型：RELATIONSHIP、DEPENDENCY等 + - 关系属性：id、label、type、properties、style（JSON） + +- [ ] 编写数据迁移脚本 `scripts/migrate_to_neo4j.go` + - 读取现有 `data.json` + - 创建 Neo4j 节点（批量 CREATE） + - 创建 Neo4j 关系（批量 CREATE） + - 添加唯一约束：`CREATE CONSTRAINT FOR (n:Node) REQUIRE n.id IS UNIQUE` + - 输出迁移统计（节点数、边数） + +- [ ] 执行迁移并验证数据完整性 + - 运行迁移脚本 + - 在 Neo4j Browser 中执行 Cypher 验证： + - `MATCH (n) RETURN count(n) AS totalNodes` + - `MATCH ()-[r]->() RETURN count(r) AS totalEdges` + - 对比原始 JSON 的统计 + +### 1.3 Neo4j 数据服务层 + +- [ ] 创建 `services/neo4j_service.go` + - 实现 `NewNeo4jService(uri, username, password)` + - 实现基础 CRUD 接口： + - `GetAllNodes() ([]Node, error)` + - `GetAllEdges() ([]Edge, error)` + - `GetNodeByID(id string) (Node, error)` + - `GetNeighbors(nodeID string) (nodes []Node, edges []Edge, error)` + - `SearchNodes(query string) ([]Node, error)` + +- [ ] 在 `services/data_service.go` 中切换数据源 + - 移除 JSON 文件加载逻辑 + - 注入 `Neo4jService` 依赖 + - 更新现有方法委托给 Neo4j 查询： + - `GetGraphData()` + - `GetNodeByID()` + - `GetNeighbors()` + - `SearchNodes()` + - `GetStats()` + +- [ ] 添加并发安全与连接池管理 + - 使用 Neo4j 驱动会话管理 + - 确保连接池合理设置 + - 添加重试机制（简单指数退避） + +### 1.4 API 验证与测试 + +- [ ] 后端单元测试 + - 为 `neo4j_service.go` 编写测试 + - Mock Neo4j 驱动（如有工具支持）或使用集成测试 + - 测试搜索与邻居查询 + +- [ ] API 契约测试 + - 使用 Postman 或 curl 验证： + - GET /api/graph + - GET /api/nodes/:id + - GET /api/nodes/:id/neighbors + - GET /api/search?q=query + +- [ ] 前端兼容性验证 + - 服务端无需修改即可兼容现有前端 + - 验证图谱渲染、搜索、节点详情、邻居加载 + +--- + +## 阶段二：图谱构建功能增强 + +### 2.1 后端 CRUD API + +- [ ] 创建节点 API + - POST /api/nodes + - 请求体：`{label, type, properties, x, y, style}` + - 实现：调用 Neo4j CREATE 节点 + - 返回：新建节点的完整信息 + +- [ ] 更新节点 API + - PUT /api/nodes/:id + - 请求体：`{label, type, properties, x, y, style}` + - 实现：调用 Neo4j SET 更新指定节点属性 + - 返回：更新后节点 + - 错误处理：节点不存在返回 404 + +- [ ] 删除节点 API + - DELETE /api/nodes/:id + - 实现：先删除相关关系，再删除节点（或使用 DETACH DELETE） + - 返回：204 No Content 或 404 + +- [ ] 创建边（关系）API + - POST /api/edges + - 请求体：`{source, target, label, type, properties, style}` + - 实现：使用 Cypher MATCH + CREATE 创建关系 + - 返回：新建边的完整信息 + - 验证 source 和 target 节点是否存在 + +- [ ] 更新边 API + - PUT /api/edges/:id + - 请求体：`{label, type, properties, style}` + - 实现：使用 Cypher 找到边并 SET 属性 + - 返回：更新后边 + +- [ ] 删除边 API + - DELETE /api/edges/:id + - 实现：Cypher DELETE 关系 + - 返回：204 No Content 或 404 + +### 2.2 批量操作 API + +- [ ] 批量创建节点 API + - POST /api/nodes/batch + - 请求体：`{nodes: [...]}` + - 实现：事务批量创建，任一失败则全部回滚 + - 返回：成功与失败清单 + +- [ ] 批量创建边 API + - POST /api/edges/batch + - 请求体：`{edges: [...]}` + - 实现：事务批量创建边 + - 返回：创建成功数量与失败清单 + +### 2.3 前端编辑界面 + +- [ ] 新增编辑状态与管理 + - 在 `graphStore.ts` 中添加 `editingNode` 与 `editingEdge` + - 添加方法：`startEditingNode(node)`、`startEditingEdge(edge)`、`cancelEditing()`、`saveNode/Edge()` + +- [ ] 节点编辑 Dialog 组件 + - 表单字段：label、type、properties（JSON 编辑器）、x、y、style（JSON 编辑器） + - 用于新建与更新节点 + - 集成到节点详情面板或右键菜单 + +- [ ] 边编辑 Dialog 组件 + - 表单字段：label、type、properties、style + - 用于新建与更新边 + - 集成到边点击交互或右键菜单 + +- [ ] 增强图谱视图交互 + - 支持从工具栏拖拽创建节点 + - 支持通过连线工具创建边（点击两个节点） + - 实时回显编辑结果 + +### 2.4 数据持久化与刷新 + +- [ ] 前端自动刷新机制 + - CRUD 操作后自动调用 `loadGraphData()` 刷新图谱 + - 优化：对于增量更新，避免全量重绘 + +- [ ] 添加撤销与重做 + - 维护操作历史栈 + - 提供 Undo/Redo 方法与按钮 + +--- + +## 阶段三：智能对话基础功能 + +### 3.1 AI 客户端封装 + +- [ ] 安装 Go HTTP 客户端库（如 resty 或使用标准库） +- [ ] 创建 `services/ai_service.go` + - 方法：`NewAIService(apiKey, baseURL)` + - 核心方法：`ChatCompletion(messages []Message) (string, error)` + - 实现：调用硅基流动 OpenAI Completion 接口 + - 添加请求超时与错误处理 + +- [ ] 请求与响应模型定义 + - `models/ai.go`：定义 `Message`、`ChatRequest`、`ChatResponse` + - 统一错误类型与包装 + +### 3.2 对话 API + +- [ ] 对话会话管理 + - 创建 `services/conversation_service.go` + - 维护会话上下文（最近 N 条消息） + - 方法：`CreateConversation()`、`AddMessage(sessionID, role, content)`、`GetHistory(sessionID)` + +- [ ] 创建对话 API + - POST /api/chat + - 请求体：`{message, sessionId?}` + - 实现： + 1. 创建或获取会话 + 2. 添加用户消息到历史 + 3. 调用 `AIService.ChatCompletion` + 4. 保存 AI 回复到历史 + 5. 返回 AI 回复 + - 支持流式返回（如平台支持） + +### 3.3 图谱查询与对话集成 + +- [ ] 知识检索策略 + - 在 `neo4j_service.go` 中添加： + - `SearchNodesByKeywords(keywords []string) ([]Node, error)` + - `GetRelevantContext(nodeID string, depth int) ([]Node, []Edge, error)` + - 支持根据消息关键词检索相关节点 + +- [ ] 上下文增强生成 + - 在对话前步骤： + - 提取消息中的关键实体或关键词 + - 检索图谱中的相关节点与关系 + - 将检索结果作为系统提示或上下文注入到对话请求 + - 输出格式：自然语言摘要或结构化 JSON（根据需求） + +- [ ] 优化 AI 提示词 + - 设计 System Prompt：介绍图谱助手角色与能力 + - 提供示例对话 + +### 3.4 前端对话界面 + +- [ ] 创建 Chat 组件 + - 消息列表（用户消息与 AI 消息） + - 输入框与发送按钮 + - 显示加载状态 + +- [ ] 集成到图谱页面 + - 在 `KnowledgeGraph.tsx` 中添加 Chat 组件 + - 配置布局（侧边或底部面板） + +- [ ] 增强交互 + - 支持快捷键发送（Enter） + - 显示会话历史切换（可选） + - 消息支持 Markdown 渲染 + +--- + +## 后续增强阶段（非 MVP） + +- 记忆与自省系统 +- 邮件自动化 +- 定时任务 +- CoT 推理增强 diff --git a/backend/go.mod b/backend/go.mod index 3a3008c..6c61f38 100644 --- a/backend/go.mod +++ b/backend/go.mod @@ -2,15 +2,20 @@ module knowledge-graph-backend go 1.26.1 +require ( + github.com/gin-contrib/cors v1.7.7 + github.com/gin-gonic/gin v1.12.0 + github.com/joho/godotenv v1.5.1 + github.com/neo4j/neo4j-go-driver/v5 v5.28.4 +) + require ( github.com/bytedance/gopkg v0.1.4 // indirect github.com/bytedance/sonic v1.15.0 // indirect github.com/bytedance/sonic/loader v0.5.1 // indirect github.com/cloudwego/base64x v0.1.6 // indirect github.com/gabriel-vasile/mimetype v1.4.13 // indirect - github.com/gin-contrib/cors v1.7.7 // indirect github.com/gin-contrib/sse v1.1.1 // indirect - github.com/gin-gonic/gin v1.12.0 // indirect github.com/go-playground/locales v0.14.1 // indirect github.com/go-playground/universal-translator v0.18.1 // indirect github.com/go-playground/validator/v10 v10.30.2 // indirect diff --git a/backend/go.sum b/backend/go.sum index a07a2c9..dfe49c3 100644 --- a/backend/go.sum +++ b/backend/go.sum @@ -7,6 +7,7 @@ github.com/bytedance/sonic/loader v0.5.1/go.mod h1:AR4NYCk5DdzZizZ5djGqQ92eEhCCc github.com/cloudwego/base64x v0.1.6 h1:t11wG9AECkCDk5fMSoxmufanudBtJ+/HemLstXDLI2M= github.com/cloudwego/base64x v0.1.6/go.mod h1:OFcloc187FXDaYHvrNIjxSe8ncn0OOM8gEHfghB2IPU= github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= github.com/gabriel-vasile/mimetype v1.4.13 h1:46nXokslUBsAJE/wMsp5gtO500a4F3Nkz9Ufpk2AcUM= github.com/gabriel-vasile/mimetype v1.4.13/go.mod h1:d+9Oxyo1wTzWdyVUPMmXFvp4F9tea18J8ufA774AB3s= @@ -16,6 +17,8 @@ github.com/gin-contrib/sse v1.1.1 h1:uGYpNwTacv5R68bSGMapo62iLTRa9l5zxGCps4hK6ko github.com/gin-contrib/sse v1.1.1/go.mod h1:QXzuVkA0YO7o/gun03UI1Q+FTI8ZV/n5t03kIQAI89s= github.com/gin-gonic/gin v1.12.0 h1:b3YAbrZtnf8N//yjKeU2+MQsh2mY5htkZidOM7O0wG8= github.com/gin-gonic/gin v1.12.0/go.mod h1:VxccKfsSllpKshkBWgVgRniFFAzFb9csfngsqANjnLc= +github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s= +github.com/go-playground/assert/v2 v2.2.0/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4= github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA= github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY= github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY= @@ -26,7 +29,11 @@ github.com/goccy/go-json v0.10.6 h1:p8HrPJzOakx/mn/bQtjgNjdTcN+/S6FcG2CTtQOrHVU= github.com/goccy/go-json v0.10.6/go.mod h1:oq7eo15ShAhp70Anwd5lgX2pLfOS3QCiwU/PULtXL6M= github.com/goccy/go-yaml v1.19.2 h1:PmFC1S6h8ljIz6gMRBopkjP1TVT7xuwrButHID66PoM= github.com/goccy/go-yaml v1.19.2/go.mod h1:XBurs7gK8ATbW4ZPGKgcbrY1Br56PdM69F7LkFRi1kA= +github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8= +github.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU= github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= +github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0= +github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4= github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM= github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo= github.com/klauspost/cpuid/v2 v2.3.0 h1:S4CRMLnYUhGeDFDqkGriYKdfoFlDnMtqTiI/sFzhA9Y= @@ -40,8 +47,11 @@ github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M= github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk= +github.com/neo4j/neo4j-go-driver/v5 v5.28.4 h1:7toxehVcYkZbyxV4W3Ib9VcnyRBQPucF+VwNNmtSXi4= +github.com/neo4j/neo4j-go-driver/v5 v5.28.4/go.mod h1:Vff8OwT7QpLm7L2yYr85XNWe9Rbqlbeb9asNXJTHO4k= github.com/pelletier/go-toml/v2 v2.3.0 h1:k59bC/lIZREW0/iVaQR8nDHxVq8OVlIzYCOJf421CaM= github.com/pelletier/go-toml/v2 v2.3.0/go.mod h1:2gIqNv+qfxSVS7cM2xJQKtLSTLUE9V8t9Stt+h56mCY= +github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= github.com/quic-go/qpack v0.6.0 h1:g7W+BMYynC1LbYLSqRt8PBg5Tgwxn214ZZR34VIOjz8= github.com/quic-go/qpack v0.6.0/go.mod h1:lUpLKChi8njB4ty2bFLX2x4gzDqXwUpaO1DP9qMDZII= @@ -56,12 +66,16 @@ github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/ github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo= github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= +github.com/stretchr/testify v1.11.1 h1:7s2iGBzp5EwR7/aIZr8ao5+dra3wiQyKjjFuvgVKu7U= +github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U= github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI= github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08= github.com/ugorji/go/codec v1.3.1 h1:waO7eEiFDwidsBN6agj1vJQ4AG7lh2yqXyOXqhgQuyY= github.com/ugorji/go/codec v1.3.1/go.mod h1:pRBVtBSKl77K30Bv8R2P+cLSGaTtex6fsA2Wjqmfxj4= go.mongodb.org/mongo-driver/v2 v2.5.0 h1:yXUhImUjjAInNcpTcAlPHiT7bIXhshCTL3jVBkF3xaE= go.mongodb.org/mongo-driver/v2 v2.5.0/go.mod h1:yOI9kBsufol30iFsl1slpdq1I0eHPzybRWdyYUs8K/0= +go.uber.org/mock v0.6.0 h1:hyF9dfmbgIX5EfOdasqLsWD6xqpNZlXblLB/Dbnwv3Y= +go.uber.org/mock v0.6.0/go.mod h1:KiVJ4BqZJaMj4svdfmHM0AUx4NJYO8ZNpPnZn1Z+BBU= golang.org/x/arch v0.25.0 h1:qnk6Ksugpi5Bz32947rkUgDt9/s5qvqDPl/gBKdMJLE= golang.org/x/arch v0.25.0/go.mod h1:0X+GdSIP+kL5wPmpK7sdkEVTt2XoYP0cSjQSbZBwOi8= golang.org/x/crypto v0.49.0 h1:+Ng2ULVvLHnJ/ZFEq4KdcDd/cfjrrjjNSXNzxg0Y4U4= @@ -77,4 +91,5 @@ google.golang.org/protobuf v1.36.11 h1:fV6ZwhNocDyBLK0dj+fg8ektcVegBBuEolpbTQyBN google.golang.org/protobuf v1.36.11/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j23XfzDpco= gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= diff --git a/backend/main.go b/backend/main.go index f7e3fe3..4d96106 100644 --- a/backend/main.go +++ b/backend/main.go @@ -1,4 +1,4 @@ -package main +package neo4j import ( "log" @@ -16,7 +16,7 @@ const ( DATA_FILE = "data.json" ) -func main() { +func test() { dataService, err := services.NewDataService(DATA_FILE) if err != nil { diff --git a/backend/neo4j/client.go b/backend/neo4j/client.go new file mode 100644 index 0000000..3b9c28c --- /dev/null +++ b/backend/neo4j/client.go @@ -0,0 +1,48 @@ +package neo4j + +import ( + "context" + "fmt" + "log" + "os" + + "github.com/joho/godotenv" + "github.com/neo4j/neo4j-go-driver/v5/neo4j" +) + +var ( + NEO4J_URI string + NEO4J_USERNAME string + NEO4J_PASSWORD string +) + +func init() { + err := godotenv.Load() + if err != nil { + log.Fatal("Error loading .env file") + } + + NEO4J_URI = os.Getenv("NEO4J_URI") + NEO4J_USERNAME = os.Getenv("NEO4J_USERNAME") + NEO4J_PASSWORD = os.Getenv("NEO4J_PASSWORD") +} + +func doConnect() { + ctx := context.Background() + // dbUri := "" + // dbUser := "" + // dbPassword := "" + driver, err := neo4j.NewDriverWithContext( + NEO4J_URI, + neo4j.BasicAuth(NEO4J_USERNAME, NEO4J_PASSWORD, "")) + if err != nil { + panic(err) + } + defer driver.Close(ctx) + + err = driver.VerifyConnectivity(ctx) + if err != nil { + panic(err) + } + fmt.Println("Connection established.") +} diff --git a/backend/neo4j/client_test.go b/backend/neo4j/client_test.go new file mode 100644 index 0000000..651b49d --- /dev/null +++ b/backend/neo4j/client_test.go @@ -0,0 +1,7 @@ +package neo4j + +import "testing" + +func TestDoConnect(t *testing.T){ + doConnect() +} \ No newline at end of file