Refine search
private boolean poolHasChildren(MWMResourcePlan resourcePlan, String poolPath) { boolean commited = false; Query query = null; try { openTransaction(); query = pm.newQuery(MWMPool.class, "resourcePlan == rp && path.startsWith(poolPath)"); query.declareParameters("MWMResourcePlan rp, java.lang.String poolPath"); query.setResult("count(this)"); query.setUnique(true); Long count = (Long) query.execute(resourcePlan, poolPath + "."); commited = commitTransaction(); return count != null && count > 0; } finally { rollbackAndCleanup(commited, query); } }
Map<String, Object> params = new HashMap<>(); String queryFilterString = makeQueryFilterString(catName, dbName, null, filter, params); query = pm.newQuery(MTable.class); query.declareImports("import java.lang.String"); query.setResult("tableName"); query.setResultClass(java.lang.String.class); if (maxTables >= 0) { query.setRange(0, maxTables); query.declareParameters(parameterDeclaration); query.setFilter(queryFilterString); Collection<String> names = (Collection<String>)query.executeWithMap(params);
private List<MPartition> listMPartitions(String catName, String dbName, String tableName, int max, QueryWrapper queryWrapper) { boolean success = false; List<MPartition> mparts = null; try { openTransaction(); LOG.debug("Executing listMPartitions"); dbName = normalizeIdentifier(dbName); tableName = normalizeIdentifier(tableName); Query query = queryWrapper.query = pm.newQuery(MPartition.class, "table.tableName == t1 && table.database.name == t2 && table.database.catalogName == t3"); query.declareParameters("java.lang.String t1, java.lang.String t2, java.lang.String t3"); query.setOrdering("partitionName ascending"); if (max >= 0) { query.setRange(0, max); } mparts = (List<MPartition>) query.execute(tableName, dbName, catName); LOG.debug("Done executing query for listMPartitions"); pm.retrieveAll(mparts); success = commitTransaction(); LOG.debug("Done retrieving all objects for listMPartitions {}", mparts); } finally { if (!success) { rollbackTransaction(); } } return mparts; }
private MDelegationToken getTokenFrom(String tokenId) { Query query = pm.newQuery(MDelegationToken.class, "tokenIdentifier == tokenId"); query.declareParameters("java.lang.String tokenId"); query.setUnique(true); MDelegationToken delegationToken = (MDelegationToken) query.execute(tokenId); if (query != null) { query.closeAll(); } return delegationToken; }
private Query createActivePlanQuery() { openTransaction(); Query query = pm.newQuery(MWMResourcePlan.class, "status == activeStatus && ns == nsname"); query.declareParameters("java.lang.String activeStatus, java.lang.String nsname"); query.setUnique(true); return query; }
@SuppressWarnings("unchecked") private List<User> findUsers( String searchField, String searchKey, boolean ascendingUsername ) { PersistenceManager pm = getPersistenceManager(); Transaction tx = pm.currentTransaction(); try { tx.begin(); Extent extent = pm.getExtent( JdoUser.class, true ); Query query = pm.newQuery( extent ); String ordering = ascendingUsername ? "username ascending" : "username descending"; query.setOrdering( ordering ); query.declareImports( "import java.lang.String" ); query.declareParameters( "String searchKey" ); query.setFilter( "this." + searchField + ".toLowerCase().indexOf(searchKey.toLowerCase()) > -1" ); List<User> result = (List<User>) query.execute( searchKey ); result = (List<User>) pm.detachCopyAll( result ); tx.commit(); return result; } finally { rollback( tx ); } }
@SuppressWarnings("unchecked") public List<MRoleMap> listMRoleMembers(String roleName) { boolean success = false; Query query = null; List<MRoleMap> mRoleMemeberList = new ArrayList<>(); try { LOG.debug("Executing listRoleMembers"); openTransaction(); query = pm.newQuery(MRoleMap.class, "role.roleName == t1"); query.declareParameters("java.lang.String t1"); query.setUnique(false); List<MRoleMap> mRoles = (List<MRoleMap>) query.execute(roleName); pm.retrieveAll(mRoles); success = commitTransaction(); mRoleMemeberList.addAll(mRoles); LOG.debug("Done retrieving all objects for listRoleMembers"); } finally { rollbackAndCleanup(success, query); } return mRoleMemeberList; }
query = pm.newQuery(MTableColumnStatistics.class); String filter; String parameters; query.setFilter(filter); query.declareParameters(parameters); if (colName != null) { query.setUnique(true); mStatsObj = (MTableColumnStatistics) query.executeWithArray(normalizeIdentifier(tableName), normalizeIdentifier(dbName), normalizeIdentifier(catName), normalizeIdentifier(colName)); pm.retrieve(mStatsObj); pm.deletePersistent(mStatsObj); } else { throw new NoSuchObjectException("Column stats doesn't exist for db=" + dbName + " table=" (List<MTableColumnStatistics>) query.execute( normalizeIdentifier(tableName), normalizeIdentifier(dbName),
/** * Get the MAuthzPathsMapping object from authzObj */ private MAuthzPathsMapping getMAuthzPathsMappingCore(PersistenceManager pm, long authzSnapshotID, String authzObj) { Query query = pm.newQuery(MAuthzPathsMapping.class); query.setFilter("this.authzSnapshotID == authzSnapshotID && this.authzObjName == authzObjName"); query.declareParameters("long authzSnapshotID, java.lang.String authzObjName"); query.setUnique(true); return (MAuthzPathsMapping) query.execute(authzSnapshotID, authzObj); }
public List<BuildResult> getAllBuildsForAProjectByDate( int projectId ) { PersistenceManager pm = getPersistenceManager(); Transaction tx = pm.currentTransaction(); try { tx.begin(); Query query = pm.newQuery( "SELECT FROM " + BuildResult.class.getName() + " WHERE project.id == projectId PARAMETERS int projectId ORDER BY endTime DESC" ); query.declareImports( "import java.lang.Integer" ); query.declareParameters( "Integer projectId" ); List result = (List) query.execute( new Integer( projectId ) ); result = (List) pm.detachCopyAll( result ); tx.commit(); return result; } finally { rollback( tx ); } }
String dbName = normalizeIdentifier(schemaName.getDbName()); String catName = normalizeIdentifier(schemaName.getCatName()); query = pm.newQuery(MSchemaVersion.class, "iSchema.name == schemaName && iSchema.db.name == dbName && iSchema.db.catalogName == cat"); query.declareParameters("java.lang.String schemaName, java.lang.String dbName, " + "java.lang.String cat"); query.setUnique(true); query.setOrdering("version descending"); query.setRange(0, 1); MSchemaVersion mSchemaVersion = (MSchemaVersion)query.execute(name, dbName, catName); pm.retrieve(mSchemaVersion); if (mSchemaVersion != null) { pm.retrieveAll(mSchemaVersion.getCols()); if (mSchemaVersion.getSerDe() != null) { pm.retrieve(mSchemaVersion.getSerDe());
lowered_tbl_names.add(normalizeIdentifier(t)); query = pm.newQuery(MTable.class); query.setFilter("database.name == db && database.catalogName == cat && tbl_names.contains(tableName)"); query.declareParameters("java.lang.String db, java.lang.String cat, java.util.Collection tbl_names"); Collection mtables = (Collection) query.execute(db, catName, lowered_tbl_names); if (mtables == null || mtables.isEmpty()) { dbExistsQuery = pm.newQuery(MDatabase.class, "name == db && catalogName == cat"); dbExistsQuery.declareParameters("java.lang.String db, java.lang.String cat"); dbExistsQuery.setUnique(true); dbExistsQuery.setResult("name"); String dbNameIfExists = (String) dbExistsQuery.execute(db, catName); if (org.apache.commons.lang.StringUtils.isEmpty(dbNameIfExists)) { throw new UnknownDBException("Could not find database " + rollbackAndCleanup(committed, query); if (dbExistsQuery != null) { dbExistsQuery.closeAll();
query = pm.newQuery("select constraintName from org.apache.hadoop.hive.metastore.model.MConstraint where " + "((parentTable.tableName == ptblname && parentTable.database.name == pdbname && " + "parentTable.database.catalogName == pcatname) || " "childTable.database.name == cdbname && childTable.database.catalogName == ccatname)) " + (constraintname != null ? " && constraintName == constraintname" : "")); query.declareParameters("java.lang.String ptblname, java.lang.String pdbname," + "java.lang.String pcatname, java.lang.String ctblname, java.lang.String cdbname," + "java.lang.String ccatname" + constraintname != null ? ((Collection<?>) query. executeWithArray(tableName, dbName, catName, tableName, dbName, catName, constraintname)): ((Collection<?>) query. executeWithArray(tableName, dbName, catName, tableName, dbName, catName)); for (Iterator<?> i = constraintNamesColl.iterator(); i.hasNext();) { String currName = (String) i.next(); constraintNames.add(currName); query = pm.newQuery(MConstraint.class); query.setFilter("param.contains(constraintName)"); query.declareParameters("java.util.Collection param"); Collection<?> constraints = (Collection<?>)query.execute(constraintNames); mConstraints = new ArrayList<>(); for (Iterator<?> i = constraints.iterator(); i.hasNext();) { query.closeAll();
Query query = pm.newQuery( "select partitionName from org.apache.hadoop.hive.metastore.model.MPartition " + "where " + queryFilterString); query.setRange(0, maxParts); query.declareParameters(parameterDeclaration); if (ascending) { query.setOrdering("partitionName ascending"); } else { query.setOrdering("partitionName descending"); query.setResult("partitionName"); Collection<String> names = (Collection<String>) query.executeWithMap(params); partNames = new ArrayList<String>(names); success = commitTransaction(); LOG.debug("Done retrieving all objects for getPartitionNamesByFilter, size: {}", partNames.size()); query.closeAll(); } finally { if (!success) {
return null; Query query = pm.newQuery(MPartition.class, jdoFilter); if (maxParts >= 0) { query.setRange(0, maxParts); query.declareParameters(parameterDeclaration); query.setOrdering("partitionName ascending"); @SuppressWarnings("unchecked") List<MPartition> mparts = (List<MPartition>) query.executeWithMap(params); LOG.debug("Done executing query for getPartitionsViaOrmFilter"); pm.retrieveAll(mparts); // TODO: why is this inconsistent with what we get by names? LOG.debug("Done retrieving all objects for getPartitionsViaOrmFilter"); List<Partition> results = convertToParts(mparts); query.closeAll(); return results;
private List<RuntimeStat> getMRuntimeStats(int maxEntries, int maxCreateTime) { Query<MRuntimeStat> query = pm.newQuery(MRuntimeStat.class); query.setOrdering("createTime descending"); if (maxCreateTime > 0) { query.setFilter("createTime < "+maxCreateTime); } if (maxEntries < 0) { maxEntries = Integer.MAX_VALUE; } List<RuntimeStat> ret = new ArrayList<>(); List<MRuntimeStat> res = (List<MRuntimeStat>) query.execute(); int totalEntries = 0; for (MRuntimeStat mRuntimeStat : res) { pm.retrieve(mRuntimeStat); totalEntries += mRuntimeStat.getWeight(); ret.add(MRuntimeStat.toThrift(mRuntimeStat)); if (totalEntries >= maxEntries) { break; } } return ret; }
private MSchemaVersion getMSchemaVersion(String catName, String dbName, String schemaName, int version) { Query query = null; try { dbName = normalizeIdentifier(dbName); schemaName = normalizeIdentifier(schemaName); query = pm.newQuery(MSchemaVersion.class, "iSchema.name == schemaName && iSchema.db.name == dbName &&" + "iSchema.db.catalogName == cat && version == schemaVersion"); query.declareParameters( "java.lang.String schemaName, java.lang.String dbName," + "java.lang.String cat, java.lang.Integer schemaVersion"); query.setUnique(true); MSchemaVersion mSchemaVersion = (MSchemaVersion)query.executeWithArray(schemaName, dbName, catName, version); pm.retrieve(mSchemaVersion); if (mSchemaVersion != null) { pm.retrieveAll(mSchemaVersion.getCols()); if (mSchemaVersion.getSerDe() != null) { pm.retrieve(mSchemaVersion.getSerDe()); } } return mSchemaVersion; } finally { if (query != null) { query.closeAll(); } } }
Query query = queryWrapper.query = pm.newQuery(MPartition.class); query.setFilter(filter); query.declareParameters(makeParameterDeclarationString(params)); if (max_parts >= 0) { query.setRange(0, max_parts); query.setResult(resultsCol); return (Collection) query.executeWithMap(params);
query = pm.newQuery(MNotificationLog.class, filterBuilder.toString()); query.declareParameters(parameterBuilder.toString()); query.setOrdering("eventId ascending"); int maxEventResponse = MetastoreConf.getIntVar(conf, ConfVars.METASTORE_MAX_EVENT_RESPONSE); int maxEvents = (rqst.getMaxEvents() < maxEventResponse && rqst.getMaxEvents() > 0) ? rqst.getMaxEvents() : maxEventResponse; query.setRange(0, maxEvents); Collection<MNotificationLog> events = (Collection) query.executeWithArray(parameterVals.toArray(new Object[parameterVals.size()])); commited = commitTransaction(); if (events == null) {
LOG.warn("The table does not have the same column definition as its partition."); Query query = queryWrapper.query = pm.newQuery(MPartitionColumnStatistics.class); String paramStr = "java.lang.String t1, java.lang.String t2, java.lang.String t3"; String filter = "tableName == t1 && dbName == t2 && catName == t3 && ("; query.setFilter(filter); query.declareParameters(paramStr); query.setOrdering("partitionName ascending"); @SuppressWarnings("unchecked") List<MPartitionColumnStatistics> result = (List<MPartitionColumnStatistics>) query.executeWithArray(params); pm.retrieveAll(result); committed = commitTransaction(); return result;