kafka CdCommandHandler 源码
kafka CdCommandHandler 代码
文件路径:/shell/src/main/java/org/apache/kafka/shell/CdCommandHandler.java
/*
* 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.shell;
import net.sourceforge.argparse4j.inf.ArgumentParser;
import net.sourceforge.argparse4j.inf.Namespace;
import org.apache.kafka.shell.MetadataNode.DirectoryNode;
import org.jline.reader.Candidate;
import java.io.PrintWriter;
import java.util.List;
import java.util.Optional;
import java.util.function.Consumer;
/**
* Implements the cd command.
*/
public final class CdCommandHandler implements Commands.Handler {
public final static Commands.Type TYPE = new CdCommandType();
public static class CdCommandType implements Commands.Type {
private CdCommandType() {
}
@Override
public String name() {
return "cd";
}
@Override
public String description() {
return "Set the current working directory.";
}
@Override
public boolean shellOnly() {
return true;
}
@Override
public void addArguments(ArgumentParser parser) {
parser.addArgument("target").
nargs("?").
help("The directory to change to.");
}
@Override
public Commands.Handler createHandler(Namespace namespace) {
return new CdCommandHandler(Optional.ofNullable(namespace.getString("target")));
}
@Override
public void completeNext(MetadataNodeManager nodeManager, List<String> nextWords,
List<Candidate> candidates) throws Exception {
if (nextWords.size() == 1) {
CommandUtils.completePath(nodeManager, nextWords.get(0), candidates);
}
}
}
private final Optional<String> target;
public CdCommandHandler(Optional<String> target) {
this.target = target;
}
@Override
public void run(Optional<InteractiveShell> shell,
PrintWriter writer,
MetadataNodeManager manager) throws Exception {
String effectiveTarget = target.orElse("/");
manager.visit(new Consumer<MetadataNodeManager.Data>() {
@Override
public void accept(MetadataNodeManager.Data data) {
new GlobVisitor(effectiveTarget, entryOption -> {
if (entryOption.isPresent()) {
if (!(entryOption.get().node() instanceof DirectoryNode)) {
writer.println("cd: " + effectiveTarget + ": not a directory.");
} else {
data.setWorkingDirectory(entryOption.get().absolutePath());
}
} else {
writer.println("cd: " + effectiveTarget + ": no such directory.");
}
}).accept(data);
}
});
}
@Override
public int hashCode() {
return target.hashCode();
}
@Override
public boolean equals(Object other) {
if (!(other instanceof CdCommandHandler)) return false;
CdCommandHandler o = (CdCommandHandler) other;
if (!o.target.equals(target)) return false;
return true;
}
}
相关信息
相关文章
0
赞