com.yammer.dropwizard.jdbi.DBIHealthCheck Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dropwizard-jdbi Show documentation

The newest version!

package com.yammer.dropwizard.jdbi;

import com.yammer.metrics.core.HealthCheck;
import org.skife.jdbi.v2.DBI;
import org.skife.jdbi.v2.Handle;

public class DBIHealthCheck extends HealthCheck {
    private final DBI dbi;
    private final String validationQuery;

    public DBIHealthCheck(DBI dbi, String name, String validationQuery) {
        super(name + "-db");
        this.dbi = dbi;
        this.validationQuery = validationQuery;
    }

    @Override
    protected Result check() throws Exception {
        final Handle handle = dbi.open();
        try {
            handle.execute(validationQuery);
        } finally {
            handle.close();
        }
        return Result.healthy();
    }
}