Vertx 接入Mysql数据库 (六)

项目github地址：https://github.com/fotocj007/VertxWebApi

前面的项目实现了一个httpserver，上传返回的消息都是json格式。现在开始实现mysql，redis，mongodb的使用。关于mysql,redis,mongodb服务器的创建这里不做详解。

一：添加：  gradle.compile group: 'io.vertx', name: 'vertx-mysql-client', version: '3.9.8'

二：新建mysql连接json。在项目外建立资源文件夹，这样就不需要打到包里面去。

{
  "config": [
    {
      "url": "127.0.0.1",
      "port": 3306,
      "user": "root",
      "password": "root123456",
      "max_pool_size": 20,
      "max_idle_time": 1800
    }
  ],
  "configDb": "demo_db"
}

三：创建json文件加载帮助类

public class ConfigUtil {
    public static JsonObject loadJsonConfig(Vertx vertx, String path) {
        FileSystem fs = vertx.fileSystem();
        if (!fs.propsBlocking(path).isDirectory()) {
            Buffer buffer = fs.readFileBlocking(path);
            if (isJsonArray(buffer)) {
                JsonArray array = new JsonArray(buffer);
                JsonObject ob = new JsonObject();
                ob.put("__CONFIGS__", array);
                return ob;
            } else {
                return new JsonObject(buffer);
            }

        }
        return new JsonObject();
    }

    private static boolean isJsonArray(Buffer buffer) {
        return buffer.getByte(0) == "[".getBytes()[0];
    }
}

public abstract class JsonObjectConfig {
    private final String path;

    public JsonObjectConfig(Vertx vertx, String path){
        this.path = path;
        loadJson(vertx);
    }

    private void loadJson(Vertx vertx){
        JsonObject configJson = ConfigUtil.loadJsonConfig(vertx, path);
        decode(configJson);
    }

    private void decode(JsonObject jsonObject) {
        parse(jsonObject);
    }

    public abstract void parse(JsonObject jsonObject);
}

四：mysql json文件加载

public class MysqlConfig extends JsonObjectConfig {
    public JsonArray configs;

    public String configDbName;

    public MysqlConfig(Vertx vertx, String path){
        super(vertx,path);
    }

    @Override
    public void parse(JsonObject jsonObject) {
        configs = jsonObject.getJsonArray("config");
        configDbName = jsonObject.getString("configDb");
    }
}

五：创建mysql连接

public class MySQLUtil {
    private final Logger logger = LoggerFactory.getLogger(MySQLUtil.class);

    private final Vertx vertx;

    private final int poolSize;

    private final List<JsonObject> dataConfig;

    private List<MySQLPool> clients;

    public MySQLUtil(Vertx vertx, int poolSize, List<JsonObject> dataConfig){
        this.vertx = vertx;
        this.poolSize = poolSize;
        this.dataConfig = dataConfig;

        initPool();
    }

    private void initPool(){
        clients = new ArrayList<>();
        for (JsonObject dataSource : dataConfig) {
            for (int j = 0; j < poolSize; j++) {
                MySQLConnectOptions connectOptions = new MySQLConnectOptions()
                        .setPort(dataSource.getInteger("port"))
                        .setHost(dataSource.getString("url"))
                        .setUser(dataSource.getString("user"))
                        .setPassword(dataSource.getString("password"))
                        .setCharset("utf8")
                        .setCollation("utf8_general_ci")
                        .setReconnectAttempts(3)//连接无法建立时重试
                        .setReconnectInterval(1000);

                // 连接池选项
                PoolOptions poolOptions = new PoolOptions()
                        .setMaxSize(dataSource.getInteger("max_pool_size"))
                        .setIdleTimeout(dataSource.getInteger("max_idle_time"));

                // 创建带连接池的客户端
                MySQLPool client = MySQLPool.pool(vertx,connectOptions, poolOptions);

                clients.add(client);
            }
        }
    }

    public MySQLPool getConfigClient(){
        return clients.get(ThreadLocalRandom.current().nextInt(clients.size()));
    }

    public void close(){
        for(MySQLPool client : clients){
            client.close();
        }

        logger.warn("mysql 关闭连接池---");
    }
}

六：创建mysql查询帮助类

public class PlayerDao {
    protected Logger logger = LoggerFactory.getLogger(PlayerDao.class);

    protected String DB_SPLIT = "";
    protected MySQLUtil mySQLPool;

    public PlayerDao(String DB_SPLIT, MySQLUtil mySQLPool) {
        this.DB_SPLIT = DB_SPLIT;
        this.mySQLPool = mySQLPool;
    }

    /*************************
     * 查询数据
     * 根据 实体类T获取数据并实例化
     */
    public <T> void queryConfigList(String sql, Class<T> classes, Handler<AsyncResult<List<T>>> handler){
        mySQLPool.getConfigClient().query(sql)
                .execute(qRes -> {
                    if(qRes.succeeded()){
                        List<T> lists = new ArrayList<>();

                        RowSet<Row> result = qRes.result();
                        List<String> col = qRes.result().columnsNames();

                        for (Row row : result) {
                            JsonObject json = new JsonObject();
                            for (String str : col) {
                                json.put(str,row.getValue(str));
                            }
                            T entity = new JsonObject(json.toString()).mapTo(classes);
                            lists.add(entity);
                        }

                        handler.handle(Future.succeededFuture(lists));
                    }else {
                        handler.handle(Future.failedFuture(qRes.cause()));
                        logger.error("--error queryConfigList----- " + sql, qRes.cause());
                    }
                });
    }
}

七：创建mysql管理类，管理查询帮助类。

public class DaoManager {
    private final MysqlConfig mysqlConfig;

    private final MySQLUtil mySQLPool;

    private PlayerDao playerDao;

    public DaoManager(MysqlConfig mysqlConfig, MySQLUtil mySQLPool){
        this.mysqlConfig = mysqlConfig;
        this.mySQLPool = mySQLPool;

        init();
    }

    private void init(){
        playerDao = new PlayerDao(mysqlConfig.configDbName,mySQLPool);
    }

    public PlayerDao getPlayerDao(){return playerDao;}
}

八：修改Configure配置文件，加载json和初始化mysql

public class Configure {
    private static final Configure ourInstance = new Configure();

    public static Configure getInstance() {
        return ourInstance;
    }

    protected Vertx vertx;

    public MysqlConfig mysqlConfig;
    private MySQLUtil mySQLPool;
    public DaoManager daoManager;

    public void init(Vertx vertx){
        this.vertx = vertx;

        initHandler();

        loadConfig();

        initDb();
    }

    private void initHandler(){
        HandlerManager.getInstance().addHandler(new DemoHandler());
    }

    /**
     *  加载db和Redis配置文件
     */
    protected void loadConfig(){
        mysqlConfig = new MysqlConfig(vertx, "res/mysql.json");
    }

    protected void initDb(){
        List<JsonObject> list = new ArrayList<>();
        for(int i = 0; i< mysqlConfig.configs.size();i++){
            list.add(mysqlConfig.configs.getJsonObject(i));
        }
        mySQLPool = new MySQLUtil(vertx,2,list);

        daoManager = new DaoManager(mysqlConfig,mySQLPool);
    }
}

九：修改启动类

十：测试一下

1：新建数据库表

2：新建个实体类 PlayerInfo

public class PlayerInfo {
    private long id;

    private String userName;

    private int age;

    public long getId() {
        return id;
    }

    public void setId(long id) {
        this.id = id;
    }

    public String getUserName() {
        return userName;
    }

    public void setUserName(String userName) {
        this.userName = userName;
    }

    public int getAge() {
        return age;
    }

    public void setAge(int age) {
        this.age = age;
    }
}

3：修改DemoHandler，重新数据库

public class DemoHandler implements InterHandler {
    @Override
    public void handler(AbstractUpMessage up, HttpServerResponse resp) {
        //上传参数
        DemoRequest request = (DemoRequest)up;
        System.out.println("上传参数:"+ request.name + "-" + request.age);


 9         String sql = "select * from " + Configure.getInstance().mysqlConfig.configDbName + ".player_info ";
10         PlayerDao client = Configure.getInstance().daoManager.getPlayerDao();
11         client.queryConfigList(sql, PlayerInfo.class, res -> {
12             List<PlayerInfo> lists = res.result();
13             for(PlayerInfo item : lists){
14                 System.out.println(item.getUserName()+"---" + item.getAge());
15             }
16         });


        //返回数据
        String n = "cscscs---";
        String in = "info ---";
        //编码返回json
        DemoResponse response = new DemoResponse(getMessageId(),n,in);
        response.encode();
        resp.end(response.SendMessage());
    }

    @Override
    public short getMessageId() {
        return HandlerCode.DEMO_V1;
    }
}

4：运行，调用接口试试

项目结构