ELK for User Service

Author: DucBaoUIT

.gitignore

@@ -31,3 +31,4 @@ build/
 
 ### VS Code ###
 .vscode/
+#logback-*.xml

pom.xml

@@ -116,6 +116,11 @@
 		</dependency>
 
 
+		<dependency>
+    		<groupId>net.logstash.logback</groupId>
+    		<artifactId>logstash-logback-encoder</artifactId>
+    		<version>7.0.1</version>
+		</dependency>
 
 
 

src/main/java/com/programming/userService/controller/UserController.java

@@ -22,6 +22,8 @@
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.Paths;
+
+import org.slf4j.MDC;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.cache.annotation.Cacheable;
 import org.springframework.mail.MailException;
@@ -36,6 +38,9 @@
 import com.programming.userService.entity.AuthUser;
 import org.springframework.data.redis.core.RedisTemplate;
 
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
 @RestController
 @AllArgsConstructor
 @RequestMapping("/user")
@@ -45,6 +50,8 @@ public class UserController {
     private final PasswordEncoder passwordEncoder;
     private final RedisTemplate<String, Object> redisTemplate;
     private static final String USER_CACHE = "USER_CACHE";
+
+    private static final Logger logger = LoggerFactory.getLogger(UserController.class);
     @Autowired
     private JwtUtil jwtUtil;
 
@@ -60,6 +67,9 @@ public String getEmail() {
     }
     @GetMapping("/")
     public String getServiceName(){
+        //ELK 
+        MDC.put("type", "userservice");
+        logger.info("User Service Start");
         return "User Service";
     }
 
@@ -92,6 +102,11 @@ public ResponseEntity registerUser(@RequestBody AuthUser user) {
             user.setTimestamp(new Date());
             user.setAvatar(getDefaultAvatar());
             AuthUser save = userRepository.save(user);
+            AuthUser userFromDb = userRepository.findByUsername(user.getUsername())
+                    .orElseThrow(() -> new Exception("User not found"));
+            MDC.put("type", "userservice");
+            MDC.put("action", "register");
+            logger.info("UserID: " + userFromDb.getId());
             return ResponseEntity.ok(save);
         } catch (Exception e) {
             return ResponseEntity.internalServerError().body(e.getMessage());
@@ -132,6 +147,9 @@ public ResponseEntity loginUser2(@RequestBody AuthUser user) {
             AuthUser userFromDb = userRepository.findByUsername(user.getUsername())
                     .orElseThrow(() -> new Exception("User not found"));
             if (passwordEncoder.matches(user.getPassword(), userFromDb.getPassword())) {
+                MDC.put("type", "userservice");
+                MDC.put("action", "login");
+                logger.info("UserID: " + userFromDb.getId());
                 return ResponseEntity.ok(userFromDb);
             } else {
                 return ResponseEntity.status(HttpStatus.UNAUTHORIZED).body("Invalid credentials");
@@ -203,7 +221,12 @@ public ResponseEntity updateProfile(@PathVariable("id") String id, @RequestBody
             userFromDb.setFirstName(user.getFirstName());
             userFromDb.setLastName(user.getLastName());
             AuthUser save = userRepository.save(userFromDb);
+            String tempUserId = userFromDb.getId();
 
+            MDC.put("type", "userservice");
+            MDC.put("action", "update-profile");
+            logger.info("UserID: " + userFromDb.getId());
+            
             return ResponseEntity.ok(HttpStatus.OK);
         } catch (Exception e) {
             return ResponseEntity.internalServerError().body(e.getMessage());
@@ -224,6 +247,10 @@ public ResponseEntity changePassword(@PathVariable("id") String id,
             userFromDb.setPassword(passwordEncoder.encode(changePasswordRequest.getNewPassword()));
             AuthUser save = userRepository.save(userFromDb);
 
+            MDC.put("type", "userservice");
+            MDC.put("action", "change-password");
+            logger.info("UserID: " + userFromDb.getId());
+
             return ResponseEntity.ok("Password changed successfully");
         } catch (Exception e) {
             return ResponseEntity.internalServerError().body(e.getMessage());

src/main/resources/logback-spring.xml

@@ -0,0 +1,11 @@
+<configuration>
+    <appender name="LOGSTASH" class="net.logstash.logback.appender.LogstashTcpSocketAppender">
+        <destination>192.168.120.213:5050</destination>
+        <encoder class="net.logstash.logback.encoder.LogstashEncoder">
+        </encoder>
+    </appender>
+
+    <root level="INFO">
+        <appender-ref ref="LOGSTASH"/>
+    </root>
+</configuration>