package com.netsdk.demo.customize;

import static com.netsdk.lib.Utils.getOsPrefix;

import java.io.File;
import java.io.UnsupportedEncodingException;

import com.netsdk.demo.customize.analyseTaskDemo.AnalyseTaskUtils;
import com.netsdk.demo.util.CaseMenu;
import com.netsdk.lib.NetSDKLib;
import com.netsdk.lib.ToolKits;
import com.netsdk.lib.callback.impl.DefaultDisconnectCallback;
import com.netsdk.lib.callback.impl.DefaultHaveReconnectCallBack;
import com.netsdk.lib.structure.NET_DEV_IO_STAT_INFO;
import com.netsdk.lib.structure.NET_IN_DEV_IO_STATS;
import com.netsdk.lib.structure.NET_OUT_DEV_IO_STATS;

/**
 * @author 251823
 * @description 获取所有存储设备IO信息
 * @date 2021/07/10
 */
public class QueryDevIOStatsDemo {
	// SDk对象初始化
		public static final NetSDKLib netsdk = NetSDKLib.NETSDK_INSTANCE;
		public static final NetSDKLib configsdk = NetSDKLib.CONFIG_INSTANCE;

		// 判断是否初始化
		private static boolean bInit = false;
		// 判断log是否打开
		private static boolean bLogOpen = false;
		// 设备信息
		//private NetSDKLib.NET_DEVICEINFO_Ex deviceInfo = new NetSDKLib.NET_DEVICEINFO_Ex();
		// 登录句柄
		private NetSDKLib.LLong m_hLoginHandle = new NetSDKLib.LLong(0);

		// 回调函数需要是静态的，防止被系统回收
		// 断线回调
		private static NetSDKLib.fDisConnect disConnectCB = DefaultDisconnectCallback.getINSTANCE();
		// 重连回调
		private static NetSDKLib.fHaveReConnect haveReConnectCB = DefaultHaveReconnectCallBack.getINSTANCE();

		// 编码格式
		public static String encode;

		static {
			String osPrefix = getOsPrefix();
			if (osPrefix.toLowerCase().startsWith("win32-amd64")) {
				encode = "GBK";
			} else if (osPrefix.toLowerCase().startsWith("linux-amd64")) {
				encode = "UTF-8";
			}
		}

		/**
		 * 初始化SDK库
		 */
		public static boolean Init() {
			bInit = netsdk.CLIENT_Init(disConnectCB, null);
			if (!bInit) {
				System.out.println("Initialize SDK failed");
				return false;
			}
			// 配置日志
			QueryDevIOStatsDemo.enableLog();

			// 设置断线重连回调接口, 此操作为可选操作，但建议用户进行设置
			netsdk.CLIENT_SetAutoReconnect(haveReConnectCB, null);

			// 设置登录超时时间和尝试次数，可选
			// 登录请求响应超时时间设置为3S
			int waitTime = 3000;
			// 登录时尝试建立链接 1 次
			int tryTimes = 1;
			netsdk.CLIENT_SetConnectTime(waitTime, tryTimes);
			// 设置更多网络参数， NET_PARAM 的nWaittime ， nConnectTryNum 成员与 CLIENT_SetConnectTime
			// 接口设置的登录设备超时时间和尝试次数意义相同,可选
			NetSDKLib.NET_PARAM netParam = new NetSDKLib.NET_PARAM();
			// 登录时尝试建立链接的超时时间
			netParam.nConnectTime = 10000;
			// 设置子连接的超时时间
			netParam.nGetConnInfoTime = 3000;
			netsdk.CLIENT_SetNetworkParam(netParam);
			return true;
		}

		/**
		 * 打开 sdk log
		 */
		private static void enableLog() {
			NetSDKLib.LOG_SET_PRINT_INFO setLog = new NetSDKLib.LOG_SET_PRINT_INFO();
			File path = new File("sdklog/");
			if (!path.exists())
				path.mkdir();

			// 这里的log保存地址依据实际情况自己调整
			String logPath = path.getAbsoluteFile().getParent() + "\\sdklog\\" + "sdklog" + AnalyseTaskUtils.getDate()
					+ ".log";
			setLog.nPrintStrategy = 0;
			setLog.bSetFilePath = 1;
			System.arraycopy(logPath.getBytes(), 0, setLog.szLogFilePath, 0, logPath.getBytes().length);
			System.out.println(logPath);
			setLog.bSetPrintStrategy = 1;
			bLogOpen = netsdk.CLIENT_LogOpen(setLog);
			if (!bLogOpen)
				System.err.println("Failed to open NetSDK log");
		}

		/**
		 * 高安全登录
		 */
		public void loginWithHighLevel() {
			// 输入结构体参数
			NetSDKLib.NET_IN_LOGIN_WITH_HIGHLEVEL_SECURITY pstlnParam = new NetSDKLib.NET_IN_LOGIN_WITH_HIGHLEVEL_SECURITY() {
				{
					szIP = m_strIpAddr.getBytes();
					nPort = m_nPort;
					szUserName = m_strUser.getBytes();
					szPassword = m_strPassword.getBytes();
				}
			};
			// 输出结构体参数
			NetSDKLib.NET_OUT_LOGIN_WITH_HIGHLEVEL_SECURITY pstOutParam = new NetSDKLib.NET_OUT_LOGIN_WITH_HIGHLEVEL_SECURITY();

			// 写入sdk
			m_hLoginHandle = netsdk.CLIENT_LoginWithHighLevelSecurity(pstlnParam, pstOutParam);
			if (m_hLoginHandle.longValue() == 0) {
				System.err.printf("Login Device[%s] Port[%d]Failed. %s\n", m_strIpAddr, m_nPort,
						netsdk.CLIENT_GetLastError());
			} else {
				// deviceInfo = pstOutParam.stuDeviceInfo; // 获取设备信息
				System.out.println("Login Success");
				System.out.println("Device Address：" + m_strIpAddr);
				// System.out.println("设备包含：" + deviceInfo.byChanNum + "个通道");
			}
		}

		/**
		 * 退出
		 */
		public void logOut() {
			if (m_hLoginHandle.longValue() != 0) {
				netsdk.CLIENT_Logout(m_hLoginHandle);
				System.out.println("LogOut Success");
			}
		}

		/**
		 * 清除 sdk环境
		 */
		public static void cleanup() {
			if (bLogOpen) {
				netsdk.CLIENT_LogClose();
			}
			if (bInit) {
				netsdk.CLIENT_Cleanup();
			}
		}

		/**
		 * 清理并退出
		 */
		public static void cleanAndExit() {
			netsdk.CLIENT_Cleanup();
			System.exit(0);
		}

		/**
		 * 获取存储设备IO信息
		 */
		public void getDevIOStats() {
			int nQueryType = NetSDKLib.NET_QUERY_DEV_IO_STATS;
			// 入参
			NET_IN_DEV_IO_STATS stIn = new NET_IN_DEV_IO_STATS();
			
			// 出参
			NET_OUT_DEV_IO_STATS stOut = new NET_OUT_DEV_IO_STATS();

			stIn.write();
			stOut.write();
			
	    	boolean bRet = netsdk.CLIENT_QueryDevInfo(m_hLoginHandle, nQueryType, stIn.getPointer(), stOut.getPointer(), null, 3000);
			if(bRet) {
				stOut.read();
				System.out.println("存储设备io信息个数:"+stOut.nInfoNum);
				NET_DEV_IO_STAT_INFO[] stuInfo = stOut.stuInfo;
				for (int i = 0; i < stOut.nInfoNum; i++) {
					System.out.println("------第"+(i+1)+"个信息----------");
					try {
						System.out.println("存储设备名称:"+new String(stuInfo[i].szDeviceName,encode));
					} catch (UnsupportedEncodingException e) {
						e.printStackTrace();
					}
					System.out.println("每秒完成的读次数:"+stuInfo[i].nReadIos);
					System.out.println("每秒对该设备的读请求被合并次数:"+stuInfo[i].nReadMerges);
					System.out.println("每秒读数据量(kB为单位):"+stuInfo[i].nReadSectors);
					System.out.println("每秒完成的写次数:"+stuInfo[i].nWriteIos);
					System.out.println("每秒对该设备的写请求被合并次数:"+stuInfo[i].nWriteMerges);
					System.out.println("每秒写数据量(kB为单位):"+stuInfo[i].nWriteSectors);
					System.out.println("平均每次IO操作的数据量(扇区数为单位):"+stuInfo[i].nSize);
					System.out.println("平均等待处理的IO请求队列长度:"+stuInfo[i].nQueue);
					System.out.println("平均每次IO请求等待时间(包括等待时间和处理时间，毫秒为单位):"+stuInfo[i].nWait);
					System.out.println("平均每次IO请求的处理时间(毫秒为单位):"+stuInfo[i].nSvctm);
					System.out.println("采用周期内用于IO操作的时间比率，即IO队列非空的时间比率,范围[0,100]:"+stuInfo[i].nUtil);					
				}
				
				
			} else {
				System.err.println("getDevIOStats Failed!" + ToolKits.getErrorCode());
			}
		}

		/******************************** 测试控制台 ***************************************/

		// 配置登陆地址，端口，用户名，密码
		private String m_strIpAddr = "172.12.10.18";
		private int m_nPort = 37777;
		private String m_strUser = "admin";
		private String m_strPassword = "admin123";

		public static void main(String[] args) {
			QueryDevIOStatsDemo demo = new QueryDevIOStatsDemo();
			demo.InitTest();
			demo.RunTest();
			demo.EndTest();

		}

		/**
		 * 初始化测试
		 */
		public void InitTest() {
			QueryDevIOStatsDemo.Init();
			this.loginWithHighLevel();
		}

		/**
		 * 加载测试内容
		 */
		public void RunTest() {
			CaseMenu menu = new CaseMenu();
			menu.addItem(new CaseMenu.Item(this, "获取存储设备IO信息", "getDevIOStats"));

			menu.run();
		}

		/**
		 * 结束测试
		 */
		public void EndTest() {
			System.out.println("End Test");
			this.logOut(); // 退出
			System.out.println("See You...");
			QueryDevIOStatsDemo.cleanAndExit(); // 清理资源并退出
		}
		/******************************** 结束 ***************************************/


}