import&export

@RequestMapping(value = "/import")
public void importBlackWhiteList(MultipartFile file, HttpServletRequest request, HttpServletResponse response) {
String tenantId = CurrentTenant.getInstance(request).getTenant().getId();
OutputStream os = null;
InputStream is = null;
try {
response.reset();
response.setContentType("application/octet-stream;charset=UTF-8");
is = file.getInputStream();
os = response.getOutputStream();
iPAssignService.importIPAssigns(new TenantId(tenantId), is, os);
} catch (Exception e) {
_logger.error(e.getMessage(), e);
} finally {
try {
if (os != null) {
os.close();
}
if (is != null) {
is.close();
}
} catch (IOException e) {
}
}
}

@RequestMapping(value = "/export")
public void exportAuthLogs(HttpServletRequest request, HttpServletResponse response) throws IOException {
String tenantId = CurrentTenant.getInstance(request).getTenant().getId();
OutputStream os = null;
String fileName = "ip_assignment_template.xlsx";
File excelFile = null;
try {
response.reset();
response.setContentType("application/octet-stream;charset=UTF-8");
os = response.getOutputStream();
String fileAbsolutePath = request.getSession().getServletContext().getRealPath("/") + "/page/resources/" + fileName;
excelFile = iPAssignService.exportIPAssigns(new TenantId(tenantId), fileAbsolutePath, os);
} catch (Exception e) {
_logger.error(e.getMessage(), e);
} finally {
if (os != null) {
os.close();
}
if (excelFile != null) {
FileUtils.forceDelete(excelFile);
}
}
}

package com.ndkey.am.access.ip;

import com.ndkey.am.access.AccessSession;
import com.ndkey.am.access.device.AccessServer;
import com.ndkey.am.access.device.AccessServerId;
import com.ndkey.am.access.device.AccessServerRepository;
import com.ndkey.am.access.device.Ssid;
import com.ndkey.am.access.device.SsidId;
import com.ndkey.am.access.device.SsidRepository;
import com.ndkey.am.access.identity.IdentityStore;
import com.ndkey.am.access.identity.IdentityStoreRepository;
import com.ndkey.am.access.identity.User;
import com.ndkey.am.access.identity.UserInspectionService;
import com.ndkey.am.access.identity.UserRepository;
import com.ndkey.am.tenant.TenantId;
import com.ndkey.mq.client.util.StringUtils;
import com.ndkey.net.MacAddress;
import com.ndkey.utils.RandomString;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.UnknownHostException;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;
import org.apache.commons.io.FileUtils;
import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
import org.apache.poi.openxml4j.opc.OPCPackage;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.transaction.annotation.Transactional;

public class IPAssignService {

private final static int LOGIN_NAME_INDEX = 0;
private final static int IDENTITY_STORE_NAME_INDEX = 1;
private final static int TERMINAL_MAC_INDEX = 2;
private final static int NAS_INDEX = 3;
private final static int IP_INDEX = 4;
private final static int NET_MASK_INDEX = 5;
private final static int GATE_WAY_INDEX = 6;
private final static int RESULT_INDEX = 7;
private final static int FAILURE_REASON_INDEX = 8;

private final static int TERMINAL_MAC_POWER = 4;
private final static int ACCESS_SERVER_POWER = 1;
private final static int SSID_POWER = 2;

private IPAssignPolicyRepository iPAssignPolicyRepository;
private UserRepository userRepository;
private IdentityStoreRepository identityStoreRepository;
private SsidRepository ssidRepository;
private AccessServerRepository accessServerRepository;
private UserInspectionService userInspectionService;

public IPAssignment assignIP(AccessSession accessSession) throws UnknownHostException {
List<IPAssignPolicy> list = iPAssignPolicyRepository.list(accessSession.getUser().getId());
if (list.isEmpty()) {
return null;
}
Collections.sort(list, new Comparator<IPAssignPolicy>() {
@Override
public int compare(IPAssignPolicy o1, IPAssignPolicy o2) {
return calcIPAssignPower(o2) - calcIPAssignPower(o1);  //降序
}
});
for (IPAssignPolicy item : list) {
if (matchsMacAddress(accessSession.getTerminalMac(), item.getTerminalMac())
&& matchsNAS(accessSession.getAccessServer(), item.getAccessServerId())
&& matchsSsid(accessSession.getSsid(), item.getSsidId())) {
return new IPAssignment(item.getIp(), item.getNetmask(), item.getGateway());
}
}
return null;
}

private boolean matchsMacAddress(MacAddress terminalEnd, MacAddress deliverMac) {
if (deliverMac == null) {
return true;
}
if (terminalEnd != null) {
return deliverMac.equals(terminalEnd);
} else {
return false;
}
}

private boolean matchsNAS(AccessServer terminalEnd, AccessServerId accessServerId) {
if (accessServerId == null) {
return true;
}
if (terminalEnd != null) {
return accessServerId.getId().equals(terminalEnd.getId().getId());
} else {
return false;
}
}

private boolean matchsSsid(Ssid terminalEnd, SsidId ssidId) {
if (ssidId == null) {
return true;
}
if (terminalEnd != null) {
return ssidId.getId().equals(terminalEnd.getId().getId());
} else {
return false;
}
}

private int calcIPAssignPower(IPAssignPolicy ipAssign) {
int power = 0;
power += ipAssign.getTerminalMac() == null ? 0 : IPAssignService.TERMINAL_MAC_POWER;
power += ipAssign.getAccessServerId() == null ? 0 : IPAssignService.ACCESS_SERVER_POWER;
power += ipAssign.getSsidId() == null ? 0 : IPAssignService.SSID_POWER;
return power;
}

@Transactional(timeout = 60 * 60 * 8)
public void importIPAssigns(TenantId tenantId, InputStream is, OutputStream os) throws IOException {
XSSFWorkbook wb = new XSSFWorkbook(is);
XSSFSheet sheet = wb.getSheetAt(0);
int maxRowNum = sheet.getLastRowNum();
XSSFRow xSSFRow = null;
for (int index = 3; index <= maxRowNum; index++) {
xSSFRow = sheet.getRow(index);
if (xSSFRow != null) {
try {
String identityStoreName = getCellValue(xSSFRow.getCell(IDENTITY_STORE_NAME_INDEX));
IdentityStore identityStore = identityStoreRepository.getByName(tenantId, identityStoreName);
if (StringUtils.isBlank(identityStoreName) || identityStore == null) {
markFailureRow(xSSFRow, "identity store not exist");
continue;
}
String loginName = getCellValue(xSSFRow.getCell(LOGIN_NAME_INDEX));
User user = userRepository.getByLoginName(identityStore.getId(), loginName);
if (StringUtils.isBlank(loginName) || user == null) {
markFailureRow(xSSFRow, "user not exist");
continue;
}
String macAddress = getCellValue(xSSFRow.getCell(TERMINAL_MAC_INDEX));
MacAddress terminalMac = StringUtils.isBlank(macAddress) ? null : MacAddress.parseMacAddress(macAddress);
String nasName = getCellValue(xSSFRow.getCell(NAS_INDEX));
AccessServer nas = StringUtils.isBlank(nasName) ? null : accessServerRepository.getByName(tenantId, nasName);
AccessServerId accessServerId = nas == null ? null : nas.getId();
String ipAddress = getCellValue(xSSFRow.getCell(IP_INDEX));
String netmaskAddress = getCellValue(xSSFRow.getCell(NET_MASK_INDEX));
String gatewayAddress = getCellValue(xSSFRow.getCell(GATE_WAY_INDEX));
if (StringUtils.isBlank(ipAddress) || StringUtils.isBlank(netmaskAddress) || StringUtils.isBlank(gatewayAddress)) {
markFailureRow(xSSFRow, "ip netmask gateway is null");
continue;
}
IPAssignPolicy ipAssignPolicy = new IPAssignPolicy(tenantId, user.getId(), terminalMac, accessServerId, null, ipAddress, netmaskAddress, gatewayAddress);
iPAssignPolicyRepository.save(ipAssignPolicy);
// String mobileOperator = xSSFRow.getCell(USER_MOBILE_OPERATOR_INDEX) == null ? "" : getCellValue(xSSFRow.getCell(USER_MOBILE_OPERATOR_INDEX));
// String mobileLocation = xSSFRow.getCell(USER_MOBILE_LOCATION_INDEX) == null ? "" : getCellValue(xSSFRow.getCell(USER_MOBILE_LOCATION_INDEX));
this.markSuccessRow(xSSFRow);
} catch (Exception e) {
markFailureRow(xSSFRow, e.getMessage());
}
}
}
wb.write(os);
wb = null;
}

public File exportIPAssigns(TenantId tenantId, String templatePath, OutputStream os) throws IOException, InvalidFormatException {
String tempPath = FileUtils.getTempDirectoryPath();
File excelFile = new File(tempPath + "\\" + generateFileName() + ".xlsx");
File templateFile = new File(templatePath);
FileUtils.copyFile(templateFile, excelFile);
XSSFWorkbook wb;
try (OPCPackage opk = OPCPackage.open(excelFile)) {
wb = new XSSFWorkbook(opk);
XSSFSheet sheet = wb.getSheetAt(0);
XSSFRow row = null;
List<IPAssignPolicy> list = iPAssignPolicyRepository.list(tenantId, 0, Integer.MAX_VALUE);
int i = 3;
for (IPAssignPolicy item : list) {
row = sheet.createRow(i);
User user = userRepository.get(item.getUserId());
IdentityStore identityStore = userInspectionService.inspectForIdentityStore(user);
AccessServer accessServer = item.getAccessServerId() == null ? null : accessServerRepository.get(item.getAccessServerId());
row.createCell(LOGIN_NAME_INDEX, Cell.CELL_TYPE_STRING).setCellValue(user.getLoginName());
row.createCell(IDENTITY_STORE_NAME_INDEX, Cell.CELL_TYPE_STRING).setCellValue(identityStore == null ? "" : identityStore.getName());
row.createCell(TERMINAL_MAC_INDEX, Cell.CELL_TYPE_STRING).setCellValue(item.getTerminalMac() == null ? "" : item.getTerminalMac().getAddress());
row.createCell(NAS_INDEX, Cell.CELL_TYPE_STRING).setCellValue(accessServer == null ? "" : accessServer.getName());
row.createCell(IP_INDEX, Cell.CELL_TYPE_STRING).setCellValue(StringUtils.isBlank(item.getIp())?"":item.getIp());
row.createCell(NET_MASK_INDEX, Cell.CELL_TYPE_STRING).setCellValue(StringUtils.isBlank(item.getNetmask())?"":item.getNetmask());
row.createCell(GATE_WAY_INDEX, Cell.CELL_TYPE_STRING).setCellValue(StringUtils.isBlank(item.getGateway())?"":item.getGateway());
i++;
}
wb.write(os);
}
wb = null;
// 3. return file
return excelFile;
}

public String getCellValue(Cell cell) {
cell.setCellType(Cell.CELL_TYPE_STRING);
return cell.getStringCellValue();
}

private void markFailureRow(XSSFRow row, String message) {
if (row.getCell(FAILURE_REASON_INDEX) == null) {
row.createCell(FAILURE_REASON_INDEX).setCellValue(message);
} else {
row.getCell(FAILURE_REASON_INDEX).setCellValue(message);
}
if (row.getCell(RESULT_INDEX) == null) {
row.createCell(RESULT_INDEX).setCellValue("失败");
} else {
row.getCell(RESULT_INDEX).setCellValue("失败");
}
}

private void markSuccessRow(XSSFRow row) {
if (row.getCell(RESULT_INDEX) == null) {
row.createCell(RESULT_INDEX).setCellValue("成功");
} else {
row.getCell(RESULT_INDEX).setCellValue("成功");
}
}

private String generateFileName() {
return RandomString.getRandomString("qazwsxedcrfvtgbyhnujmikl1234567890", 6);
}

public void setiPAssignPolicyRepository(IPAssignPolicyRepository iPAssignPolicyRepository) {
this.iPAssignPolicyRepository = iPAssignPolicyRepository;
}

public void setUserRepository(UserRepository userRepository) {
this.userRepository = userRepository;
}

public void setIdentityStoreRepository(IdentityStoreRepository identityStoreRepository) {
this.identityStoreRepository = identityStoreRepository;
}

public void setSsidRepository(SsidRepository ssidRepository) {
this.ssidRepository = ssidRepository;
}

public void setAccessServerRepository(AccessServerRepository accessServerRepository) {
this.accessServerRepository = accessServerRepository;
}

public void setUserInspectionService(UserInspectionService userInspectionService) {
this.userInspectionService = userInspectionService;
}

}