1
zhanghua
2024-09-26 c775c6953d9759e70f08acbfa8f6d7490aaae3d1
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
package com.netsdk.demo.customize.transmitInfoDemo;
 
import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.TimeUnit;
 
import com.netsdk.demo.customize.transmitInfoDemo.callback.RegisterServiceCallBack;
import com.netsdk.demo.customize.transmitInfoDemo.entity.DeviceInfo;
import com.netsdk.demo.customize.transmitInfoDemo.entity.ListenInfo;
import com.netsdk.demo.customize.transmitInfoDemo.module.AutoRegisterModule;
import com.netsdk.demo.customize.transmitInfoDemo.module.ConfigModule;
import com.netsdk.demo.customize.transmitInfoDemo.module.LoginModule;
import com.netsdk.demo.customize.transmitInfoDemo.module.SdkUtilModule;
import com.netsdk.demo.util.CaseMenu;
import com.netsdk.lib.NetSDKLib;
import com.netsdk.lib.ToolKits;
import com.netsdk.lib.NetSDKLib.CFG_DVRIP_INFO;
import com.netsdk.lib.NetSDKLib.CFG_REGISTER_SERVER_INFO;
import com.netsdk.lib.NetSDKLib.CFG_SERVER_INFO;
import com.netsdk.lib.enumeration.ENUMERROR;
 
/**
 * @author 251823
 * @description AutoRegister Demo
 * @date 2022/3/23
 */
public class AutoRegisterExDemo {    
 
    // 初始化测试
    public void InitTest() {
        SdkUtilModule.Init();  // 初始化SDK库
    }
 
    ///////////////////////////// 主动注册 /////////////////////////////
    ///////////////////////////////////////////////////////////////////
 
    // 用户存储注册上来的设备信息的缓存 Map 项目中请替换成其他中间件
    private final Map<String, DeviceInfo> deviceInfoMap = new ConcurrentHashMap<>();
 
    // 监听句柄
    private NetSDKLib.LLong m_hListenHandle = new NetSDKLib.LLong(0);
    private NetSDKLib.LLong m_hLoginHandle = new NetSDKLib.LLong(0);
 
    /**
     * 开启监听
     */
    public void serverStartListen() {
        m_hListenHandle = AutoRegisterModule.ServerStartListen(serverIpAddr, serverPort, RegisterServiceCallBack.getInstance());
        if (m_hListenHandle.longValue() == 0) return;
        taskIsOpen = true;
        new Thread(this::eventListTask).start();
    }
 
    /**
     * 结束监听
     */
    public void serverStopListen() {
        AutoRegisterModule.ServerStopListen(m_hListenHandle);
        // 清空队列
        taskIsOpen = false;
        deviceInfoMap.clear();
        RegisterServiceCallBack.ServerInfoQueue.clear();
    }
 
    private volatile Boolean taskIsOpen = false;
 
    // 获取监听回调数据并放入缓存
    public void eventListTask() {
        while (taskIsOpen) {
            try {
                // 稍微延迟一下,避免循环的太快
                Thread.sleep(10);
                // 阻塞获取
                ListenInfo listenInfo = RegisterServiceCallBack.ServerInfoQueue.poll(50, TimeUnit.MILLISECONDS);
                if (listenInfo == null) continue;
                // 结果放入缓存
                if (deviceInfoMap.containsKey(listenInfo.devSerial)) {
                     deviceInfoMap.remove(key);                    
                }
                deviceInfoMap.put(listenInfo.devSerial, new DeviceInfo(listenInfo.devIpAddress, listenInfo.devPort));
                System.out.println("New Device AutoRegister Info... Serial:" + listenInfo.devSerial);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }
 
    /**
     * 查看上报的监听数据
     */
    public void checkRegisterDevInfo() {
        // 浅拷贝一个快照
        Map<String, DeviceInfo> deviceInfos = new HashMap<>(deviceInfoMap);
        if (deviceInfos.isEmpty()) System.out.println("没有任何设备上报注册信息.");
 
        StringBuilder info = new StringBuilder();
        for (String label : deviceInfos.keySet()) {
            info.append(String.format("Serial: %s %s\n", label, deviceInfos.get(label).printRegisterInfo()));
        }
        System.out.println(info.toString());
    }
 
    String key = "";
    /**
     * 登录指定设备
     */
    public void deviceLogin() {
        Scanner sc = new Scanner(System.in);
        System.out.println("请输入设备的注册 Serial");
        key = sc.nextLine().trim();
 
        DeviceInfo deviceInfo = deviceInfoMap.get(key);
        if (deviceInfo == null) {
            System.out.println("注册上报的设备中没有该 Serial");
            return;
        }
 
        String ipAddr = deviceInfo.ipAddress;
        int port = deviceInfo.port;
 
        System.out.println("请输入用户名");
        String username = sc.nextLine().trim();
        System.out.println("请输入密码");
        String password = sc.nextLine().trim();
        
        // 高安全登录  主动注册 
        m_hLoginHandle = LoginModule.AutoRegisterLoginWithHighSecurity(key, ipAddr, port, username, password, deviceInfo.m_stDeviceInfo);
        if (m_hLoginHandle.longValue() == 0) {    
            System.err.println("登录失败:" + ENUMERROR.getErrorMessage());
            return;
        }
        // deviceInfo.m_hLoginHandle = m_hLoginHandle;
        // deviceInfo.isLogin = true;
        // 启动业务 Demo
        System.out.println("Start business demo");
        deviceInfoMap.remove(key); // 清除此注册信息 请等待重新上报后再重新登录
    }
    
    /**
     * 登出指定设备
     */
    public void deviceLogout() {
        // 结束测试后登出设备
        LoginModule.logout(m_hLoginHandle);
    }
    
    
    /**
     * 查询设备在线状态
     *
     */
    public void QueryOnlineState() {
        ConfigModule.QueryOnlineState(m_hLoginHandle);
    }
    
    /**
     * 获取网络协议配置
     *
     */
    public void getDVRIPConfig() {
        CFG_DVRIP_INFO msg = ConfigModule.getDVRIPConfig(m_hLoginHandle);
        if(msg != null) {
            System.out.println("TCP服务端口 : " + msg.nTcpPort);
            System.out.println("主动注册配置个数 "+msg.nRegistersNum);
            CFG_REGISTER_SERVER_INFO[] stuRegisters = msg.stuRegisters;
            for (int i = 0; i < msg.nRegistersNum; i++) {
                System.out.println("主动注册使能: "+stuRegisters[i].bEnable);
                System.out.println("设备ID: "+new String(stuRegisters[i].szDeviceID));
                System.out.println("服务器个数: "+stuRegisters[i].nServersNum);
                CFG_SERVER_INFO[] stuServers = stuRegisters[i].stuServers;
                for (int j = 0; j < stuRegisters[i].nServersNum; j++) {
                    System.out.println("服务器端口号: "+stuServers[j].nPort);
                    System.out.println("IP地址或网络名: "+new String(stuServers[j].szAddress));
                }                
            }
        }
    }
    
    /**
     * 下发网络协议配置
     *
     */
    public void setDVRIPConfig() {
        // 获取网络协议配置
        CFG_DVRIP_INFO msg = ConfigModule.getDVRIPConfig(m_hLoginHandle);
        /******修改配置信息******/  
        // 主动注册配置个数
        msg.nRegistersNum = 1;  
        // 主动注册使能
        boolean enable = true;
        msg.stuRegisters[0].bEnable = enable? 1:0;     
        // 服务器个数
        msg.stuRegisters[0].nServersNum = 1;        
        // 服务器地址
        String address = "10.33.121.50";  
        
        StringToByteArray(address, msg.stuRegisters[0].stuServers[0].szAddress);        
        // 服务器端口号
        int nPort = 9500;
        msg.stuRegisters[0].stuServers[0].nPort = nPort;
        
        // 设备ID
        String deviceId = "none";        
        ByteArrayToByteArray(deviceId.getBytes(), msg.stuRegisters[0].szDeviceID);        
        ConfigModule.setDVRIPConfig(m_hLoginHandle, msg);        
    }
    
    /**
     * 字符串拷贝,用于先获取,再设置(src → dst)
     * @param src 
     * @param dst
     */
    public static void StringToByteArray(String src, byte[] dst) {
        for(int i = 0; i < dst.length; i++) {
            dst[i] = 0;
        }
        
        System.arraycopy(src.getBytes(), 0, dst, 0, src.getBytes().length);
    }
    
    /**
     * 数组拷贝, 用于先获取,再设置(src → dst)
     * @param b
     * @param dst
     */
    public static void ByteArrayToByteArray(byte[] src, byte[] dst) {
        for(int i = 0; i < dst.length; i++) {
            dst[i] = 0;
        }
        
        System.arraycopy(src, 0, dst, 0, src.length);
    }
    
    
    // 加载测试内容
    public void RunTest() {
        CaseMenu menu = new CaseMenu();
        menu.addItem(new CaseMenu.Item(this, "开始监听", "serverStartListen"));
        menu.addItem(new CaseMenu.Item(this, "结束监听", "serverStopListen"));
        menu.addItem(new CaseMenu.Item(this, "查看上报信息", "checkRegisterDevInfo"));
        menu.addItem(new CaseMenu.Item(this, "登录设备", "deviceLogin")); // 登录成功后启动业务demo    
        menu.addItem(new CaseMenu.Item(this, "登出设备", "deviceLogout"));   
        
        // 登录成功
        menu.addItem(new CaseMenu.Item(this, "查询设备在线状态", "QueryOnlineState"));   
        menu.addItem(new CaseMenu.Item(this, "获取网络协议配置", "getDVRIPConfig"));
        menu.addItem(new CaseMenu.Item(this, "下发网络协议配置", "setDVRIPConfig"));  
        
        menu.run();        
        System.out.println("主测试已退出");
    }
 
    // 结束测试
    public void EndTest() {
        System.out.println("End Test");
        if (m_hListenHandle.longValue() != 0) serverStopListen();
        System.out.println("See You...");
        SdkUtilModule.cleanup();             // 清理资源
    }
 
    /////////////// 注册地址(服务器 这里是运行此Demo的电脑IP) 监听端口 //////////////////////
    private final String serverIpAddr = "10.33.121.50";
    private final int serverPort = 9500;   // 注意不要和其他程序发生冲突
    /////////////////////////////////////////////////////////////////////////////////////
 
    public static void main(String[] args) {
        AutoRegisterExDemo launcher = new AutoRegisterExDemo();
        launcher.InitTest();
        launcher.RunTest();
        launcher.EndTest();
    }
 
}