/*
* 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.kafka.common.requests;
import org.apache.kafka.clients.admin.AccessControlEntry;
import org.apache.kafka.clients.admin.AclBinding;
import org.apache.kafka.clients.admin.Resource;
import org.apache.kafka.common.protocol.ApiKeys;
import org.apache.kafka.common.protocol.types.Struct;
import org.apache.kafka.common.utils.Utils;
import java.nio.ByteBuffer;
import java.util.ArrayList;
import java.util.List;
public class CreateAclsRequest extends AbstractRequest {
private final static String CREATIONS = "creations";
public static class AclCreation {
private final AclBinding acl;
public AclCreation(AclBinding acl) {
this.acl = acl;
}
static AclCreation fromStruct(Struct struct) {
Resource resource = RequestUtils.resourceFromStructFields(struct);
AccessControlEntry entry = RequestUtils.aceFromStructFields(struct);
return new AclCreation(new AclBinding(resource, entry));
}
public AclBinding acl() {
return acl;
}
void setStructFields(Struct struct) {
RequestUtils.resourceSetStructFields(acl.resource(), struct);
RequestUtils.aceSetStructFields(acl.entry(), struct);
}
@Override
public String toString() {
return "(acl=" + acl + ")";
}
}
public static class Builder extends AbstractRequest.Builder<CreateAclsRequest> {
private final List<AclCreation> creations;
public Builder(List<AclCreation> creations) {
super(ApiKeys.CREATE_ACLS);
this.creations = creations;
}
@Override
public CreateAclsRequest build(short version) {
return new CreateAclsRequest(version, creations);
}
@Override
public String toString() {
return "(type=CreateAclsRequest, creations=" + Utils.join(creations, ", ") + ")";
}
}
private final List<AclCreation> aclCreations;
CreateAclsRequest(short version, List<AclCreation> aclCreations) {
super(version);
this.aclCreations = aclCreations;
}
public CreateAclsRequest(Struct struct, short version) {
super(version);
this.aclCreations = new ArrayList<>();
for (Object creationStructObj : struct.getArray(CREATIONS)) {
Struct creationStruct = (Struct) creationStructObj;
aclCreations.add(AclCreation.fromStruct(creationStruct));
}
}
@Override
protected Struct toStruct() {
Struct struct = new Struct(ApiKeys.CREATE_ACLS.requestSchema(version()));
List<Struct> requests = new ArrayList<>();
for (AclCreation creation : aclCreations) {
Struct creationStruct = struct.instance(CREATIONS);
creation.setStructFields(creationStruct);
requests.add(creationStruct);
}
struct.set(CREATIONS, requests.toArray());
return struct;
}
public List<AclCreation> aclCreations() {
return aclCreations;
}
@Override
public AbstractResponse getErrorResponse(int throttleTimeMs, Throwable throwable) {
short versionId = version();
switch (versionId) {
case 0:
List<CreateAclsResponse.AclCreationResponse> responses = new ArrayList<>();
for (int i = 0; i < aclCreations.size(); i++) {
responses.add(new CreateAclsResponse.AclCreationResponse(throwable));
}
return new CreateAclsResponse(throttleTimeMs, responses);
default:
throw new IllegalArgumentException(String.format("Version %d is not valid. Valid versions for %s are 0 to %d",
versionId, this.getClass().getSimpleName(), ApiKeys.CREATE_ACLS.latestVersion()));
}
}
public static CreateAclsRequest parse(ByteBuffer buffer, short version) {
return new CreateAclsRequest(ApiKeys.CREATE_ACLS.parseRequest(version, buffer), version);
}
}