Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.openmetadata.service.jdbi3.QueryRepository Maven / Gradle / Ivy
package org.openmetadata.service.jdbi3;
import static org.openmetadata.common.utils.CommonUtil.listOrEmpty;
import static org.openmetadata.common.utils.CommonUtil.nullOrEmpty;
import static org.openmetadata.schema.type.EventType.ENTITY_FIELDS_CHANGED;
import static org.openmetadata.schema.type.EventType.ENTITY_UPDATED;
import static org.openmetadata.service.Entity.USER;
import java.util.*;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.UriInfo;
import lombok.SneakyThrows;
import org.jdbi.v3.sqlobject.transaction.Transaction;
import org.openmetadata.schema.entity.data.Query;
import org.openmetadata.schema.entity.services.DatabaseService;
import org.openmetadata.schema.entity.teams.User;
import org.openmetadata.schema.type.ChangeDescription;
import org.openmetadata.schema.type.ChangeEvent;
import org.openmetadata.schema.type.EntityReference;
import org.openmetadata.schema.type.FieldChange;
import org.openmetadata.schema.type.Include;
import org.openmetadata.schema.type.Relationship;
import org.openmetadata.service.Entity;
import org.openmetadata.service.resources.query.QueryResource;
import org.openmetadata.service.util.EntityUtil;
import org.openmetadata.service.util.FullyQualifiedName;
import org.openmetadata.service.util.JsonUtils;
import org.openmetadata.service.util.RestUtil;
public class QueryRepository extends EntityRepository {
private static final String QUERY_USED_IN_FIELD = "queryUsedIn";
private static final String QUERY_USERS_FIELD = "users";
private static final String QUERY_PATCH_FIELDS = "users,query,queryUsedIn,processedLineage";
private static final String QUERY_UPDATE_FIELDS = "users,queryUsedIn,processedLineage";
public QueryRepository() {
super(
QueryResource.COLLECTION_PATH,
Entity.QUERY,
Query.class,
Entity.getCollectionDAO().queryDAO(),
QUERY_PATCH_FIELDS,
QUERY_UPDATE_FIELDS);
supportsSearch = true;
}
@Override
public void setFullyQualifiedName(Query query) {
query.setFullyQualifiedName(
FullyQualifiedName.add(query.getService().getFullyQualifiedName(), query.getName()));
}
@Override
public void setFields(Query entity, EntityUtil.Fields fields) {
entity.setQueryUsedIn(
fields.contains(QUERY_USED_IN_FIELD) ? getQueryUsage(entity) : entity.getQueryUsedIn());
entity.withUsers(fields.contains("users") ? getQueryUsers(entity) : entity.getUsers());
}
@Override
public void clearFields(Query entity, EntityUtil.Fields fields) {
entity.withQueryUsedIn(fields.contains(QUERY_USED_IN_FIELD) ? entity.getQueryUsedIn() : null);
entity.withUsers(fields.contains("users") ? this.getQueryUsers(entity) : null);
}
public List getQueryUsage(Query queryEntity) {
return queryEntity == null
? Collections.emptyList()
: findFrom(queryEntity.getId(), Entity.QUERY, Relationship.MENTIONED_IN, null);
}
public List getQueryUsers(Query queryEntity) {
return queryEntity == null
? Collections.emptyList()
: findFrom(queryEntity.getId(), Entity.QUERY, Relationship.USES, USER);
}
@Override
@SneakyThrows
public void prepare(Query entity, boolean update) {
if (nullOrEmpty(entity.getName())) {
String checkSum = EntityUtil.hash(entity.getQuery());
entity.setChecksum(checkSum);
entity.setName(checkSum);
}
entity.setUsers(EntityUtil.populateEntityReferences(entity.getUsers()));
DatabaseService service = Entity.getEntity(entity.getService(), "", Include.ALL);
entity.setService(service.getEntityReference());
}
@Override
public void storeEntity(Query queryEntity, boolean update) {
List queryUsage = queryEntity.getQueryUsedIn();
List queryUsers = queryEntity.getUsers();
queryEntity.withQueryUsedIn(null).withUsers(null);
store(queryEntity, update);
// Restore relationships
queryEntity.withQueryUsedIn(queryUsage).withUsers(queryUsers);
}
@Override
public void storeRelationships(Query queryEntity) {
// Store Query Users Relation
if (queryEntity.getUsers() != null) {
for (EntityReference entityRef : queryEntity.getUsers()) {
addRelationship(
entityRef.getId(), queryEntity.getId(), USER, Entity.QUERY, Relationship.USES);
}
}
// Store Query Used in Relation
storeQueryUsedIn(queryEntity.getId(), queryEntity.getQueryUsedIn(), null);
// The service contains the query
addServiceRelationship(queryEntity, queryEntity.getService());
}
@Override
public EntityUpdater getUpdater(Query original, Query updated, Operation operation) {
return new QueryUpdater(original, updated, operation);
}
private void storeQueryUsedIn(
UUID queryId, List addQueryUsedIn, List deleteQueryUsedIn) {
for (EntityReference entityRef : listOrEmpty(addQueryUsedIn)) {
addRelationship(
entityRef.getId(), queryId, entityRef.getType(), Entity.QUERY, Relationship.MENTIONED_IN);
}
for (EntityReference entityRef : listOrEmpty(deleteQueryUsedIn)) {
deleteRelationship(
entityRef.getId(), entityRef.getType(), queryId, Entity.QUERY, Relationship.MENTIONED_IN);
}
}
public RestUtil.PutResponse addQueryUser(
UriInfo uriInfo, String updatedBy, UUID queryId, List userFqnList) {
Query query = Entity.getEntity(Entity.QUERY, queryId, QUERY_USERS_FIELD, Include.NON_DELETED);
List oldValue = query.getUsers();
for (String userFqn : userFqnList) {
User user = Entity.getEntityByName(USER, userFqn, "", Include.NON_DELETED);
EntityReference entityRef = user.getEntityReference();
addRelationship(
entityRef.getId(), queryId, entityRef.getType(), Entity.QUERY, Relationship.USES);
}
// Populate Fields
setFieldsInternal(query, new EntityUtil.Fields(allowedFields, QUERY_USERS_FIELD));
Entity.withHref(uriInfo, query.getUsers());
ChangeEvent changeEvent =
getQueryChangeEvent(
updatedBy, QUERY_USERS_FIELD, oldValue, query.getUsers(), withHref(uriInfo, query));
return new RestUtil.PutResponse<>(Response.Status.CREATED, changeEvent, ENTITY_FIELDS_CHANGED);
}
public RestUtil.PutResponse addQueryUsedBy(
UriInfo uriInfo, String updatedBy, UUID queryId, List userList) {
Query query = Entity.getEntity(Entity.QUERY, queryId, QUERY_UPDATE_FIELDS, Include.NON_DELETED);
Query oldQuery = JsonUtils.readValue(JsonUtils.pojoToJson(query), Query.class);
query.getUsedBy().addAll(userList);
ChangeEvent changeEvent =
getQueryChangeEvent(
updatedBy,
QUERY_USERS_FIELD,
oldQuery.getUsedBy(),
query.getUsers(),
withHref(uriInfo, query));
update(uriInfo, oldQuery, query);
return new RestUtil.PutResponse<>(Response.Status.CREATED, changeEvent, ENTITY_FIELDS_CHANGED);
}
public RestUtil.PutResponse addQueryUsage(
UriInfo uriInfo, String updatedBy, UUID queryId, List entityIds) {
Query query = Entity.getEntity(Entity.QUERY, queryId, QUERY_USED_IN_FIELD, Include.NON_DELETED);
List oldValue = query.getQueryUsedIn();
// Create Relationships
entityIds.forEach(
entityRef ->
addRelationship(
entityRef.getId(),
queryId,
entityRef.getType(),
Entity.QUERY,
Relationship.MENTIONED_IN));
// Populate Fields
setFieldsInternal(query, new EntityUtil.Fields(allowedFields, QUERY_USED_IN_FIELD));
Entity.withHref(uriInfo, query.getQueryUsedIn());
ChangeEvent changeEvent =
getQueryChangeEvent(
updatedBy,
QUERY_USED_IN_FIELD,
oldValue,
query.getQueryUsedIn(),
withHref(uriInfo, query));
return new RestUtil.PutResponse<>(Response.Status.CREATED, changeEvent, ENTITY_FIELDS_CHANGED);
}
public RestUtil.PutResponse removeQueryUsedIn(
UriInfo uriInfo, String updatedBy, UUID queryId, List entityIds) {
Query query = Entity.getEntity(Entity.QUERY, queryId, QUERY_USED_IN_FIELD, Include.NON_DELETED);
List oldValue = query.getQueryUsedIn();
for (EntityReference ref : entityIds) {
deleteRelationship(
ref.getId(), ref.getType(), queryId, Entity.QUERY, Relationship.MENTIONED_IN);
}
// Populate Fields
setFieldsInternal(query, new EntityUtil.Fields(allowedFields, QUERY_USED_IN_FIELD));
Entity.withHref(uriInfo, query.getQueryUsedIn());
ChangeEvent changeEvent =
getQueryChangeEvent(
updatedBy,
QUERY_USED_IN_FIELD,
oldValue,
query.getQueryUsedIn(),
withHref(uriInfo, query));
return new RestUtil.PutResponse<>(Response.Status.CREATED, changeEvent, ENTITY_FIELDS_CHANGED);
}
private ChangeEvent getQueryChangeEvent(
String updatedBy, String fieldUpdated, Object oldValue, Object newValue, Query updatedQuery) {
FieldChange fieldChange =
new FieldChange().withName(fieldUpdated).withNewValue(newValue).withOldValue(oldValue);
ChangeDescription change =
new ChangeDescription().withPreviousVersion(updatedQuery.getVersion());
change.getFieldsUpdated().add(fieldChange);
return new ChangeEvent()
.withId(UUID.randomUUID())
.withEntity(updatedQuery)
.withChangeDescription(change)
.withEventType(ENTITY_UPDATED)
.withEntityType(entityType)
.withEntityId(updatedQuery.getId())
.withEntityFullyQualifiedName(updatedQuery.getFullyQualifiedName())
.withUserName(updatedBy)
.withTimestamp(System.currentTimeMillis())
.withCurrentVersion(updatedQuery.getVersion())
.withPreviousVersion(updatedQuery.getVersion());
}
public class QueryUpdater extends EntityUpdater {
public QueryUpdater(Query original, Query updated, Operation operation) {
super(original, updated, operation);
}
@Transaction
@Override
public void entitySpecificUpdate() {
updateFromRelationships(
"users",
USER,
original.getUsers(),
updated.getUsers(),
Relationship.USES,
Entity.QUERY,
original.getId());
List added = new ArrayList<>();
List deleted = new ArrayList<>();
recordListChange(
"queryUsedIn",
original.getQueryUsedIn(),
updated.getQueryUsedIn(),
added,
deleted,
EntityUtil.entityReferenceMatch);
// Store processed Lineage
recordChange(
"processedLineage", original.getProcessedLineage(), updated.getProcessedLineage());
// Store Query Used in Relation
recordChange("usedBy", original.getUsedBy(), updated.getUsedBy(), true);
storeQueryUsedIn(updated.getId(), added, deleted);
// Query is a required field. Cannot be removed.
if (updated.getQuery() != null) {
String originalChecksum = EntityUtil.hash(original.getQuery());
String updatedChecksum = EntityUtil.hash(updated.getQuery());
if (!originalChecksum.equals(updatedChecksum)) {
updated.setChecksum(updatedChecksum);
recordChange("query", original.getQuery(), updated.getQuery());
recordChange("checksum", original.getChecksum(), updated.getChecksum());
}
}
}
}
}
