xichen01 commented on code in PR #8275: URL: https://github.com/apache/ozone/pull/8275#discussion_r2047297700
########## hadoop-ozone/common/src/main/java/org/apache/hadoop/ozone/om/helpers/OmLifecycleRuleAndOperator.java: ########## @@ -0,0 +1,119 @@ +/* + * 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.hadoop.ozone.om.helpers; + +import jakarta.annotation.Nonnull; +import jakarta.annotation.Nullable; +import java.util.HashMap; +import java.util.Map; +import org.apache.hadoop.ozone.om.exceptions.OMException; + +/** + * A class that encapsulates lifecycleRule andOperator. + */ +public final class OmLifecycleRuleAndOperator { + + private final Map<String, String> tags; + private final String prefix; + + private OmLifecycleRuleAndOperator(Map<String, String> tags, String prefix) { + this.tags = tags; + this.prefix = prefix; + } + + @Nonnull + public Map<String, String> getTags() { + return tags; + } + + @Nullable + public String getPrefix() { + return prefix; + } + + /** + * Validates the OmLifecycleRuleAndOperator. + * Ensures the following: + * - Either tags or prefix must be specified. + * - If there are tags and no prefix, the tags should be more than one. + * - Prefix alone is not allowed. + * + * @throws OMException if the validation fails. + */ + public void valid() throws OMException { + if ((tags == null || tags.isEmpty()) && (prefix == null || prefix.isEmpty())) { + throw new OMException("Invalid lifecycle rule andOperator configuration: " + + "Either 'Tags' or 'Prefix' must be specified.", + OMException.ResultCodes.INVALID_REQUEST); + } + + if (tags != null && !tags.isEmpty()) { + if (prefix == null || prefix.isEmpty()) { + if (tags.size() == 1) { Review Comment: Rewrote here, reduced duplicate code. ########## hadoop-ozone/common/src/main/java/org/apache/hadoop/ozone/om/helpers/OmLCRule.java: ########## @@ -0,0 +1,238 @@ +/* + * 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.hadoop.ozone.om.helpers; + +import java.util.ArrayList; +import java.util.List; +import org.apache.commons.lang3.RandomStringUtils; +import org.apache.commons.lang3.StringUtils; +import org.apache.hadoop.ozone.om.exceptions.OMException; + +/** + * A class that encapsulates lifecycle rule. + */ +public class OmLCRule { + + public static final int LC_ID_LENGTH = 48; + // Ref: https://docs.aws.amazon.com/AmazonS3/latest/userguide/intro-lifecycle-rules.html#intro-lifecycle-rule-id + public static final int LC_ID_MAX_LENGTH = 255; + + private String id; + private String prefix; + private boolean enabled; + private boolean isPrefixEnable; + private boolean isTagEnable; + // List of actions for this rule + private List<OmLCAction> actions; + private OmLCFilter filter; + + OmLCRule(String id, String prefix, boolean enabled, + List<OmLCAction> actions, OmLCFilter filter) { + this.id = id; + this.prefix = prefix; + this.enabled = enabled; + this.actions = actions; + this.filter = filter; + + if (StringUtils.isEmpty(this.id)) { + this.id = RandomStringUtils.randomAlphanumeric(LC_ID_LENGTH); + } + + OmLifecycleRuleAndOperator andOperator = filter != null ? filter.getAndOperator() : null; + + if (prefix != null || + (filter != null && filter.getPrefix() != null) || + (andOperator != null && andOperator.getPrefix() != null)) { + isPrefixEnable = true; + } + + if ((filter != null && filter.getTag() != null) || + (andOperator != null && !andOperator.getTags().isEmpty())) { + isTagEnable = true; + } + } + + public String getId() { + return id; + } + + public void setId(String id) { + this.id = id; + } + + public String getPrefix() { + return prefix; + } + + public void setPrefix(String prefix) { + this.prefix = prefix; + } + + public boolean isEnabled() { + return enabled; + } + + public void setEnabled(boolean enabled) { + this.enabled = enabled; + } + + public List<OmLCAction> getActions() { + return actions; + } + + public void setActions(List<OmLCAction> actions) { + this.actions = actions; + } + + /** + * Get the expiration action if present. + * + * @return the expiration action if present, null otherwise + */ + public OmLCExpiration getExpiration() { + if (actions == null || actions.isEmpty()) { + return null; + } + + for (OmLCAction action : actions) { + if (action instanceof OmLCExpiration) { + return (OmLCExpiration) action; + } + } + return null; + } + + public OmLCFilter getFilter() { + return filter; + } + + public boolean isPrefixEnable() { + return isPrefixEnable; + } + + public boolean isTagEnable() { + return isTagEnable; + } + + public void setFilter(OmLCFilter filter) { + this.filter = filter; + } + + /** + * Validates the lifecycle rule. + * - ID length should not exceed the allowed limit + * - At least one action must be specified + * - Filter and Prefix cannot be used together + * - Actions must be valid + * - Filter must be valid + * - There must be at most one Expiration action per rule + * + * @throws OMException if the validation fails + */ + public void valid() throws OMException { + if (id.length() > LC_ID_MAX_LENGTH) { + throw new OMException("ID length should not exceed allowed limit of " + LC_ID_MAX_LENGTH, + OMException.ResultCodes.INVALID_REQUEST); + } + + if (actions == null || actions.isEmpty()) { + throw new OMException("At least one action needs to be specified in a rule.", + OMException.ResultCodes.INVALID_REQUEST); + } + + // Check that there is at most one Expiration action + for (OmLCAction action : actions) { + if (action.getActionType() == OmLCAction.ActionType.EXPIRATION) { + if (actions.size() > 1) { Review Comment: Yes, This branch only checks if the number of EXPIRATION is 1 -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected] --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
