gsd
1 year ago
19 changed files with 194 additions and 19 deletions
@ -0,0 +1,41 @@ |
|||
package app.annotations.impl; |
|||
|
|||
import app.entities.db.CollectableStatistic; |
|||
import app.services.db.CollectStatisticService; |
|||
import jakarta.servlet.http.HttpServletRequest; |
|||
import jakarta.servlet.http.HttpServletResponse; |
|||
import org.aspectj.lang.JoinPoint; |
|||
import org.aspectj.lang.annotation.After; |
|||
import org.aspectj.lang.annotation.AfterReturning; |
|||
import org.aspectj.lang.annotation.Aspect; |
|||
import org.aspectj.lang.annotation.Before; |
|||
import org.slf4j.Logger; |
|||
import org.slf4j.LoggerFactory; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.context.annotation.Configuration; |
|||
|
|||
import java.util.Arrays; |
|||
|
|||
@Aspect |
|||
@Configuration |
|||
public class CollectStatisticAspect { |
|||
|
|||
private final Logger logger = LoggerFactory.getLogger(this.getClass()); |
|||
|
|||
private CollectStatisticService collectStatisticService; |
|||
|
|||
private HttpServletRequest request; |
|||
|
|||
@Autowired |
|||
public CollectStatisticAspect(CollectStatisticService collectStatisticService, HttpServletRequest request) { |
|||
this.collectStatisticService = collectStatisticService; |
|||
this.request = request; |
|||
} |
|||
|
|||
@Before("@annotation(app.annotations.interfaces.CollectStatistic)") |
|||
public void before() { |
|||
CollectableStatistic st = new CollectableStatistic(request); |
|||
collectStatisticService.add(st); |
|||
logger.info(st.toString()); |
|||
} |
|||
} |
|||
@ -0,0 +1,11 @@ |
|||
package app.annotations.interfaces; |
|||
|
|||
import java.lang.annotation.ElementType; |
|||
import java.lang.annotation.Retention; |
|||
import java.lang.annotation.RetentionPolicy; |
|||
import java.lang.annotation.Target; |
|||
|
|||
@Retention(RetentionPolicy.RUNTIME) |
|||
@Target(ElementType.METHOD) |
|||
public @interface CollectStatistic { |
|||
} |
|||
@ -0,0 +1,52 @@ |
|||
package app.entities.db; |
|||
|
|||
import jakarta.servlet.http.HttpServletRequest; |
|||
import lombok.AllArgsConstructor; |
|||
import lombok.Data; |
|||
|
|||
import java.sql.Timestamp; |
|||
import java.time.Instant; |
|||
import java.util.Arrays; |
|||
|
|||
@Data |
|||
public class CollectableStatistic { |
|||
Long id; |
|||
Timestamp timestamp; |
|||
///
|
|||
String client_ip; |
|||
String steam64; |
|||
///
|
|||
String method; |
|||
String path; |
|||
String query; |
|||
///
|
|||
String useragent; |
|||
|
|||
public CollectableStatistic(HttpServletRequest request) { |
|||
this.client_ip = request.getHeader("X-Forwarded-For"); |
|||
this.method = request.getMethod(); |
|||
this.path = request.getRequestURI(); |
|||
this.query = request.getQueryString(); |
|||
this.steam64 = getCookie(request, "steam64"); |
|||
this.useragent = request.getHeader("User-Agent"); |
|||
this.timestamp = Timestamp.from(Instant.now()); |
|||
} |
|||
|
|||
@Override |
|||
public String toString() { |
|||
return |
|||
"Time: " + timestamp.toString() + "\n" + |
|||
"IP: " + client_ip + "\n" + |
|||
"Steam64: " + steam64 + "\n" + |
|||
"Method: " + method + "\n" + |
|||
"Path: " + path + "\n" + |
|||
"Query: " + query + "\n" + |
|||
"UA: " + useragent + "\n"; |
|||
} |
|||
|
|||
private String getCookie(HttpServletRequest request, String cookie_name) { |
|||
if (request.getHeader("Cookie") == null) return null; |
|||
String[] rawCookieParams = request.getHeader("Cookie").split(";"); |
|||
return Arrays.stream(rawCookieParams).filter(s -> s.contains(cookie_name + "=")).map(s -> s.split(cookie_name + "=")[1]).findFirst().orElse(null); |
|||
} |
|||
} |
|||
@ -0,0 +1,38 @@ |
|||
package app.services.db; |
|||
|
|||
import app.entities.db.CollectableStatistic; |
|||
import app.updates.BaseUpdater; |
|||
import jakarta.annotation.PostConstruct; |
|||
import jakarta.persistence.EntityManager; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.beans.factory.annotation.Qualifier; |
|||
import org.springframework.stereotype.Service; |
|||
import org.springframework.transaction.annotation.Transactional; |
|||
|
|||
import java.util.ArrayList; |
|||
import java.util.Queue; |
|||
import java.util.concurrent.ConcurrentLinkedQueue; |
|||
|
|||
@Service |
|||
@Transactional("RwTransactionManager") |
|||
public class CollectStatisticService extends BaseUpdater { |
|||
|
|||
private EntityManager entityManager; |
|||
|
|||
@Autowired |
|||
public CollectStatisticService(@Qualifier(value = "RwEntityManager") EntityManager entityManager) { |
|||
this.entityManager = entityManager; |
|||
} |
|||
|
|||
@Transactional("RwTransactionManager") |
|||
public void add(CollectableStatistic collectableStatistic) { |
|||
entityManager.createNativeQuery("INSERT INTO `web_statistic` (`steam64`, `client_ip`, `method`, `path`, `query`, `useragent`) VALUES (?1, ?2, ?3, ?4, ?5, ?6)") |
|||
.setParameter(1, collectableStatistic.getSteam64()) |
|||
.setParameter(2, collectableStatistic.getClient_ip()) |
|||
.setParameter(3, collectableStatistic.getMethod()) |
|||
.setParameter(4, collectableStatistic.getPath()) |
|||
.setParameter(5, collectableStatistic.getQuery()) |
|||
.setParameter(6, collectableStatistic.getUseragent()) |
|||
.executeUpdate(); |
|||
} |
|||
} |
|||
Loading…
Reference in new issue