sharding-jdbc + druid 实现mysql 主备份

功能需求

master 正常时所有请求router到master;
master不正常时，DQL路由到slaves(slaves 会进行负载均衡,算法自定),DML DDL等抛异常;

直接贴代码简单点

Demo.java

package com.yiwugou.demo.sharding;

import java.sql.SQLException;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.sql.DataSource;

import org.springframework.jdbc.core.JdbcTemplate;

import com.alibaba.druid.filter.Filter;
import com.alibaba.druid.pool.DruidDataSource;

import io.shardingsphere.core.api.MasterSlaveDataSourceFactory;
import io.shardingsphere.core.api.config.MasterSlaveRuleConfiguration;

public class Demo {
    private static YiwugouConfig config = new YiwugouConfig();

    public static void main(String[] args) throws Exception {

        DataSource masterSlaveDataSource = initMasterSlaveDataSource();
        DataSource dataSource = initYiwugouDataSource(masterSlaveDataSource);
        JdbcTemplate jdbcTemplate = initJdbcTemplate(dataSource);

        for (int i = 0; i <= 100000; i++) {
            try {
                List<Map<String, Object>> objs = jdbcTemplate.queryForList("select * from T_DEMO");
                System.err.println(i + "=" + objs);

                Thread.sleep(1000L);
            } catch (Exception e) {
                e.printStackTrace();
            }
            // try {
            // jdbcTemplate.update("insert into T_DEMO (name) values ('abcd" + i
            // + "')");
            // System.err.println("execute=" + i);
            // Thread.sleep(1000L);
            // } catch (Exception e) {
            // e.printStackTrace();
            // }
        }
    }

    public static DataSource initYiwugouDataSource(DataSource dataSource) {
        YiwugouDataSource ds = new YiwugouDataSource(dataSource, config);
        return ds;
    }

    public static DataSource initMasterSlaveDataSource() throws SQLException {
        DataSource ds_72 = initDataSource("com.mysql.jdbc.Driver", "jdbc:mysql://10.6.2.72:3306/demo", "root",
                "admin123");
        DataSource ds_127 = initDataSource("com.mysql.jdbc.Driver", "jdbc:mysql://127.0.0.1:3306/demo", "root", "");

        Map<String, DataSource> dataSourceMap = new HashMap<>();
        dataSourceMap.put("ds_master", ds_127);
        dataSourceMap.put("ds_slave_0", ds_72);
        // dataSourceMap.put("ds_slave_1", slaveDataSource1);

        MasterSlaveRuleConfiguration masterSlaveRuleConfig = new MasterSlaveRuleConfiguration("ds_master_slave",
                "ds_master", Arrays.asList("ds_slave_0"));

        DataSource dataSource = MasterSlaveDataSourceFactory.createDataSource(dataSourceMap, masterSlaveRuleConfig,
                new HashMap<String, Object>());
        return dataSource;
    }

    public static DataSource initDataSource(String driver, String url, String username, String password) {
        DruidDataSource dataSource = new DruidDataSource();
        dataSource.setDriverClassName(driver);
        dataSource.setUrl(url);
        dataSource.setUsername(username);
        dataSource.setPassword(password);
        dataSource.setMaxActive(100);
        dataSource.setMinIdle(5);

        dataSource.setFailFast(true); // 重要 不然会卡住
        Filter yiwugouFilter = new YiwugouFilter(new Runnable() {
            @Override
            public void run() {
                config.setAlived(false);
            }
        });
        dataSource.setProxyFilters(Arrays.asList(yiwugouFilter));
        // dataSource.setBreakAfterAcquireFailure(true);
        // dataSource.setAsyncInit(true);
        return dataSource;
    }

    public static JdbcTemplate initJdbcTemplate(DataSource dataSource) {
        JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);
        return jdbcTemplate;
    }

}

YiwugouConfig.java

package com.yiwugou.demo.sharding;

public class YiwugouConfig {
    private boolean isAlived = true;

    public boolean isAlived() {
        return isAlived;
    }

    public void setAlived(boolean isAlived) {
        this.isAlived = isAlived;
    }
}

YiwugouDataSource

package com.yiwugou.demo.sharding;

import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.SQLException;
import java.sql.SQLFeatureNotSupportedException;
import java.util.logging.Logger;

import javax.sql.DataSource;

import io.shardingsphere.core.api.HintManager;

public class YiwugouDataSource implements DataSource {
    private DataSource dataSource;

    private YiwugouConfig config;

    public YiwugouDataSource(DataSource dataSource, YiwugouConfig config) {
        this.dataSource = dataSource;
        this.config = config;
    }

    @Override
    public Connection getConnection() throws SQLException {
        if (config.isAlived()) {
            HintManager hintManager = HintManager.getInstance();
            hintManager.setMasterRouteOnly();
        }
        Connection con = this.getDataSource().getConnection();
        return con;

    }

    @Override
    public Connection getConnection(String username, String password) throws SQLException {
        if (config.isAlived()) {
            HintManager hintManager = HintManager.getInstance();
            hintManager.setMasterRouteOnly();
        }
        Connection con = this.getDataSource().getConnection(username, password);
        return con;
    }

    private DataSource getDataSource() {
        return this.dataSource;
    }

    @Override
    public void setLoginTimeout(int timeout) throws SQLException {
        this.getDataSource().setLoginTimeout(timeout);
    }

    @Override
    public PrintWriter getLogWriter() throws SQLException {
        return this.getDataSource().getLogWriter();
    }

    @Override
    public void setLogWriter(PrintWriter pw) throws SQLException {
        this.getDataSource().setLogWriter(pw);
    }

    @Override
    public <T> T unwrap(Class<T> iface) throws SQLException {
        return this.getDataSource().unwrap(iface);
    }

    @Override
    public boolean isWrapperFor(Class<?> iface) throws SQLException {
        return this.getDataSource().isWrapperFor(iface);
    }

    @Override
    public Logger getParentLogger() throws SQLFeatureNotSupportedException {
        return this.getDataSource().getParentLogger();
    }

    @Override
    public int getLoginTimeout() throws SQLException {
        return this.getDataSource().getLoginTimeout();
    }
}

YiwugouFilter.java

package com.yiwugou.demo.sharding;

import java.sql.SQLException;

import com.alibaba.druid.filter.FilterAdapter;
import com.alibaba.druid.filter.FilterChain;
import com.alibaba.druid.proxy.jdbc.PreparedStatementProxy;
import com.alibaba.druid.proxy.jdbc.ResultSetProxy;
import com.alibaba.druid.proxy.jdbc.StatementProxy;

public class YiwugouFilter extends FilterAdapter {
    public YiwugouFilter(Runnable runnable) {
        this.runnable = runnable;
    }

    private Runnable runnable;

    @Override
    public void preparedStatement_addBatch(FilterChain chain, PreparedStatementProxy statement) throws SQLException {
        try {
            super.preparedStatement_addBatch(chain, statement);
        } catch (Exception e) {
            runnable.run();
            throw e;
        }
    }

    @Override
    public boolean preparedStatement_execute(FilterChain chain, PreparedStatementProxy statement) throws SQLException {
        try {
            return super.preparedStatement_execute(chain, statement);
        } catch (Exception e) {
            runnable.run();
            throw e;
        }
    }

    @Override
    public ResultSetProxy preparedStatement_executeQuery(FilterChain chain, PreparedStatementProxy statement)
            throws SQLException {
        try {
            return super.preparedStatement_executeQuery(chain, statement);
        } catch (Exception e) {
            runnable.run();
            throw e;
        }
    }

    @Override
    public int preparedStatement_executeUpdate(FilterChain chain, PreparedStatementProxy statement)
            throws SQLException {
        try {
            return super.preparedStatement_executeUpdate(chain, statement);
        } catch (Exception e) {
            runnable.run();
            throw e;
        }
    }

    @Override
    public void statement_addBatch(FilterChain chain, StatementProxy statement, String sql) throws SQLException {
        try {
            super.statement_addBatch(chain, statement, sql);
        } catch (Exception e) {
            runnable.run();
            throw e;
        }
    }

    @Override
    public boolean statement_execute(FilterChain chain, StatementProxy statement, String sql) throws SQLException {
        try {
            return super.statement_execute(chain, statement, sql);
        } catch (Exception e) {
            runnable.run();
            throw e;
        }
    }

    @Override
    public boolean statement_execute(FilterChain chain, StatementProxy statement, String sql, int autoGeneratedKeys)
            throws SQLException {
        try {
            return super.statement_execute(chain, statement, sql, autoGeneratedKeys);
        } catch (Exception e) {
            runnable.run();
            throw e;
        }
    }

    @Override
    public boolean statement_execute(FilterChain chain, StatementProxy statement, String sql, int[] columnIndexes)
            throws SQLException {
        try {
            return super.statement_execute(chain, statement, sql, columnIndexes);
        } catch (Exception e) {
            runnable.run();
            throw e;
        }
    }

    @Override
    public boolean statement_execute(FilterChain chain, StatementProxy statement, String sql, String[] columnNames)
            throws SQLException {
        try {
            return super.statement_execute(chain, statement, sql, columnNames);
        } catch (Exception e) {
            runnable.run();
            throw e;
        }
    }

    @Override
    public int[] statement_executeBatch(FilterChain chain, StatementProxy statement) throws SQLException {
        try {
            return super.statement_executeBatch(chain, statement);
        } catch (Exception e) {
            runnable.run();
            throw e;
        }
    }

    @Override
    public ResultSetProxy statement_executeQuery(FilterChain chain, StatementProxy statement, String sql)
            throws SQLException {
        try {
            return super.statement_executeQuery(chain, statement, sql);
        } catch (Exception e) {
            runnable.run();
            throw e;
        }
    }

    @Override
    public int statement_executeUpdate(FilterChain chain, StatementProxy statement, String sql) throws SQLException {
        try {
            return super.statement_executeUpdate(chain, statement, sql);
        } catch (Exception e) {
            runnable.run();
            throw e;
        }
    }

    @Override
    public int statement_executeUpdate(FilterChain chain, StatementProxy statement, String sql, int autoGeneratedKeys)
            throws SQLException {
        try {
            return super.statement_executeUpdate(chain, statement, sql, autoGeneratedKeys);
        } catch (Exception e) {
            runnable.run();
            throw e;
        }
    }

    @Override
    public int statement_executeUpdate(FilterChain chain, StatementProxy statement, String sql, int[] columnIndexes)
            throws SQLException {
        try {
            return super.statement_executeUpdate(chain, statement, sql, columnIndexes);
        } catch (Exception e) {
            runnable.run();
            throw e;
        }
    }

    @Override
    public int statement_executeUpdate(FilterChain chain, StatementProxy statement, String sql, String[] columnNames)
            throws SQLException {
        try {
            return super.statement_executeUpdate(chain, statement, sql, columnNames);
        } catch (Exception e) {
            runnable.run();
            throw e;
        }
    }

}

人面猴
序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
沈念sama阅读 204,293评论 6赞 478
死咒
序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
沈念sama阅读 85,604评论 2赞 381
救了他两次的神仙让他今天三更去死
文/潘晓璐我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
开封第一讲书人阅读 150,958评论 0赞 337
道士缉凶录：失踪的卖姜人
文/不坏的土叔我叫张陵，是天一观的道长。经常有香客问我，道长，这世上最难降的妖魔是什么？我笑而不...
开封第一讲书人阅读 54,729评论 1赞 277
港岛之恋（遗憾婚礼）
正文为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
茶点故事阅读 63,719评论 5赞 366
恶毒庶女顶嫁案：这布局不是一般人想出来的
文/花漫我一把揭开白布。她就那样静静地躺着，像睡着了一般。火红的嫁衣衬着肌肤如雪。梳的纹丝不乱的头发上，一...
开封第一讲书人阅读 48,630评论 1赞 281
城市分裂传说
那天，我揣着相机与录音，去河边找鬼。笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。我是一名探鬼主播，决...
沈念sama阅读 38,000评论 3赞 397
双鸳鸯连环套：你想象不到人心有多黑
文/苍兰香墨我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
开封第一讲书人阅读 36,665评论 0赞 258
万荣杀人案实录
序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
沈念sama阅读 40,909评论 1赞 299
护林员之死
正文独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋以下内容为张勋视角年9月15日...
茶点故事阅读 35,646评论 2赞 321
白月光启示录
正文我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
茶点故事阅读 37,726评论 1赞 330
活死人
序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
沈念sama阅读 33,400评论 4赞 321
日本核电站爆炸内幕
正文年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
茶点故事阅读 38,986评论 3赞 307
男人毒药：我在死后第九天来索命
文/蒙蒙一、第九天我趴在偏房一处隐蔽的房顶上张望。院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
开封第一讲书人阅读 29,959评论 0赞 19
一桩弑父案，背后竟有这般阴谋
文/苍兰香墨我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。一阵脚步声响...
开封第一讲书人阅读 31,197评论 1赞 260
情欲美人皮
我被黑心中介骗来泰国打工，没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。一个月前我还...
沈念sama阅读 44,996评论 2赞 349
代替公主和亲
正文我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
茶点故事阅读 42,481评论 2赞 342

sharding-jdbc + druid 实现mysql 主备份

功能需求

直接贴代码 简单点

Demo.java

YiwugouConfig.java

YiwugouDataSource

YiwugouFilter.java

推荐阅读更多精彩内容

直接贴代码简单点