You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
63 lines
2.4 KiB
63 lines
2.4 KiB
package com.example.sshd.service;
|
|
|
|
import java.sql.ResultSet;
|
|
import java.sql.SQLException;
|
|
import java.util.Date;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.jdbc.core.JdbcTemplate;
|
|
import org.springframework.jdbc.core.RowMapper;
|
|
import org.springframework.stereotype.Service;
|
|
|
|
import jakarta.annotation.PostConstruct;
|
|
|
|
@Service
|
|
public class JdbcService {
|
|
|
|
private static final String createRemoteIpLookupTableSql = "CREATE TABLE IF NOT EXISTS public.remote_ip_lookup (id BIGINT not null, "
|
|
+ "remote_ip_address CHARACTER VARYING not null, remote_ip_info CHARACTER VARYING not null, PRIMARY KEY (id));";
|
|
private static final String createRemoteIpLookupIndexSql = "CREATE INDEX IF NOT EXISTS public.remote_ip_lookup_idx ON "
|
|
+ "public.remote_ip_lookup (remote_ip_address);";
|
|
|
|
@Autowired
|
|
private JdbcTemplate jdbcTemplate;
|
|
|
|
@PostConstruct
|
|
private void init() {
|
|
jdbcTemplate.execute(createRemoteIpLookupTableSql);
|
|
jdbcTemplate.execute(createRemoteIpLookupIndexSql);
|
|
}
|
|
|
|
public List<Map<String, Object>> getRemoteIpInfo(String remoteIp) {
|
|
return jdbcTemplate.query(
|
|
"SELECT id, remote_ip_address, remote_ip_info from public.remote_ip_lookup WHERE remote_ip_address = ? ",
|
|
new RowMapper<Map<String, Object>>() {
|
|
@Override
|
|
public Map<String, Object> mapRow(ResultSet rs, int rowNum) throws SQLException {
|
|
return Map.of("insert_time", new Date(rs.getLong(1)), "remote_ip_address", rs.getString(2),
|
|
"remote_ip_info", rs.getString(3));
|
|
}
|
|
}, remoteIp);
|
|
}
|
|
|
|
public List<Map<String, Object>> getAllRemoteIpInfo() {
|
|
return jdbcTemplate.query(
|
|
"SELECT id, remote_ip_address, remote_ip_info from public.remote_ip_lookup order by id",
|
|
new RowMapper<Map<String, Object>>() {
|
|
@Override
|
|
public Map<String, Object> mapRow(ResultSet rs, int rowNum) throws SQLException {
|
|
return Map.of("insert_time", new Date(rs.getLong(1)), "remote_ip_address", rs.getString(2),
|
|
"remote_ip_info", rs.getString(3));
|
|
}
|
|
});
|
|
}
|
|
|
|
public int insertRemoteIpInfo(String remoteIpAddress, String remoteIpInfo) {
|
|
return jdbcTemplate.update(
|
|
"INSERT INTO public.remote_ip_lookup (id, remote_ip_address, remote_ip_info) VALUES (?, ?, ?)",
|
|
System.currentTimeMillis(), remoteIpAddress, remoteIpInfo);
|
|
}
|
|
}
|