Function ListDatabaseColumnsV2

api/handler/v2/database.go:185–231

@Id ListDatabaseColumnsV2 @Description list columns of database source instance. @Tags database @Security ApiKeyAuth @Param host query string true "database host" @Param port query int true "database port" @Param user query string true "database user" @Param password query string true "database pass

(c echo.Context)

// @Success 200 {object} models.ListColumnsRespV2
// @Router /v2/database/columns [get]
func ListDatabaseColumnsV2(c echo.Context) error {
	logger := handler.NewLogger().Named("ListDatabaseColumnsV2")
	reqParam := new(models.ListColumnsReqV2)
	err := handler.BindAndValidate(logger, c, reqParam)
	if err != nil {
		return c.JSON(http.StatusInternalServerError, models.BuildBaseResp(err))
	}

	columns := make([]string, 0)
	switch reqParam.DatabaseType {
	case DB_TYPE_MYSQL:
		columns, err = listMySQLColumns(logger, reqParam)
		if err != nil {
			return c.JSON(http.StatusInternalServerError, models.BuildBaseResp(fmt.Errorf("list %s columns failed : %v", DB_TYPE_MYSQL, err)))
		}
	case DB_TYPE_ORACLE:
		if reqParam.IsPasswordEncrypted && reqParam.Password != "" {
			realPwd, err := handler.DecryptPassword(reqParam.Password, g.RsaPrivateKey)
			if nil != err {
				return c.JSON(http.StatusOK, &models.ConnectionRespV2{
					BaseResp: models.BuildBaseResp(err),
				})
			}
			reqParam.Password = realPwd
		}
		oracleDb, err := config.NewDB(context.TODO(), &config.OracleConfig{
			User:        reqParam.User,
			Password:    reqParam.Password,
			Host:        reqParam.Host,
			Port:        reqParam.Port,
			ServiceName: reqParam.ServiceName,
		})
		if err != nil {
			return c.JSON(http.StatusInternalServerError, models.BuildBaseResp(fmt.Errorf("list %s columns failed : %v", DB_TYPE_ORACLE, err)))
		}
		defer oracleDb.Close()
		columns, err = oracleDb.GetColumns(reqParam.Schema, reqParam.Table)
		if err != nil {
			return c.JSON(http.StatusInternalServerError, models.BuildBaseResp(fmt.Errorf("list %s columns failed : %v", DB_TYPE_ORACLE, err)))
		}
	}

	return c.JSON(http.StatusOK, &models.ListColumnsRespV2{
		Columns:  columns,
		BaseResp: models.BuildBaseResp(nil),
	})
}

func listMySQLColumns(logger hclog.Logger, reqParam *models.ListColumnsReqV2) ([]string, error) {
	uri, err := buildMysqlUri(reqParam.Host, reqParam.User, reqParam.Password,