package com.rongyichuang.config;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.CommandLineRunner;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.stereotype.Component;

@Component
public class DatabaseMigration implements CommandLineRunner {

    @Autowired
    private JdbcTemplate jdbcTemplate;

    @Override
    public void run(String... args) throws Exception {
        System.out.println("🔄 开始执行数据库迁移...");
        try {
            // 检查description字段是否已存在
            String checkSql = "SELECT COUNT(*) FROM INFORMATION_SCHEMA.COLUMNS " +
                             "WHERE TABLE_SCHEMA = 'ryc' AND TABLE_NAME = 't_employee' AND COLUMN_NAME = 'description'";
            
            Integer count = jdbcTemplate.queryForObject(checkSql, Integer.class);
            
            if (count == 0) {
                // 添加description字段到表末尾
                String alterSql = "ALTER TABLE t_employee ADD COLUMN description varchar(255)";
                jdbcTemplate.execute(alterSql);
                System.out.println("✅ 成功添加description字段到t_employee表");
            } else {
                System.out.println("ℹ️ description字段已存在，无需添加");
            }
            
        } catch (Exception e) {
            System.err.println("❌ 数据库迁移失败: " + e.getMessage());
            e.printStackTrace();
        }
    }
}