forked from KNEMC/KLALB
227 lines
7.6 KiB
Java
227 lines
7.6 KiB
Java
package org.kne.cloud.network.klalb;
|
||
|
||
import java.io.File;
|
||
import java.io.IOException;
|
||
import java.net.InetSocketAddress;
|
||
import java.nio.channels.ServerSocketChannel;
|
||
import java.util.ArrayList;
|
||
import java.util.Collections;
|
||
import java.util.Iterator;
|
||
import java.util.List;
|
||
import java.util.NoSuchElementException;
|
||
import java.util.Scanner;
|
||
|
||
import org.kne.cloud.network.MultipurposeSocketAddress;
|
||
import org.kne.cloud.network.SocketChannelListener;
|
||
import org.kne.cloud.network.ipv6.IPv6NetworkLink;
|
||
import org.kne.cloud.network.ipv6.RouteItem;
|
||
import org.kne.cloud.network.klalb.ui.KLALBStateGUI3;
|
||
import org.kne.cloud.network.klalb.ui.UIEnv;
|
||
import org.kne.cloud.network.perf.Kperf;
|
||
import org.kne.cloud.network.perf.NodeBenchmark;
|
||
import org.kne.debug.TimeDebugger;
|
||
import org.kne.membandboost.MembandBenchmark;
|
||
import org.kne.membandboost.ObjectPoolBandwidthTest;
|
||
|
||
public class KLALBMain {
|
||
public static KLALBStateGUI3 ksg;
|
||
public static void main(String[] args) throws IOException {
|
||
try {
|
||
UIEnv.inituie();
|
||
}catch(Throwable e) {
|
||
e.printStackTrace();
|
||
}
|
||
System.out.println(CONST.klalb+" V"+CONST.klalbver);
|
||
Scanner scn=new Scanner(System.in);
|
||
TimeDebugger dtb=new TimeDebugger();
|
||
dtb.putTime("Start");
|
||
File configJson=new File("klalb-config.json");
|
||
|
||
KLALBProxySystem kpcje=new KLALBProxySystem();
|
||
dtb.putTime("Create");
|
||
kpcje.loadConfigJson(configJson);
|
||
dtb.putTime("Load");
|
||
System.out.println("=".repeat(65));
|
||
System.out.println(" - IPv6 Address / End SID: "+kpcje.getKlalbController().getSelf().getAddress());
|
||
System.out.println("=".repeat(65));
|
||
try {
|
||
if(!kpcje.getControllerConfig().isNogui())
|
||
openGUI(kpcje);
|
||
}catch(Throwable e) {
|
||
e.printStackTrace();
|
||
}
|
||
dtb.putTime("UI");
|
||
//dtb.print();
|
||
/*MultipurposeSocketAddress mpa=new MultipurposeSocketAddress("127.9.9.9", 49573);
|
||
kpcje.enableRemoteManagement(mpa);
|
||
System.out.println("远程管理端口已在"+mpa+"端口上开启");*/
|
||
/*if(true)
|
||
return;*/
|
||
ServerSocketChannel kpsvr=KLALBVirtualServerSocketChannel.open(kpcje.getKlalbController());
|
||
kpsvr.bind(new InetSocketAddress("::0", 4564));
|
||
SocketChannelListener stlr=new SocketChannelListener(kpsvr);
|
||
stlr.setCon((scl)->{
|
||
try {
|
||
new Kperf(new StreamChannelKLALBPacketLink(scl,false)).startPerfing();
|
||
} catch (IOException e) {
|
||
e.printStackTrace();
|
||
}
|
||
});
|
||
System.out.println("Done!");
|
||
while(true) {
|
||
System.out.print("KLALB>");
|
||
try {
|
||
String s=scn.nextLine();
|
||
String tri=s.trim();
|
||
if(tri.isEmpty()) {
|
||
continue;
|
||
}
|
||
String[]sc=tri.split(" ");
|
||
switch(sc[0]) {
|
||
case "?":
|
||
case "help":
|
||
System.out.println(" help / ?: see help");
|
||
System.out.println(" monitor: show monitor GUI");
|
||
System.out.println(" links-state: query link states");
|
||
System.out.println(" links-add <addr:port>: add link");
|
||
System.out.println(" links-remove <addr:port>: remove link");
|
||
System.out.println(" links-reconnect: reconnect all link");
|
||
System.out.println(" kltp-state: query kltp states");
|
||
System.out.println(" route: display internal route table");
|
||
System.out.println(" kperf <addr:port>: performance benchmark");
|
||
System.out.println(" exit: Exit");
|
||
break;
|
||
|
||
case "monitor":
|
||
try {
|
||
openGUI(kpcje);
|
||
}catch(Throwable e) {
|
||
e.printStackTrace();
|
||
}
|
||
break;
|
||
case "links-state":
|
||
System.out.println("links state：");
|
||
//System.out.println("状态\t上传流量\t下载流量\t上传速度\t下载速度\t上传延迟\t下载延迟\t上传抖动\t下载抖动");
|
||
System.out.println("state\tspd↑\tspd↓\tPPS↑\tPPS↓\tdelay↑\tdelay↓\tjitter↑\tjitter↓");
|
||
synchronized (kpcje.getKlalbController().getLines()) {
|
||
for (Iterator<IPv6NetworkLink> iterator = kpcje.getKlalbController().getLines().iterator(); iterator.hasNext();) {
|
||
IPv6NetworkLink link=iterator.next();
|
||
if(link instanceof KLALBRemoteLink) {
|
||
KLALBRemoteLink hostPort = (KLALBRemoteLink) link;
|
||
System.out.println(hostPort .toString());
|
||
}
|
||
}
|
||
}
|
||
break;
|
||
case "links-add":
|
||
if(sc.length>=2) {
|
||
MultipurposeSocketAddress mpsa=new MultipurposeSocketAddress(sc[1]);
|
||
List<KLALBRemoteLink>addl=kpcje.getKlalbController().addRemoteLines(mpsa);
|
||
if(addl.isEmpty()) {
|
||
//System.out.println("添加失败，线路已存在！");
|
||
System.out.println("link already exist!");
|
||
}else {
|
||
for (Iterator<KLALBRemoteLink> iterator = addl.iterator(); iterator.hasNext();) {
|
||
KLALBRemoteLink klalbRemoteLine = (KLALBRemoteLink) iterator.next();
|
||
|
||
//System.out.println("添加成功："+klalbRemoteLine.getMonitor().getName());
|
||
System.out.println("Successfully added: "+klalbRemoteLine.getName());
|
||
}
|
||
}
|
||
}else {
|
||
// System.out.println("请输入要添加地址:端口!");
|
||
System.out.println("Input Address:port");
|
||
}
|
||
break;
|
||
case "links-remove":
|
||
if(sc.length>=2) {
|
||
MultipurposeSocketAddress mpsa=new MultipurposeSocketAddress(sc[1]);
|
||
List<KLALBRemoteLink>rmvl=kpcje.getKlalbController().removeRemoteLines(mpsa);
|
||
if(rmvl.isEmpty()) {
|
||
System.out.println("No matched link has been found.");
|
||
}else {
|
||
for (Iterator iterator = rmvl.iterator(); iterator.hasNext();) {
|
||
KLALBRemoteLink klalbRemoteLine = (KLALBRemoteLink) iterator.next();
|
||
|
||
System.out.println("Successfully removed: "+klalbRemoteLine.getName());
|
||
}
|
||
}
|
||
}else {
|
||
System.out.println("Invalid arguments.");
|
||
}
|
||
break;
|
||
case "exit":
|
||
System.out.println("Exited.");
|
||
System.exit(0);
|
||
break;
|
||
case "links-reconnect":
|
||
System.out.println("Trying to reconnect all links.");
|
||
kpcje.getKlalbController().reconnectImmediately();
|
||
break;
|
||
case "kltp-state":
|
||
System.out.println(kpcje.getKlalbController().getKLTPregister().toString());
|
||
break;
|
||
case "route":
|
||
List<RouteItem> lri=new ArrayList<>( kpcje.getKlalbController().getIpv6Router().getCurrentRouteTabel());
|
||
Collections.sort(lri);
|
||
System.out.println("Route Table:");
|
||
System.out.println("Prefix\tProtocol\tPref\tCost\tFlag\tNext Hop\tInterface");
|
||
for (Iterator<RouteItem> iterator = lri.iterator(); iterator.hasNext();) {
|
||
RouteItem routeItem = (RouteItem) iterator.next();
|
||
System.out.println(routeItem.getDestination()+"\t"+routeItem.getProto()+"\t"+routeItem.getPre()+"\t"+routeItem.getCost()+"\t"+routeItem.getFlag()+"\t"+routeItem.getNexthop()+"\t"+routeItem.getDestlink().getName());
|
||
}
|
||
|
||
break;
|
||
case "kperf":
|
||
if(sc.length>=2) {
|
||
MultipurposeSocketAddress mpsa=new MultipurposeSocketAddress(sc[1]);
|
||
Kperf kp=new Kperf(mpsa);
|
||
kp.startPerfing();
|
||
}else {
|
||
System.out.println("Invalid arguments.");
|
||
}
|
||
|
||
break;
|
||
case "membandbenchmark":
|
||
MembandBenchmark.main(args);
|
||
break;
|
||
case "memqueuebenchmark":
|
||
|
||
ObjectPoolBandwidthTest.main(args);
|
||
break;
|
||
//case "$$SYSTEM:":
|
||
//System.out.println();
|
||
//break;
|
||
case "nodebenchmark":
|
||
int nodenumber=100;
|
||
if(sc.length>=2) {
|
||
nodenumber=Integer.parseInt(sc[1]);
|
||
}
|
||
NodeBenchmark nbc=new NodeBenchmark(kpcje.getKlalbController(),nodenumber);
|
||
nbc.startPerfing();
|
||
break;
|
||
default:
|
||
System.out.println("Unknown command. See \"help\" or \"?\".");
|
||
}
|
||
}catch (NoSuchElementException err) {
|
||
System.out.println("Exited.");
|
||
System.exit(0);
|
||
}catch(RuntimeException e) {
|
||
System.out.println("Exception Occured!");
|
||
e.printStackTrace();
|
||
}
|
||
}
|
||
|
||
}
|
||
private static void openGUI(KLALBProxySystem kpcje) throws RuntimeException{
|
||
Thread t=new Thread(()->{
|
||
if(ksg==null) {
|
||
ksg=kpcje.getKLALBGUI();
|
||
}
|
||
ksg.setVisible(true);
|
||
});
|
||
t.start();
|
||
}
|
||
|
||
}
|