yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-09-25 08:50:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merging if statements (#7851)

Browse Source
This commit is contained in:
André Oliveira 2022-10-01 19:51:39 +01:00 committed by GitHub
parent 03f1d4f2fb
commit 736637a598
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
9 changed files with 64 additions and 82 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
openmetadata-service/src/main/java/org/openmetadata/service/events/AbstractEventPublisher.java
View File

@ -50,10 +50,8 @@ public abstract class AbstractEventPublisher implements EventPublisher {
throws Exception {
// Ignore events that don't match the webhook event filters
ChangeEvent changeEvent = changeEventHolder.get();
if (!filter.isEmpty()) {
if (!FilterUtil.shouldProcessRequest(changeEvent, filter)) {
return;
}
if (!filter.isEmpty() && !FilterUtil.shouldProcessRequest(changeEvent, filter)) {
return;
}
// Batch until either the batch has ended or batch size has reached the max size

18
openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/CollectionDAO.java
View File

@ -967,10 +967,8 @@ public interface CollectionDAO {
return listAnnouncementsByEntityLinkBefore(
fqnPrefix, toType, limit, before, type, relation, mysqlCondition, postgresCondition);
}
if (userName != null) {
if (filterType == FilterType.MENTIONS) {
filterRelation = Relationship.MENTIONED_IN.ordinal();
}
if (userName != null && filterType == FilterType.MENTIONS) {
filterRelation = Relationship.MENTIONED_IN.ordinal();
}
return listThreadsByEntityLinkBefore(
fqnPrefix, toType, limit, before, type, status, resolved, relation, userName, teamNames, filterRelation);
@ -1058,10 +1056,8 @@ public interface CollectionDAO {
return listAnnouncementsByEntityLinkAfter(
fqnPrefix, toType, limit, after, type, relation, mysqlCondition, postgresCondition);
}
if (userName != null) {
if (filterType == FilterType.MENTIONS) {
filterRelation = Relationship.MENTIONED_IN.ordinal();
}
if (userName != null && filterType == FilterType.MENTIONS) {
filterRelation = Relationship.MENTIONED_IN.ordinal();
}
return listThreadsByEntityLinkAfter(
fqnPrefix, toType, limit, after, type, status, resolved, relation, userName, teamNames, filterRelation);
@ -1146,10 +1142,8 @@ public interface CollectionDAO {
}
return listCountAnnouncementsByEntityLink(fqnPrefix, toType, type, relation, mysqlCondition, postgresCondition);
}
if (userName != null) {
if (filterType == FilterType.MENTIONS) {
filterRelation = Relationship.MENTIONED_IN.ordinal();
}
if (userName != null && filterType == FilterType.MENTIONS) {
filterRelation = Relationship.MENTIONED_IN.ordinal();
}
return listCountThreadsByEntityLink(
fqnPrefix, toType, type, status, resolved, relation, userName, teamNames, filterRelation);

7
openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/EntityRepository.java
View File

@ -1221,11 +1221,10 @@ public abstract class EntityRepository<T extends EntityInterface> {
private void updateOwner() throws JsonProcessingException {
EntityReference origOwner = original.getOwner();
EntityReference updatedOwner = updated.getOwner();
if (operation.isPatch() || updatedOwner != null) {
if ((operation.isPatch() || updatedOwner != null)
&& recordChange(FIELD_OWNER, origOwner, updatedOwner, true, entityReferenceMatch)) {
// Update owner for all PATCH operations. For PUT operations, ownership can't be removed
if (recordChange(FIELD_OWNER, origOwner, updatedOwner, true, entityReferenceMatch)) {
EntityRepository.this.updateOwner(original, origOwner, updatedOwner);
}
EntityRepository.this.updateOwner(original, origOwner, updatedOwner);
}
}

20
openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/TableRepository.java
View File

@ -1053,22 +1053,22 @@ public class TableRepository extends EntityRepository<Table> {
private void updateColumnPrecision(Column origColumn, Column updatedColumn) throws JsonProcessingException {
String columnField = getColumnField(original, origColumn, "precision");
boolean updated = recordChange(columnField, origColumn.getPrecision(), updatedColumn.getPrecision());
if (origColumn.getPrecision() != null) { // Previously precision was set
if (updated && updatedColumn.getPrecision() < origColumn.getPrecision()) {
// The precision was reduced. Treat it as backward-incompatible change
majorVersionChange = true;
}
if (origColumn.getPrecision() != null
&& updated
&& updatedColumn.getPrecision() < origColumn.getPrecision()) { // Previously precision was set
// The precision was reduced. Treat it as backward-incompatible change
majorVersionChange = true;
}
}
private void updateColumnScale(Column origColumn, Column updatedColumn) throws JsonProcessingException {
String columnField = getColumnField(original, origColumn, "scale");
boolean updated = recordChange(columnField, origColumn.getScale(), updatedColumn.getScale());
if (origColumn.getScale() != null) { // Previously scale was set
if (updated && updatedColumn.getScale() < origColumn.getScale()) {
// The scale was reduced. Treat it as backward-incompatible change
majorVersionChange = true;
}
if (origColumn.getScale() != null
&& updated
&& updatedColumn.getScale() < origColumn.getScale()) { // Previously scale was set
// The scale was reduced. Treat it as backward-incompatible change
majorVersionChange = true;
}
}
}

8
openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/UserRepository.java
View File

@ -349,10 +349,10 @@ public class UserRepository extends EntityRepository<User> {
AuthenticationMechanism updatedAuthMechanism = updated.getAuthenticationMechanism();
if (origAuthMechanism == null && updatedAuthMechanism != null) {
recordChange("authenticationMechanism", original.getAuthenticationMechanism(), "new-encrypted-value");
} else if (origAuthMechanism != null && updatedAuthMechanism != null) {
if (!JsonUtils.areEquals(origAuthMechanism, updatedAuthMechanism)) {
recordChange("authenticationMechanism", "old-encrypted-value", "new-encrypted-value");
}
} else if (origAuthMechanism != null
&& updatedAuthMechanism != null
&& !JsonUtils.areEquals(origAuthMechanism, updatedAuthMechanism)) {
recordChange("authenticationMechanism", "old-encrypted-value", "new-encrypted-value");
}
}
}

24
openmetadata-service/src/main/java/org/openmetadata/service/resources/teams/UserResource.java
View File

@ -426,11 +426,9 @@ public class UserResource extends EntityResource<User, UserRepository> {
.withUsername(securityContext.getUserPrincipal().getName())
.withToken(request.getToken())
.withLogoutTime(logoutTime));
if (isBasicAuth()) {
if (isBasicAuth() && request.getRefreshToken() != null) {
// need to clear the refresh token as well
if (request.getRefreshToken() != null) {
tokenRepository.deleteToken(request.getRefreshToken());
}
tokenRepository.deleteToken(request.getRefreshToken());
}
return Response.status(200).entity("Logout Successful").build();
}
@ -1441,16 +1439,16 @@ public class UserResource extends EntityResource<User, UserRepository> {
String botName = create.getBotName();
EntityInterface bot = retrieveBot(botName);
// check if the bot user exists
if (!botHasRelationshipWithUser(bot, original)) {
if (!botHasRelationshipWithUser(bot, original)
&& original != null
&& userHasRelationshipWithAnyBot(original, bot)) {
// throw an exception if user already has a relationship with a bot
if (original != null && userHasRelationshipWithAnyBot(original, bot)) {
List<CollectionDAO.EntityRelationshipRecord> userBotRelationship = retrieveBotRelationshipsFor(original);
bot =
Entity.getEntityRepository(Entity.BOT)
.get(null, userBotRelationship.stream().findFirst().orElseThrow().getId(), Fields.EMPTY_FIELDS);
throw new IllegalArgumentException(
String.format("Bot user [%s] is already used by [%s] bot.", user.getName(), bot.getName()));
}
List<CollectionDAO.EntityRelationshipRecord> userBotRelationship = retrieveBotRelationshipsFor(original);
bot =
Entity.getEntityRepository(Entity.BOT)
.get(null, userBotRelationship.stream().findFirst().orElseThrow().getId(), Fields.EMPTY_FIELDS);
throw new IllegalArgumentException(
String.format("Bot user [%s] is already used by [%s] bot.", user.getName(), bot.getName()));
}
addAuthMechanismToBot(user, create, uriInfo);
RestUtil.PutResponse<User> response = dao.createOrUpdate(uriInfo, user);

8
openmetadata-service/src/main/java/org/openmetadata/service/security/JwtFilter.java
View File

@ -201,11 +201,9 @@ public class JwtFilter implements ContainerRequestFilter {
}
// validate principal domain
if (enforcePrincipalDomain) {
if (!domain.equals(principalDomain)) {
throw new AuthenticationException(
String.format("Not Authorized! Email does not match the principal domain %s", principalDomain));
}
if (enforcePrincipalDomain && !domain.equals(principalDomain)) {
throw new AuthenticationException(
String.format("Not Authorized! Email does not match the principal domain %s", principalDomain));
}
return userName;
}

24
openmetadata-service/src/main/java/org/openmetadata/service/security/policyevaluator/CompiledRule.java
View File

@ -132,11 +132,9 @@ public class CompiledRule extends Rule {
Iterator<MetadataOperation> iterator = operationContext.getOperations().listIterator();
while (iterator.hasNext()) {
MetadataOperation operation = iterator.next();
if (matchOperation(operation)) {
if (matchExpression(policyContext, subjectContext, resourceContext)) {
LOG.info("operation {} allowed", operation);
iterator.remove();
}
if (matchOperation(operation) && matchExpression(policyContext, subjectContext, resourceContext)) {
LOG.info("operation {} allowed", operation);
iterator.remove();
}
}
}
@ -154,15 +152,13 @@ public class CompiledRule extends Rule {
Access access = getAccess();
// Walk through all the operations in the rule and set permissions
for (Permission permission : resourcePermission.getPermissions()) {
if (matchOperation(permission.getOperation())) {
if (overrideAccess(access, permission.getAccess())) {
permission
.withAccess(access)
.withRole(policyContext.getRoleName())
.withPolicy(policyContext.getPolicyName())
.withRule(this);
LOG.debug("Updated permission {}", permission);
}
if (matchOperation(permission.getOperation()) && overrideAccess(access, permission.getAccess())) {
permission
.withAccess(access)
.withRole(policyContext.getRoleName())
.withPolicy(policyContext.getPolicyName())
.withRule(this);
LOG.debug("Updated permission {}", permission);
}
}
}

31
openmetadata-service/src/main/java/org/openmetadata/service/util/ElasticSearchClientUtils.java
View File

@ -62,22 +62,21 @@ public final class ElasticSearchClientUtils {
private static SSLContext createSSLContext(ElasticSearchConfiguration elasticSearchConfiguration)
throws KeyStoreException {
if (elasticSearchConfiguration.getScheme().equals("https")) {
if (elasticSearchConfiguration.getTruststorePath() != null
&& !elasticSearchConfiguration.getTruststorePath().isEmpty()) {
Path trustStorePath = Paths.get(elasticSearchConfiguration.getTruststorePath());
KeyStore truststore = KeyStore.getInstance("jks");
try (InputStream is = Files.newInputStream(trustStorePath)) {
truststore.load(is, elasticSearchConfiguration.getTruststorePassword().toCharArray());
SSLContextBuilder sslBuilder = SSLContexts.custom().loadTrustMaterial(truststore, null);
return sslBuilder.build();
} catch (IOException
| NoSuchAlgorithmException
| CertificateException
| KeyStoreException
| KeyManagementException e) {
throw new RuntimeException("Failed to crete SSLContext to for ElasticSearch Client", e);
}
if (elasticSearchConfiguration.getScheme().equals("https")
&& elasticSearchConfiguration.getTruststorePath() != null
&& !elasticSearchConfiguration.getTruststorePath().isEmpty()) {
Path trustStorePath = Paths.get(elasticSearchConfiguration.getTruststorePath());
KeyStore truststore = KeyStore.getInstance("jks");
try (InputStream is = Files.newInputStream(trustStorePath)) {
truststore.load(is, elasticSearchConfiguration.getTruststorePassword().toCharArray());
SSLContextBuilder sslBuilder = SSLContexts.custom().loadTrustMaterial(truststore, null);
return sslBuilder.build();
} catch (IOException
| NoSuchAlgorithmException
| CertificateException
| KeyStoreException
| KeyManagementException e) {
throw new RuntimeException("Failed to crete SSLContext to for ElasticSearch Client", e);
}
}
return null;