potiuk commented on code in PR #40916:
URL: https://github.com/apache/airflow/pull/40916#discussion_r1686885216
##########
airflow/api_internal/internal_api_call.py:
##########
@@ -42,67 +42,57 @@
class InternalApiConfig:
"""Stores and caches configuration for Internal API."""
- _initialized = False
_use_internal_api = False
_internal_api_endpoint = ""
@staticmethod
- def force_database_direct_access(message: str):
+ def set_use_database_access(component: str):
"""
Block current component from using Internal API.
All methods decorated with internal_api_call will always be executed
locally.`
This mode is needed for "trusted" components like Scheduler,
Webserver, Internal Api server
"""
- InternalApiConfig._initialized = True
InternalApiConfig._use_internal_api = False
- if _ENABLE_AIP_44:
- logger.info("Forcing database direct access. %s", message)
+ if not _ENABLE_AIP_44:
+ raise RuntimeError("The AIP_44 is not enabled so you cannot use
it. ")
+ logger.info(
+ "DB isolation mode. But this is a trusted component and DB
connection is set. "
+ "Using database direct access when running %s.",
+ component,
+ )
@staticmethod
- def force_api_access(api_endpoint: str):
- """
- Force using Internal API with provided endpoint.
+ def set_use_internal_api(component: str):
+ if not _ENABLE_AIP_44:
+ raise RuntimeError("The AIP_44 is not enabled so you cannot use
it. ")
+ internal_api_endpoint = ""
+ internal_api_url = conf.get("core", "internal_api_url")
+ internal_api_endpoint = internal_api_url + "/internal_api/v1/rpcapi"
+ if not internal_api_endpoint.startswith("http://";):
+ raise AirflowConfigException("[core]internal_api_url must start
with http://";)
+ url_conf = urlparse(conf.get("core", "internal_api_url"))
+ api_path = url_conf.path
+ if api_path in ["", "/"]:
+ # Add the default path if not given in the configuration
+ api_path = "/internal_api/v1/rpcapi"
+ if url_conf.scheme not in ["http", "https"]:
+ raise AirflowConfigException("[core]internal_api_url must start
with http:// or https://";)
Review Comment:
Yeah. Bad merge.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
