LAPTOP-S9HJSOEB\昊天
3 years ago
parent
6cb148ff3e
commit
5736eb5a6a
@ -0,0 +1,37 @@
|
|||||||
|
package com.leaper.web.config;
|
||||||
|
|
||||||
|
|
||||||
|
import com.leaper.web.service.sftp.SftpFactory;
|
||||||
|
import com.leaper.web.service.sftp.SftpHelper;
|
||||||
|
import com.leaper.web.service.sftp.SftpPool;
|
||||||
|
import org.springframework.boot.context.properties.EnableConfigurationProperties;
|
||||||
|
import org.springframework.context.annotation.Bean;
|
||||||
|
import org.springframework.context.annotation.Configuration;
|
||||||
|
|
||||||
|
import javax.annotation.Resource;
|
||||||
|
|
||||||
|
// ftp配置
|
||||||
|
@Configuration
|
||||||
|
public class SftpConfig {
|
||||||
|
@Resource
|
||||||
|
ConfigProperties properties;
|
||||||
|
|
||||||
|
// 工厂
|
||||||
|
@Bean
|
||||||
|
public SftpFactory sftpFactory(ConfigProperties properties) {
|
||||||
|
return new SftpFactory(properties);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 连接池
|
||||||
|
@Bean
|
||||||
|
public SftpPool sftpPool(SftpFactory sftpFactory) {
|
||||||
|
return new SftpPool(sftpFactory);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 辅助类
|
||||||
|
@Bean
|
||||||
|
public SftpHelper sftpHelper(SftpPool sftpPool) {
|
||||||
|
return new SftpHelper(sftpPool);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
@ -0,0 +1,23 @@
|
|||||||
|
package com.leaper.web.entity;
|
||||||
|
|
||||||
|
import com.baomidou.mybatisplus.annotation.IdType;
|
||||||
|
import com.baomidou.mybatisplus.annotation.TableId;
|
||||||
|
import com.fasterxml.jackson.annotation.JsonFormat;
|
||||||
|
import lombok.Data;
|
||||||
|
|
||||||
|
import java.time.LocalDateTime;
|
||||||
|
|
||||||
|
@Data
|
||||||
|
public class PicData {
|
||||||
|
|
||||||
|
@TableId(type = IdType.AUTO)
|
||||||
|
private Integer id;
|
||||||
|
|
||||||
|
private String url;
|
||||||
|
|
||||||
|
private String type;
|
||||||
|
|
||||||
|
@JsonFormat(pattern="yyyy-MM-dd HH:mm:ss")
|
||||||
|
private LocalDateTime createTime;
|
||||||
|
|
||||||
|
}
|
||||||
@ -0,0 +1,8 @@
|
|||||||
|
package com.leaper.web.mapper;
|
||||||
|
|
||||||
|
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
|
||||||
|
import com.leaper.web.entity.LightSource;
|
||||||
|
import com.leaper.web.entity.OrderLive;
|
||||||
|
|
||||||
|
public interface OrderLiveMapper extends BaseMapper<OrderLive> {
|
||||||
|
}
|
||||||
@ -0,0 +1,8 @@
|
|||||||
|
package com.leaper.web.mapper;
|
||||||
|
|
||||||
|
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
|
||||||
|
import com.leaper.web.entity.PicData;
|
||||||
|
|
||||||
|
|
||||||
|
public interface PicDataMapper extends BaseMapper<PicData> {
|
||||||
|
}
|
||||||
@ -0,0 +1,61 @@
|
|||||||
|
package com.leaper.web.service;
|
||||||
|
|
||||||
|
import com.sun.management.OperatingSystemMXBean;
|
||||||
|
|
||||||
|
import java.io.File;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.lang.management.ManagementFactory;
|
||||||
|
import java.text.DecimalFormat;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author layman
|
||||||
|
*/
|
||||||
|
public class Demo01 {
|
||||||
|
public static void main(String[] args) throws IOException {
|
||||||
|
getDiskInfo();
|
||||||
|
getMemoryInfo();
|
||||||
|
}
|
||||||
|
/**
|
||||||
|
* 获取系统各个硬盘的总容量、已经使用的容量、剩余容量和使用率
|
||||||
|
* @throws IOException
|
||||||
|
*/
|
||||||
|
public static void getDiskInfo() throws IOException {
|
||||||
|
DecimalFormat df = new DecimalFormat("#0.00");
|
||||||
|
File[] disks = File.listRoots();
|
||||||
|
for (File file : disks) {
|
||||||
|
// 获取盘符
|
||||||
|
System.out.print(file.getCanonicalPath() + " ");
|
||||||
|
// 获取总容量
|
||||||
|
long totalSpace = file.getTotalSpace();
|
||||||
|
// 获取剩余容量
|
||||||
|
long usableSpace = file.getUsableSpace();
|
||||||
|
// 获取已经使用的容量
|
||||||
|
long freeSpace = totalSpace - usableSpace;
|
||||||
|
// 获取使用率
|
||||||
|
float useRate = (float)((freeSpace * 1.0 / totalSpace) * 100);
|
||||||
|
System.out.print("总容量： " + transformation(totalSpace));
|
||||||
|
System.out.print("已经使用： " + transformation(freeSpace));
|
||||||
|
System.out.print("剩余容量： " + transformation(usableSpace));
|
||||||
|
System.out.println("使用率： " + Double.parseDouble(df.format(useRate)) + "% ");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
/**
|
||||||
|
* 获取内存使用情况
|
||||||
|
*/
|
||||||
|
public static void getMemoryInfo() {
|
||||||
|
OperatingSystemMXBean mem = (OperatingSystemMXBean) ManagementFactory.getOperatingSystemMXBean();
|
||||||
|
// 获取内存总容量
|
||||||
|
long totalMemorySize = mem.getTotalPhysicalMemorySize();
|
||||||
|
// 获取可用内存容量
|
||||||
|
long freeMemorySize = mem.getFreePhysicalMemorySize();
|
||||||
|
System.out.println("内存总容量：" + transformation(totalMemorySize) );
|
||||||
|
System.out.println("可用容量：" + transformation(freeMemorySize));
|
||||||
|
}
|
||||||
|
/**
|
||||||
|
* 将字节容量转化为GB
|
||||||
|
*/
|
||||||
|
public static String transformation(long size){
|
||||||
|
return size / 1024 / 1024 / 1024 + "GB"+" ";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
@ -0,0 +1,100 @@
|
|||||||
|
package com.leaper.web.service.sftp;
|
||||||
|
|
||||||
|
|
||||||
|
import com.jcraft.jsch.ChannelSftp;
|
||||||
|
import com.jcraft.jsch.JSch;
|
||||||
|
import com.jcraft.jsch.JSchException;
|
||||||
|
import com.jcraft.jsch.Session;
|
||||||
|
import com.leaper.web.config.ConfigProperties;
|
||||||
|
import lombok.Data;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.apache.commons.pool2.BasePooledObjectFactory;
|
||||||
|
import org.apache.commons.pool2.PooledObject;
|
||||||
|
import org.apache.commons.pool2.impl.DefaultPooledObject;
|
||||||
|
import org.apache.commons.pool2.impl.GenericObjectPoolConfig;
|
||||||
|
|
||||||
|
import javax.annotation.Resource;
|
||||||
|
import java.util.Properties;
|
||||||
|
|
||||||
|
@Data
|
||||||
|
@Slf4j
|
||||||
|
public class SftpFactory extends BasePooledObjectFactory<ChannelSftp> {
|
||||||
|
|
||||||
|
private Pool pool = new Pool();
|
||||||
|
|
||||||
|
public static class Pool extends GenericObjectPoolConfig<ChannelSftp> {
|
||||||
|
|
||||||
|
private int maxTotal = DEFAULT_MAX_TOTAL;
|
||||||
|
private int maxIdle = DEFAULT_MAX_IDLE;
|
||||||
|
private int minIdle = DEFAULT_MIN_IDLE;
|
||||||
|
|
||||||
|
public Pool() {
|
||||||
|
super();
|
||||||
|
}
|
||||||
|
@Override
|
||||||
|
public int getMaxTotal() {
|
||||||
|
return maxTotal;
|
||||||
|
}
|
||||||
|
@Override
|
||||||
|
public void setMaxTotal(int maxTotal) {
|
||||||
|
this.maxTotal = maxTotal;
|
||||||
|
}
|
||||||
|
@Override
|
||||||
|
public int getMaxIdle() {
|
||||||
|
return maxIdle;
|
||||||
|
}
|
||||||
|
@Override
|
||||||
|
public void setMaxIdle(int maxIdle) {
|
||||||
|
this.maxIdle = maxIdle;
|
||||||
|
}
|
||||||
|
@Override
|
||||||
|
public int getMinIdle() {
|
||||||
|
return minIdle;
|
||||||
|
}
|
||||||
|
@Override
|
||||||
|
public void setMinIdle(int minIdle) {
|
||||||
|
this.minIdle = minIdle;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
ConfigProperties configProperties;
|
||||||
|
|
||||||
|
public SftpFactory(ConfigProperties properties) {
|
||||||
|
this.configProperties = properties;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public ChannelSftp create() {
|
||||||
|
try {
|
||||||
|
JSch jsch = new JSch();
|
||||||
|
Session sshSession = jsch.getSession(configProperties.getSavePath().getCloudUser(),configProperties.getSavePath().getCloudIp(), 22);
|
||||||
|
sshSession.setPassword(configProperties.getSavePath().getCloudPassword());
|
||||||
|
Properties sshConfig = new Properties();
|
||||||
|
sshConfig.put("StrictHostKeyChecking", "no");
|
||||||
|
sshSession.setConfig(sshConfig);
|
||||||
|
sshSession.connect();
|
||||||
|
ChannelSftp channel = (ChannelSftp) sshSession.openChannel("sftp");
|
||||||
|
channel.connect();
|
||||||
|
return channel;
|
||||||
|
} catch (JSchException e) {
|
||||||
|
log.info("连接sfpt失败");
|
||||||
|
e.printStackTrace();
|
||||||
|
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public PooledObject<ChannelSftp> wrap(ChannelSftp channelSftp) {
|
||||||
|
return new DefaultPooledObject<>(channelSftp);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 销毁对象
|
||||||
|
@Override
|
||||||
|
public void destroyObject(PooledObject<ChannelSftp> p) {
|
||||||
|
ChannelSftp channelSftp = p.getObject();
|
||||||
|
channelSftp.disconnect();
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
@ -0,0 +1,190 @@
|
|||||||
|
package com.leaper.web.service.sftp;
|
||||||
|
|
||||||
|
|
||||||
|
import com.jcraft.jsch.ChannelSftp;
|
||||||
|
import com.jcraft.jsch.SftpATTRS;
|
||||||
|
import com.jcraft.jsch.SftpException;
|
||||||
|
import io.swagger.models.auth.In;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.InputStream;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.stream.IntStream;
|
||||||
|
|
||||||
|
// sftp辅助类
|
||||||
|
@Slf4j
|
||||||
|
public class SftpHelper {
|
||||||
|
|
||||||
|
private SftpPool pool;
|
||||||
|
|
||||||
|
public SftpHelper(SftpPool pool) {
|
||||||
|
this.pool = pool;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 下载文件
|
||||||
|
* @param dir 远程目录
|
||||||
|
* @param name 远程文件名
|
||||||
|
* @return 文件字节数组
|
||||||
|
*/
|
||||||
|
public byte[] download(String dir, String name) {
|
||||||
|
ChannelSftp sftp = pool.borrowObject();
|
||||||
|
try {
|
||||||
|
sftp.cd(dir);
|
||||||
|
InputStream in = sftp.get(name);
|
||||||
|
|
||||||
|
return new byte[in.available()];
|
||||||
|
} catch (SftpException e) {
|
||||||
|
log.info("sftp下载文件出错");
|
||||||
|
e.printStackTrace();
|
||||||
|
return null;
|
||||||
|
} catch (IOException e) {
|
||||||
|
throw new RuntimeException(e);
|
||||||
|
} finally {
|
||||||
|
pool.returnObject(sftp);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 上传文件
|
||||||
|
* @param dir 远程目录
|
||||||
|
* @param name 远程文件名
|
||||||
|
* @param in 输入流
|
||||||
|
*/
|
||||||
|
public void upload(String dir, String name, InputStream in) {
|
||||||
|
ChannelSftp sftp = pool.borrowObject();
|
||||||
|
try {
|
||||||
|
mkdirs(dir);
|
||||||
|
sftp.cd(dir);
|
||||||
|
sftp.put(in, name);
|
||||||
|
} catch (SftpException e) {
|
||||||
|
|
||||||
|
log.info("sftp上传文件出错");
|
||||||
|
e.printStackTrace();
|
||||||
|
} finally {
|
||||||
|
pool.returnObject(sftp);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public void put(String srcFilePath, String targetFilePath) {
|
||||||
|
ChannelSftp sftp = pool.borrowObject();
|
||||||
|
try {
|
||||||
|
sftp.put(srcFilePath, targetFilePath);
|
||||||
|
} catch (SftpException e) {
|
||||||
|
|
||||||
|
log.info("sftp上传文件出错");
|
||||||
|
e.printStackTrace();
|
||||||
|
} finally {
|
||||||
|
pool.returnObject(sftp);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 删除文件
|
||||||
|
* @param dir 远程目录
|
||||||
|
* @param name 远程文件名
|
||||||
|
*/
|
||||||
|
public void delete(String dir, String name) {
|
||||||
|
ChannelSftp sftp = pool.borrowObject();
|
||||||
|
try {
|
||||||
|
sftp.cd(dir);
|
||||||
|
sftp.rm(name);
|
||||||
|
} catch (SftpException e) {
|
||||||
|
log.info("sftp删除文件出错");
|
||||||
|
e.printStackTrace();
|
||||||
|
} finally {
|
||||||
|
pool.returnObject(sftp);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
/**
|
||||||
|
* 删除文件
|
||||||
|
* @param dirPath 远程目录文件名
|
||||||
|
*/
|
||||||
|
public void delete(String dirPath) {
|
||||||
|
ChannelSftp sftp = pool.borrowObject();
|
||||||
|
try {
|
||||||
|
sftp.rm(dirPath);
|
||||||
|
} catch (SftpException e) {
|
||||||
|
log.info("sftp删除文件出错");
|
||||||
|
e.printStackTrace();
|
||||||
|
} finally {
|
||||||
|
pool.returnObject(sftp);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 查看文件夹
|
||||||
|
* @param dirPath 远程目录文件名
|
||||||
|
*/
|
||||||
|
public List<ChannelSftp.LsEntry> ls(String dirPath) {
|
||||||
|
ChannelSftp sftp = pool.borrowObject();
|
||||||
|
List<ChannelSftp.LsEntry> list = new ArrayList<>();
|
||||||
|
try {
|
||||||
|
list = sftp.ls(dirPath);
|
||||||
|
} catch (SftpException e) {
|
||||||
|
log.info("sftp查看文件夹失败");
|
||||||
|
e.printStackTrace();
|
||||||
|
} finally {
|
||||||
|
pool.returnObject(sftp);
|
||||||
|
}
|
||||||
|
return list;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 文件是否存在
|
||||||
|
* @param dir 远程文件名
|
||||||
|
*/
|
||||||
|
public boolean isExist(String dir) {
|
||||||
|
ChannelSftp sftp = pool.borrowObject();
|
||||||
|
try {
|
||||||
|
sftp.stat(dir);
|
||||||
|
return true;
|
||||||
|
} catch (SftpException e) {
|
||||||
|
log.info("sftp文件不存在");
|
||||||
|
return false;
|
||||||
|
} finally {
|
||||||
|
pool.returnObject(sftp);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 递归创建多级目录
|
||||||
|
*
|
||||||
|
* @param dir 多级目录
|
||||||
|
*/
|
||||||
|
public void mkdirs(String dir) {
|
||||||
|
String[] folders = dir.split("/");
|
||||||
|
ChannelSftp sftp = pool.borrowObject();
|
||||||
|
try {
|
||||||
|
sftp.cd("/");
|
||||||
|
for (String folder: folders) {
|
||||||
|
if (folder.length()>0) {
|
||||||
|
try {
|
||||||
|
sftp.cd(folder);
|
||||||
|
} catch (Exception e) {
|
||||||
|
sftp.mkdir(folder);
|
||||||
|
sftp.cd(folder);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (SftpException e) {
|
||||||
|
log.info("sftp创建目录出错");
|
||||||
|
e.printStackTrace();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public void rmdir(String dirPath) {
|
||||||
|
ChannelSftp sftp = pool.borrowObject();
|
||||||
|
try {
|
||||||
|
sftp.rmdir(dirPath);
|
||||||
|
} catch (SftpException e) {
|
||||||
|
log.info("sftp删除文件夹出错");
|
||||||
|
e.printStackTrace();
|
||||||
|
} finally {
|
||||||
|
pool.returnObject(sftp);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
@ -0,0 +1,45 @@
|
|||||||
|
package com.leaper.web.service.sftp;
|
||||||
|
|
||||||
|
import com.jcraft.jsch.ChannelSftp;
|
||||||
|
import lombok.Data;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.apache.commons.pool2.impl.GenericObjectPool;
|
||||||
|
import org.apache.commons.pool2.impl.GenericObjectPoolConfig;
|
||||||
|
import org.springframework.boot.context.properties.ConfigurationProperties;
|
||||||
|
|
||||||
|
@Slf4j
|
||||||
|
@Data
|
||||||
|
public class SftpPool {
|
||||||
|
|
||||||
|
private GenericObjectPool<ChannelSftp> pool;
|
||||||
|
|
||||||
|
public SftpPool(SftpFactory factory) {
|
||||||
|
this.pool = new GenericObjectPool<>(factory, factory.getPool());
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取一个sftp连接对象
|
||||||
|
* @return sftp连接对象
|
||||||
|
*/
|
||||||
|
public ChannelSftp borrowObject() {
|
||||||
|
try {
|
||||||
|
return pool.borrowObject();
|
||||||
|
} catch (Exception e) {
|
||||||
|
log.info("获取ftp连接失败");
|
||||||
|
e.printStackTrace();
|
||||||
|
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 归还一个sftp连接对象
|
||||||
|
* @param channelSftp sftp连接对象
|
||||||
|
*/
|
||||||
|
public void returnObject(ChannelSftp channelSftp) {
|
||||||
|
if (channelSftp!=null) {
|
||||||
|
pool.returnObject(channelSftp);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue