12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152 |
- package controllers
- import (
- "fmt"
- "xg_fetl/internal/db_executor"
- "xg_fetl/internal/global"
- "xg_fetl/internal/models"
- "xg_fetl/internal/services"
- )
- func ImportController(outputDir string) {
- dbType := global.Cfg.ImportConfig.DBConfig.DBType
- dsn := global.Cfg.ImportConfig.DBConfig.GetDSN()
- dbName := global.Cfg.ImportConfig.DBConfig.DBName
-
-
- delimiter := global.Cfg.ImportConfig.Delimiter
- pageSize := global.Cfg.ImportConfig.PageSize
-
- executor, err := db_executor.NewDBExecutor(dbType, dsn)
- if err != nil {
- fmt.Println(fmt.Errorf("create db executor failed, err: %v", err))
- panic(err)
- }
- defer executor.Close()
-
- dbInfo, err := services.GetDBInfoService(dbName, executor)
- if err != nil {
- fmt.Println(fmt.Errorf("get db info failed, err: %v", err))
- panic(err)
- }
-
- allTables := make(map[string]models.TableInfo, len(dbInfo.Tables)+len(dbInfo.LobTables))
-
- for tableName, tableInfo := range dbInfo.Tables {
- allTables[tableName] = tableInfo
- }
-
- for tableName, tableInfo := range dbInfo.LobTables {
- allTables[tableName] = tableInfo
- }
- services.ReaderMain(dbName, outputDir, allTables, delimiter, pageSize, executor)
- }
|