health

zhuzhengping911 · zhuzhengping911 · commit 24680aecef5d · 2017-03-25T02:59:00.000+08:00
diff --git a/src/main/java/com/health/CusDiskSpaceHealthIndicator.java b/src/main/java/com/health/CusDiskSpaceHealthIndicator.java
@@ -0,0 +1,45 @@
+package com.health;
+
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.boot.actuate.health.AbstractHealthIndicator;
+import org.springframework.boot.actuate.health.Health;
+import org.springframework.stereotype.Component;
+
+import java.io.IOException;
+import java.nio.file.FileStore;
+import java.nio.file.Files;
+import java.nio.file.Paths;
+
+/**
+ * Created by zhuzhengping on 2017/3/25.
+ */
+@Component
+public class CusDiskSpaceHealthIndicator extends AbstractHealthIndicator {
+
+    private final FileStore fileStore;
+    private final long thresholdBytes;
+
+    @Autowired
+    public CusDiskSpaceHealthIndicator(
+        @Value("${health.filestore.path:/}") String  path,
+        @Value("${health.filestore.bytes:10485760}") long thresholdBytes
+    ) throws IOException {
+        fileStore = Files.getFileStore(Paths.get(path));
+        this.thresholdBytes = thresholdBytes;
+    }
+
+    @Override
+    protected void doHealthCheck(Health.Builder builder) throws Exception {
+        long diskFreeInBytes = fileStore.getUnallocatedSpace();
+        if(diskFreeInBytes >= thresholdBytes){
+            builder.up();
+        } else{
+            builder.down();
+        }
+
+        long totalSpaceInBytes = fileStore.getTotalSpace();
+        builder.withDetail("disk.free",diskFreeInBytes);
+        builder.withDetail("disk.totol",totalSpaceInBytes);
+    }
+}
diff --git a/src/main/java/com/health/CusStatusHealthIndicator.java b/src/main/java/com/health/CusStatusHealthIndicator.java
@@ -0,0 +1,29 @@
+package com.health;
+
+import org.springframework.boot.actuate.health.Health;
+import org.springframework.boot.actuate.health.HealthIndicator;
+import org.springframework.http.HttpStatus;
+import org.springframework.stereotype.Component;
+
+/**
+ * Created by zhuzhengping on 2017/3/25.
+ */
+@Component
+public class CusStatusHealthIndicator implements HealthIndicator {
+
+    @Override
+    public Health health() {
+        int errCode = check();
+        if(errCode != 0){
+            return Health.down()
+                    .withDetail("status",errCode)
+                    .withDetail("message","http error")
+                    .build();
+        }
+        return Health.up().build();
+    }
+
+    private int check(){
+        return HttpStatus.NOT_FOUND.value();
+    }
+}
