Skip to content

fix: Logger settings for feature servers and updated logger for permission flow #4531

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
tokoko merged 3 commits into from
Sep 19, 2024
Merged

fix: Logger settings for feature servers and updated logger for permission flow #4531

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
71267e5
Logger setting for feature servers
redhatHameed Sep 17, 2024
1b5a8fc
added/updated logger for permission flow
redhatHameed Sep 18, 2024
286a6d7
set defualt logger level for feature servers as WARNING
redhatHameed Sep 18, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
set defualt logger level for feature servers as WARNING 
Signed-off-by: Abdul Hameed <ahameed@redhat.com>
  • Loading branch information
@redhatHameed
redhatHameed committed Sep 19, 2024
commit 286a6d7f79283290bfe26db5425a2191e60d56b5
2 changes: 1 addition & 1 deletion infra/charts/feast-feature-server/values.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ image:
# image.tag -- The Docker image tag (can be overwritten if custom feature server deps are needed for on demand transforms)
tag: 0.40.0

logLevel: "DEBUG" # Set log level
logLevel: "WARNING" # Set log level DEBUG, INFO, WARNING, ERROR, and CRITICAL (case-insensitive)

imagePullSecrets: []
nameOverride: ""
Expand Down
1 change: 0 additions & 1 deletion sdk/python/feast/permissions/auth/oidc_token_parser.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,6 @@
from feast.permissions.user import User

logger = logging.getLogger(__name__)
logger.setLevel(logging.INFO)


class OidcTokenParser(TokenParser):
Expand Down
5 changes: 2 additions & 3 deletions sdk/python/feast/permissions/server/arrow.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,6 @@
from feast.permissions.user import User

logger = logging.getLogger(__name__)
logger.setLevel(logging.INFO)


class AuthorizationMiddlewareFactory(fl.ServerMiddlewareFactory):
Expand Down Expand Up @@ -49,7 +48,7 @@ def __init__(self, access_token: str, *args, **kwargs):

def call_completed(self, exception):
if exception:
logger.error(
logger.exception(
f"{AuthorizationMiddleware.__name__} encountered an exception: {exception}"
Copy link
Copy Markdown
Contributor

@lokeshrangineni lokeshrangineni Sep 19, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The better way to log exception is logging.exception I m not sure what it logs if we do the {exception}.

Copy link
Copy Markdown
Contributor Author

@redhatHameed redhatHameed Sep 19, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

set as logging.exception. These will arrow flight sever related internal exceptions.

)

Expand Down Expand Up @@ -78,7 +77,7 @@ def inject_user_details(context: ServerCallContext):
if sm is not None:
auth_middleware = cast(AuthorizationMiddleware, context.get_middleware("auth"))
current_user = asyncio.run(auth_middleware.extract_user())
logger.info(f"User extracted: {current_user}")
logger.debug(f"User extracted: {current_user}")

sm.set_current_user(current_user)

Expand Down
7 changes: 4 additions & 3 deletions sdk/python/feast/permissions/server/grpc.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@
from feast.permissions.security_manager import get_security_manager

logger = logging.getLogger(__name__)
logger.setLevel(logging.INFO)


class AuthInterceptor(grpc.ServerInterceptor):
Expand All @@ -22,11 +21,13 @@ def intercept_service(self, continuation, handler_call_details):
metadata=dict(handler_call_details.invocation_metadata)
)

logger.info(f"Fetching user details for token of length: {len(access_token)}")
logger.debug(
f"Fetching user details for token of length: {len(access_token)}"
Copy link
Copy Markdown
Contributor

@lokeshrangineni lokeshrangineni Sep 19, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

my 2 cents - may be debug log here is more appropriate.

Copy link
Copy Markdown
Contributor Author

@redhatHameed redhatHameed Sep 19, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

set as debug

)
current_user = asyncio.run(
auth_manager.token_parser.user_details_from_access_token(access_token)
)
logger.info(f"User is: {current_user.username}")
logger.debug(f"User is: {current_user}")
sm.set_current_user(current_user)

return continuation(handler_call_details)
1 change: 0 additions & 1 deletion sdk/python/feast/permissions/server/utils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,6 @@
from feast.permissions.server.rest_token_extractor import RestTokenExtractor

logger = logging.getLogger(__name__)
logger.setLevel(logging.INFO)


class ServerType(enum.Enum):
Expand Down