Skip to content

Commit

Permalink
Make PermissionTicket events marshallable
Browse files Browse the repository at this point in the history
Fixes #35328

Signed-off-by: Pedro Ruivo <[email protected]>
  • Loading branch information
pruivo authored and ahus1 committed Nov 28, 2024
1 parent fda6dde commit a65fd34
Show file tree
Hide file tree
Showing 6 changed files with 355 additions and 80 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,8 @@
import org.keycloak.keys.infinispan.PublicKeyStorageInvalidationEvent;
import org.keycloak.models.UserSessionModel;
import org.keycloak.models.cache.infinispan.ClearCacheEvent;
import org.keycloak.models.cache.infinispan.authorization.events.PermissionTicketRemovedEvent;
import org.keycloak.models.cache.infinispan.authorization.events.PermissionTicketUpdatedEvent;
import org.keycloak.models.cache.infinispan.authorization.events.PolicyRemovedEvent;
import org.keycloak.models.cache.infinispan.authorization.events.PolicyUpdatedEvent;
import org.keycloak.models.cache.infinispan.authorization.events.ResourceRemovedEvent;
Expand Down Expand Up @@ -136,6 +138,8 @@
ClearCacheEvent.class,

//models.cache.infinispan.authorization.events package
PermissionTicketRemovedEvent.class,
PermissionTicketUpdatedEvent.class,
PolicyUpdatedEvent.class,
PolicyRemovedEvent.class,
ResourceUpdatedEvent.class,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -160,6 +160,9 @@ private Marshalling() {
public static final int MAP_ENTRY_TO_KEY_FUNCTION = 65611;
public static final int SESSION_UNWRAP_MAPPER = 65612;

public static final int PERMISSION_TICKET_REMOVED_EVENT = 65613;
public static final int PERMISSION_TICKET_UPDATED_EVENT = 65614;

public static void configure(GlobalConfigurationBuilder builder) {
builder.serialization()
.addContextInitializer(KeycloakModelSchema.INSTANCE);
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,93 @@
/*
* Copyright 2024 Red Hat, Inc. and/or its affiliates
* and other contributors as indicated by the @author tags.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

package org.keycloak.models.cache.infinispan.authorization.events;

import java.util.Objects;

import org.infinispan.protostream.annotations.ProtoField;
import org.keycloak.models.cache.infinispan.events.InvalidationEvent;

abstract class BasePermissionTicketEvent extends InvalidationEvent implements AuthorizationCacheInvalidationEvent {

private final String owner;
private final String resource;
private final String scope;
private final String serverId;
private final String requester;
private final String resourceName;

BasePermissionTicketEvent(String id, String owner, String resource, String scope, String serverId, String requester, String resourceName) {
super(id);
this.owner = owner;
this.resource = resource;
this.scope = scope;
this.serverId = serverId;
this.requester = requester;
this.resourceName = resourceName;
}

@ProtoField(2)
public String getOwner() {
return owner;
}

@ProtoField(3)
public String getRequester() {
return requester;
}

@ProtoField(4)
public String getResource() {
return resource;
}

@ProtoField(5)
public String getResourceName() {
return resourceName;
}

@ProtoField(6)
public String getScope() {
return scope;
}

@ProtoField(7)
public String getServerId() {
return serverId;
}

@Override
public String toString() {
return "%s [ id=%s, name=%s]".formatted(getClass().getName(), getId(), resource);
}

@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
if (!super.equals(o)) return false;
BasePermissionTicketEvent that = (BasePermissionTicketEvent) o;
return Objects.equals(resource, that.resource) && Objects.equals(serverId, that.serverId);
}

@Override
public int hashCode() {
return Objects.hash(super.hashCode(), resource, serverId);
}

}
Original file line number Diff line number Diff line change
Expand Up @@ -17,60 +17,30 @@

package org.keycloak.models.cache.infinispan.authorization.events;

import java.util.Objects;
import java.util.Set;

import org.infinispan.protostream.annotations.ProtoFactory;
import org.infinispan.protostream.annotations.ProtoTypeId;
import org.keycloak.marshalling.Marshalling;
import org.keycloak.models.cache.infinispan.authorization.StoreFactoryCacheManager;
import org.keycloak.models.cache.infinispan.events.InvalidationEvent;

/**
* @author <a href="mailto:[email protected]">Marek Posolda</a>
*/
public class PermissionTicketRemovedEvent extends InvalidationEvent implements AuthorizationCacheInvalidationEvent {
@ProtoTypeId(Marshalling.PERMISSION_TICKET_REMOVED_EVENT)
public class PermissionTicketRemovedEvent extends BasePermissionTicketEvent {

private String owner;
private String resource;
private String scope;
private String serverId;
private String requester;
private String resourceName;

private PermissionTicketRemovedEvent(String id) {
super(id);
@ProtoFactory
PermissionTicketRemovedEvent(String id, String owner, String resource, String scope, String serverId, String requester, String resourceName) {
super(id, owner, resource, scope, serverId, requester, resourceName);
}

public static PermissionTicketRemovedEvent create(String id, String owner, String requester, String resource, String resourceName, String scope, String serverId) {
PermissionTicketRemovedEvent event = new PermissionTicketRemovedEvent(id);
event.owner = owner;
event.requester = requester;
event.resource = resource;
event.resourceName = resourceName;
event.scope = scope;
event.serverId = serverId;
return event;
}

@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
if (!super.equals(o)) return false;
PermissionTicketRemovedEvent that = (PermissionTicketRemovedEvent) o;
return Objects.equals(resource, that.resource) && Objects.equals(serverId, that.serverId);
}

@Override
public int hashCode() {
return Objects.hash(super.hashCode(), resource, serverId);
}

@Override
public String toString() {
return String.format("PermissionTicketRemovedEvent [ id=%s, name=%s]", getId(), resource);
return new PermissionTicketRemovedEvent(id, owner, resource, scope, serverId, requester, resourceName);
}

@Override
public void addInvalidations(StoreFactoryCacheManager cache, Set<String> invalidations) {
cache.permissionTicketRemoval(getId(), owner, requester, resource, resourceName, scope, serverId, invalidations);
cache.permissionTicketRemoval(getId(), getOwner(), getRequester(), getResource(), getResourceName(), getScope(), getServerId(), invalidations);
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -17,60 +17,30 @@

package org.keycloak.models.cache.infinispan.authorization.events;

import java.util.Objects;
import java.util.Set;

import org.infinispan.protostream.annotations.ProtoFactory;
import org.infinispan.protostream.annotations.ProtoTypeId;
import org.keycloak.marshalling.Marshalling;
import org.keycloak.models.cache.infinispan.authorization.StoreFactoryCacheManager;
import org.keycloak.models.cache.infinispan.events.InvalidationEvent;

/**
* @author <a href="mailto:[email protected]">Marek Posolda</a>
*/
public class PermissionTicketUpdatedEvent extends InvalidationEvent implements AuthorizationCacheInvalidationEvent {
@ProtoTypeId(Marshalling.PERMISSION_TICKET_UPDATED_EVENT)
public class PermissionTicketUpdatedEvent extends BasePermissionTicketEvent {

private String owner;
private String resource;
private String scope;
private String serverId;
private String requester;
private String resourceName;

private PermissionTicketUpdatedEvent(String id) {
super(id);
@ProtoFactory
PermissionTicketUpdatedEvent(String id, String owner, String resource, String scope, String serverId, String requester, String resourceName) {
super(id, owner, resource, scope, serverId, requester, resourceName);
}

public static PermissionTicketUpdatedEvent create(String id, String owner, String requester, String resource, String resourceName, String scope, String serverId) {
PermissionTicketUpdatedEvent event = new PermissionTicketUpdatedEvent(id);
event.owner = owner;
event.requester = requester;
event.resource = resource;
event.resourceName = resourceName;
event.scope = scope;
event.serverId = serverId;
return event;
}

@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
if (!super.equals(o)) return false;
PermissionTicketUpdatedEvent that = (PermissionTicketUpdatedEvent) o;
return Objects.equals(resource, that.resource) && Objects.equals(serverId, that.serverId);
}

@Override
public int hashCode() {
return Objects.hash(super.hashCode(), resource, serverId);
}

@Override
public String toString() {
return String.format("PermissionTicketUpdatedEvent [ id=%s, name=%s]", getId(), resource);
return new PermissionTicketUpdatedEvent(id, owner, resource, scope, serverId, requester, resourceName);
}

@Override
public void addInvalidations(StoreFactoryCacheManager cache, Set<String> invalidations) {
cache.permissionTicketUpdated(getId(), owner, requester, resource, resourceName, scope, serverId, invalidations);
cache.permissionTicketUpdated(getId(), getOwner(), getRequester(), getResource(), getResourceName(), getScope(), getServerId(), invalidations);
}
}
Loading

0 comments on commit a65fd34

Please sign in to comment.