|
|
@@ -0,0 +1,170 @@
|
|
|
+/**
|
|
|
+ * 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
|
|
|
+ * <p>
|
|
|
+ * http://www.apache.org/licenses/LICENSE-2.0
|
|
|
+ * <p>
|
|
|
+ * 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 com.google.common.base.Preconditions;
|
|
|
+import org.apache.hadoop.ozone.OzoneAcl;
|
|
|
+import org.apache.hadoop.ozone.protocol.proto.OzoneManagerProtocolProtos.PrefixInfo;
|
|
|
+import org.apache.hadoop.ozone.protocolPB.OMPBHelper;
|
|
|
+
|
|
|
+import java.util.HashMap;
|
|
|
+import java.util.LinkedList;
|
|
|
+import java.util.List;
|
|
|
+import java.util.Map;
|
|
|
+import java.util.Objects;
|
|
|
+import java.util.stream.Collectors;
|
|
|
+
|
|
|
+/**
|
|
|
+ * Wrapper class for Ozone prefix path info, currently mainly target for ACL but
|
|
|
+ * can be extended for other OzFS optimizations in future.
|
|
|
+ */
|
|
|
+// TODO: support Auditable interface
|
|
|
+public final class OmPrefixInfo extends WithMetadata {
|
|
|
+
|
|
|
+ private String name;
|
|
|
+ private List<OzoneAcl> acls;
|
|
|
+
|
|
|
+ public OmPrefixInfo(String name, List<OzoneAcl> acls,
|
|
|
+ Map<String, String> metadata) {
|
|
|
+ this.name = name;
|
|
|
+ this.acls = acls;
|
|
|
+ this.metadata = metadata;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * Returns the ACL's associated with this prefix.
|
|
|
+ * @return {@literal List<OzoneAcl>}
|
|
|
+ */
|
|
|
+ public List<OzoneAcl> getAcls() {
|
|
|
+ return acls;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * Returns the name of the prefix path.
|
|
|
+ * @return name of the prefix path.
|
|
|
+ */
|
|
|
+ public String getName() {
|
|
|
+ return name;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * Returns new builder class that builds a OmPrefixInfo.
|
|
|
+ *
|
|
|
+ * @return Builder
|
|
|
+ */
|
|
|
+ public static OmPrefixInfo.Builder newBuilder() {
|
|
|
+ return new OmPrefixInfo.Builder();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * Builder for OmPrefixInfo.
|
|
|
+ */
|
|
|
+ public static class Builder {
|
|
|
+ private String name;
|
|
|
+ private List<OzoneAcl> acls;
|
|
|
+ private Map<String, String> metadata;
|
|
|
+
|
|
|
+ public Builder() {
|
|
|
+ //Default values
|
|
|
+ this.acls = new LinkedList<>();
|
|
|
+ this.metadata = new HashMap<>();
|
|
|
+ }
|
|
|
+
|
|
|
+ public Builder setAcls(List<OzoneAcl> listOfAcls) {
|
|
|
+ this.acls = listOfAcls;
|
|
|
+ return this;
|
|
|
+ }
|
|
|
+
|
|
|
+ public Builder setName(String n) {
|
|
|
+ this.name = n;
|
|
|
+ return this;
|
|
|
+ }
|
|
|
+
|
|
|
+ public OmPrefixInfo.Builder addMetadata(String key, String value) {
|
|
|
+ metadata.put(key, value);
|
|
|
+ return this;
|
|
|
+ }
|
|
|
+
|
|
|
+ public OmPrefixInfo.Builder addAllMetadata(
|
|
|
+ Map<String, String> additionalMetadata) {
|
|
|
+ if (additionalMetadata != null) {
|
|
|
+ metadata.putAll(additionalMetadata);
|
|
|
+ }
|
|
|
+ return this;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * Constructs the OmPrefixInfo.
|
|
|
+ * @return instance of OmPrefixInfo.
|
|
|
+ */
|
|
|
+ public OmPrefixInfo build() {
|
|
|
+ Preconditions.checkNotNull(name);
|
|
|
+ Preconditions.checkNotNull(acls);
|
|
|
+ return new OmPrefixInfo(name, acls, metadata);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * Creates PrefixInfo protobuf from OmPrefixInfo.
|
|
|
+ */
|
|
|
+ public PrefixInfo getProtobuf() {
|
|
|
+ PrefixInfo.Builder pib = PrefixInfo.newBuilder().setName(name)
|
|
|
+ .addAllAcls(acls.stream().map(OMPBHelper::convertOzoneAcl)
|
|
|
+ .collect(Collectors.toList()))
|
|
|
+ .addAllMetadata(KeyValueUtil.toProtobuf(metadata));
|
|
|
+ return pib.build();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * Parses PrefixInfo protobuf and creates OmPrefixInfo.
|
|
|
+ * @param prefixInfo
|
|
|
+ * @return instance of OmPrefixInfo
|
|
|
+ */
|
|
|
+ public static OmPrefixInfo getFromProtobuf(PrefixInfo prefixInfo) {
|
|
|
+ OmPrefixInfo.Builder opib = OmPrefixInfo.newBuilder()
|
|
|
+ .setName(prefixInfo.getName())
|
|
|
+ .setAcls(prefixInfo.getAclsList().stream().map(
|
|
|
+ OMPBHelper::convertOzoneAcl).collect(Collectors.toList()));
|
|
|
+ if (prefixInfo.getMetadataList() != null) {
|
|
|
+ opib.addAllMetadata(KeyValueUtil
|
|
|
+ .getFromProtobuf(prefixInfo.getMetadataList()));
|
|
|
+ }
|
|
|
+ return opib.build();
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public boolean equals(Object o) {
|
|
|
+ if (this == o) {
|
|
|
+ return true;
|
|
|
+ }
|
|
|
+ if (o == null || getClass() != o.getClass()) {
|
|
|
+ return false;
|
|
|
+ }
|
|
|
+ OmPrefixInfo that = (OmPrefixInfo) o;
|
|
|
+ return name.equals(that.name) &&
|
|
|
+ Objects.equals(acls, that.acls) &&
|
|
|
+ Objects.equals(metadata, that.metadata);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public int hashCode() {
|
|
|
+ return Objects.hash(name);
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
Xiaoyu Yao