/*****************************************************************
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.apache.cayenne.query;
import java.util.HashMap;
import java.util.Map;
/**
* A wrapper for a QueryMetadata instance allowing that may override a subset of metadata
* properties.
*
* @since 1.2
*/
class QueryMetadataWrapper extends QueryMetadataProxy {
static final String CACHE_KEY_PROPERTY = "QueryMetadataWrapper.CacheKey";
Map<String, Object> overrides;
public QueryMetadataWrapper(QueryMetadata info) {
super(info);
}
/**
* Overrides a property with an alternative value. Property names are defined in the
* {@link QueryMetadata} interface.
*/
void override(String key, Object value) {
if (overrides == null) {
overrides = new HashMap<>();
}
overrides.put(key, value);
}
boolean overrideExists(String key) {
return overrides != null && overrides.containsKey(key);
}
public String getCacheKey() {
return (overrideExists(CACHE_KEY_PROPERTY))
? (String) overrides.get(CACHE_KEY_PROPERTY)
: super.getCacheKey();
}
/**
* @since 3.0
*/
public QueryCacheStrategy getCacheStrategy() {
return (overrideExists(QueryMetadata.CACHE_STRATEGY_PROPERTY))
? (QueryCacheStrategy) overrides.get(QueryMetadata.CACHE_STRATEGY_PROPERTY)
: super.getCacheStrategy();
}
/**
* @deprecated since 4.0, use {@link QueryMetadataWrapper#getCacheKey()}
*/
@Deprecated
public String[] getCacheGroups() {
return (overrideExists(QueryMetadata.CACHE_GROUPS_PROPERTY))
? (String[]) overrides.get(QueryMetadata.CACHE_GROUPS_PROPERTY)
: super.getCacheGroups();
}
/**
* @since 4.0
*/
public String getCacheGroup() {
if(overrideExists(QueryMetadata.CACHE_GROUPS_PROPERTY)) {
String[] cacheGroups = (String[]) overrides.get(QueryMetadata.CACHE_GROUPS_PROPERTY);
if(cacheGroups == null || cacheGroups.length == 0) {
return null;
} else {
return cacheGroups[0];
}
}
return super.getCacheGroup();
}
public boolean isFetchingDataRows() {
if (!overrideExists(QueryMetadata.FETCHING_DATA_ROWS_PROPERTY)) {
return super.isFetchingDataRows();
}
Boolean b = (Boolean) overrides.get(QueryMetadata.FETCHING_DATA_ROWS_PROPERTY);
return b != null && b;
}
public boolean isRefreshingObjects() {
return true;
}
public int getPageSize() {
if (!overrideExists(QueryMetadata.PAGE_SIZE_PROPERTY)) {
return super.getPageSize();
}
Number n = (Number) overrides.get(QueryMetadata.PAGE_SIZE_PROPERTY);
return n != null ? n.intValue() : 0;
}
public int getFetchLimit() {
if (!overrideExists(QueryMetadata.FETCH_LIMIT_PROPERTY)) {
return super.getFetchLimit();
}
Number n = (Number) overrides.get(QueryMetadata.FETCH_LIMIT_PROPERTY);
return n != null ? n.intValue() : 0;
}
public int getStatementFetchSize() {
if (!overrideExists(QueryMetadata.STATEMENT_FETCH_SIZE_PROPERTY)) {
return super.getPageSize();
}
Number n = (Number) overrides.get(QueryMetadata.STATEMENT_FETCH_SIZE_PROPERTY);
return n != null ? n.intValue() : 0;
}
}