feat: Add support for Vertica analytical database (#1538)

2025-09-13 13:10:29 +00:00 · 2024-05-20 09:45:40 +08:00
parent 6fb3d33bf4
commit 8d8411fcd3
16 changed files with 658 additions and 5 deletions
--- a/docker/examples/sqls/case_1_student_manager_vertica.sql
+++ b/docker/examples/sqls/case_1_student_manager_vertica.sql
@@ -0,0 +1,78 @@
+CREATE SCHEMA case_1_student_manager;
+COMMENT ON SCHEMA case_1_student_manager is '学校管理系统';
+
+SET SEARCH_PATH = case_1_student_manager;
+
+CREATE TABLE students (
+    student_id INT PRIMARY KEY,
+    student_name VARCHAR(100),
+    major VARCHAR(100),
+    year_of_enrollment INT,
+    student_age INT
+);
+COMMENT ON TABLE students IS '学生信息表';
+COMMENT ON COLUMN students.student_name IS '学生姓名';
+COMMENT ON COLUMN students.major IS '专业';
+COMMENT ON COLUMN students.year_of_enrollment IS '入学年份';
+COMMENT ON COLUMN students.student_age IS '学生年龄';
+
+CREATE TABLE courses (
+    course_id INT PRIMARY KEY,
+    course_name VARCHAR(100),
+    credit FLOAT
+);
+COMMENT ON TABLE courses IS '课程信息表';
+COMMENT ON COLUMN courses.course_name IS '课程名称';
+COMMENT ON COLUMN courses.credit IS '学分';
+
+CREATE TABLE scores (
+    student_id INT,
+    course_id INT,
+    score INT,
+    semester VARCHAR(50),
+    PRIMARY KEY (student_id, course_id),
+    FOREIGN KEY (student_id) REFERENCES students(student_id),
+    FOREIGN KEY (course_id) REFERENCES courses(course_id)
+);
+COMMENT ON TABLE scores IS '学生成绩表';
+COMMENT ON COLUMN scores.score IS '得分';
+COMMENT ON COLUMN scores.semester IS '学期';
+
+
+INSERT INTO students (student_id, student_name, major, year_of_enrollment, student_age) VALUES
+(1, '张三', '计算机科学', 2020, 20),
+(2, '李四', '计算机科学', 2021, 19),
+(3, '王五', '物理学', 2020, 21),
+(4, '赵六', '数学', 2021, 19),
+(5, '周七', '计算机科学', 2022, 18),
+(6, '吴八', '物理学', 2020, 21),
+(7, '郑九', '数学', 2021, 19),
+(8, '孙十', '计算机科学', 2022, 18),
+(9, '刘十一', '物理学', 2020, 21),
+(10, '陈十二', '数学', 2021, 19);
+
+INSERT INTO courses (course_id, course_name, credit) VALUES
+(1, '计算机基础', 3),
+(2, '数据结构', 4),
+(3, '高等物理', 3),
+(4, '线性代数', 4),
+(5, '微积分', 5),
+(6, '编程语言', 4),
+(7, '量子力学', 3),
+(8, '概率论', 4),
+(9, '数据库系统', 4),
+(10, '计算机网络', 4);
+
+INSERT INTO scores (student_id, course_id, score, semester) VALUES
+(1, 1, 90, '2020年秋季'),
+(1, 2, 85, '2021年春季'),
+(2, 1, 88, '2021年秋季'),
+(2, 2, 90, '2022年春季'),
+(3, 3, 92, '2020年秋季'),
+(3, 4, 85, '2021年春季'),
+(4, 3, 88, '2021年秋季'),
+(4, 4, 86, '2022年春季'),
+(5, 1, 90, '2022年秋季'),
+(5, 2, 87, '2023年春季');
+
+COMMIT;
--- a/docker/examples/sqls/case_2_ecom_vertica.sql
+++ b/docker/examples/sqls/case_2_ecom_vertica.sql
@@ -0,0 +1,78 @@
+CREATE SCHEMA case_2_ecom;
+COMMENT ON SCHEMA case_2_ecom is '电子商务系统';
+
+SET SEARCH_PATH = case_2_ecom;
+
+CREATE TABLE users (
+    user_id INT PRIMARY KEY,
+    user_name VARCHAR(100),
+    user_email VARCHAR(100),
+    registration_date DATE,
+    user_country VARCHAR(100)
+);
+COMMENT ON TABLE users IS '用户信息表';
+COMMENT ON COLUMN users.user_name IS '用户名';
+COMMENT ON COLUMN users.user_email IS '用户邮箱';
+COMMENT ON COLUMN users.registration_date IS '注册日期';
+COMMENT ON COLUMN users.user_country  IS '用户国家';
+
+CREATE TABLE products (
+    product_id INT PRIMARY KEY,
+    product_name VARCHAR(100),
+    product_price FLOAT
+);
+COMMENT ON TABLE products IS '商品信息表';
+COMMENT ON COLUMN products.product_name IS '商品名称';
+COMMENT ON COLUMN products.product_price IS '商品价格';
+
+CREATE TABLE orders (
+    order_id INT PRIMARY KEY,
+    user_id INT,
+    product_id INT,
+    quantity INT,
+    order_date DATE,
+    FOREIGN KEY (user_id) REFERENCES users(user_id),
+    FOREIGN KEY (product_id) REFERENCES products(product_id)
+);
+COMMENT ON TABLE orders IS '订单信息表';
+COMMENT ON COLUMN orders.quantity IS '数量';
+COMMENT ON COLUMN orders.order_date IS '订单日期';
+
+
+INSERT INTO users (user_id, user_name, user_email, registration_date, user_country) VALUES
+(1, 'John', 'john@gmail.com', '2020-01-01', 'USA'),
+(2, 'Mary', 'mary@gmail.com', '2021-01-01', 'UK'),
+(3, 'Bob', 'bob@gmail.com', '2020-01-01', 'USA'),
+(4, 'Alice', 'alice@gmail.com', '2021-01-01', 'UK'),
+(5, 'Charlie', 'charlie@gmail.com', '2020-01-01', 'USA'),
+(6, 'David', 'david@gmail.com', '2021-01-01', 'UK'),
+(7, 'Eve', 'eve@gmail.com', '2020-01-01', 'USA'),
+(8, 'Frank', 'frank@gmail.com', '2021-01-01', 'UK'),
+(9, 'Grace', 'grace@gmail.com', '2020-01-01', 'USA'),
+(10, 'Helen', 'helen@gmail.com', '2021-01-01', 'UK');
+
+INSERT INTO products (product_id, product_name, product_price) VALUES
+(1, 'iPhone', 699),
+(2, 'Samsung Galaxy', 599),
+(3, 'iPad', 329),
+(4, 'Macbook', 1299),
+(5, 'Apple Watch', 399),
+(6, 'AirPods', 159),
+(7, 'Echo', 99),
+(8, 'Kindle', 89),
+(9, 'Fire TV Stick', 39),
+(10, 'Echo Dot', 49);
+
+INSERT INTO orders (order_id, user_id, product_id, quantity, order_date) VALUES
+(1, 1, 1, 1, '2022-01-01'),
+(2, 1, 2, 1, '2022-02-01'),
+(3, 2, 3, 2, '2022-03-01'),
+(4, 2, 4, 1, '2022-04-01'),
+(5, 3, 5, 2, '2022-05-01'),
+(6, 3, 6, 3, '2022-06-01'),
+(7, 4, 7, 2, '2022-07-01'),
+(8, 4, 8, 1, '2022-08-01'),
+(9, 5, 9, 2, '2022-09-01'),
+(10, 5, 10, 3, '2022-10-01');
+
+COMMIT;
--- a/docker/examples/sqls/test_case_info_vertica.sql
+++ b/docker/examples/sqls/test_case_info_vertica.sql
@@ -0,0 +1,29 @@
+CREATE SCHEMA test_case_info;
+COMMENT ON SCHEMA test_case_info is '测试用例信息';
+
+SET SEARCH_PATH = test_case_info;
+
+CREATE TABLE test_cases (
+    case_id SERIAL /*INT AUTO_INCREMENT*/ PRIMARY KEY,
+    scenario_name VARCHAR(100),
+    scenario_description VARCHAR(6500),
+    test_question VARCHAR(500),
+    expected_sql VARCHAR(6500),
+    correct_output VARCHAR(6500)
+);
+COMMENT ON TABLE test_cases IS '测试用例表';
+COMMENT ON COLUMN test_cases.scenario_name IS '场景名称';
+COMMENT ON COLUMN test_cases.scenario_description IS '场景描述';
+COMMENT ON COLUMN test_cases.test_question IS '测试问题';
+COMMENT ON COLUMN test_cases.expected_sql IS '预期SQL';
+COMMENT ON COLUMN test_cases.correct_output IS '正确输出';
+
+INSERT INTO test_cases (scenario_name, scenario_description, test_question, expected_sql, correct_output) VALUES
+('学校管理系统', '测试SQL助手的联合查询，条件查询和排序功能', '查询所有学生的姓名，专业和成绩，按成绩降序排序', 'SELECT students.student_name, students.major, scores.score FROM students JOIN scores ON students.student_id = scores.student_id ORDER BY scores.score DESC;', '返回所有学生的姓名，专业和成绩，按成绩降序排序的结果'),
+('学校管理系统', '测试SQL助手的联合查询，条件查询和排序功能', '查询计算机科学专业的学生的平均成绩', 'SELECT AVG(scores.score) as avg_score FROM students JOIN scores ON students.student_id = scores.student_id WHERE students.major = ''计算机科学'';', '返回计算机科学专业学生的平均成绩'),
+('学校管理系统', '测试SQL助手的联合查询，条件查询和排序功能', '查询哪些学生在2023年秋季学期的课程学分总和超过15', 'SELECT students.student_name FROM students JOIN scores ON students.student_id = scores.student_id JOIN courses ON scores.course_id = courses.course_id WHERE scores.semester = ''2023年秋季'' GROUP BY students.student_id HAVING SUM(courses.credit) > 15;', '返回在2023年秋季学期的课程学分总和超过15的学生的姓名'),
+('电商系统', '测试SQL助手的数据聚合和分组功能', '查询每个用户的总订单数量', 'SELECT users.user_name, COUNT(orders.order_id) as order_count FROM users JOIN orders ON users.user_id = orders.user_id GROUP BY users.user_id;', '返回每个用户的总订单数量'),
+('电商系统', '测试SQL助手的数据聚合和分组功能', '查询每种商品的总销售额', 'SELECT products.product_name, SUM(products.product_price * orders.quantity) as total_sales FROM products JOIN orders ON products.product_id = orders.product_id GROUP BY products.product_id;', '返回每种商品的总销售额'),
+('电商系统', '测试SQL助手的数据聚合和分组功能', '查询2023年最受欢迎的商品（订单数量最多的商品）', 'SELECT products.product_name FROM products JOIN orders ON products.product_id = orders.product_id WHERE YEAR(orders.order_date) = 2023 GROUP BY products.product_id ORDER BY COUNT(orders.order_id) DESC LIMIT 1;', '返回2023年最受欢迎的商品（订单数量最多的商品）的名称');
+
+COMMIT;