Unverified Commit bb8f7a74 authored by 张亮's avatar 张亮 Committed by GitHub
Browse files

Merge pull request #61 from tristaZero/dev 

Optimize the common repositpry in sharding-examples
parents c5dfe16d e52ec0db

common-repository/api-repository/src/main/java/io/shardingsphere/example/repository/api/entity/Order.java

+5 −1
Original line number Diff line number Diff line
@@ -17,7 +17,11 @@ 


package io.shardingsphere.example.repository.api.entity;



public class Order {

import java.io.Serializable;



public class Order implements Serializable {

    

    private static final long serialVersionUID = 661434701950670670L;

    

    private long orderId;

common-repository/api-repository/src/main/java/io/shardingsphere/example/repository/api/entity/OrderItem.java

+5 −1
Original line number Diff line number Diff line
@@ -17,7 +17,11 @@ 


package io.shardingsphere.example.repository.api.entity;



public class OrderItem {

import java.io.Serializable;



public class OrderItem implements Serializable {

    

    private static final long serialVersionUID = 263434701950670170L;

    

    private long orderItemId;

common-repository/mybatis-repository/src/main/java/io/shardingsphere/example/repository/mybatis/repository/OrderItemRepository.java→common-repository/api-repository/src/main/java/io/shardingsphere/example/repository/api/repository/OrderItemRepository.java

+1 −4
Original line number Diff line number Diff line
@@ -15,15 +15,12 @@ 
 * </p>

 */



package io.shardingsphere.example.repository.mybatis.repository;

package io.shardingsphere.example.repository.api.repository;



import io.shardingsphere.example.repository.api.entity.OrderItem;

import io.shardingsphere.example.repository.api.repository.CommonRepository;

import org.apache.ibatis.annotations.Mapper;



import java.util.List;



@Mapper

public interface OrderItemRepository extends CommonRepository<OrderItem> {

    

    @Override

common-repository/mybatis-repository/src/main/java/io/shardingsphere/example/repository/mybatis/repository/OrderRepository.java→common-repository/api-repository/src/main/java/io/shardingsphere/example/repository/api/repository/OrderRepository.java

+1 −4
Original line number Diff line number Diff line
@@ -15,15 +15,12 @@ 
 * </p>

 */



package io.shardingsphere.example.repository.mybatis.repository;

package io.shardingsphere.example.repository.api.repository;



import io.shardingsphere.example.repository.api.entity.Order;

import io.shardingsphere.example.repository.api.repository.CommonRepository;

import org.apache.ibatis.annotations.Mapper;



import java.util.List;



@Mapper

public interface OrderRepository extends CommonRepository<Order> {

    

    @Override

common-repository/jdbc-repository/src/main/java/io/shardingsphere/example/repository/jdbc/repository/RawOrderItemRepository.java

0 → 100644
+128 −0
Original line number Diff line number Diff line 
/*

 * Copyright 2016-2018 shardingsphere.io.

 * <p>

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *     http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 * </p>

 */



package io.shardingsphere.example.repository.jdbc.repository;



import io.shardingsphere.example.repository.api.entity.OrderItem;

import io.shardingsphere.example.repository.api.repository.OrderItemRepository;



import javax.sql.DataSource;

import java.sql.Connection;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.sql.Statement;

import java.util.LinkedList;

import java.util.List;



public final class RawOrderItemRepository implements OrderItemRepository {

    

    private final DataSource dataSource;

    

    public RawOrderItemRepository(final DataSource dataSource) {

        this.dataSource = dataSource;

    }

    

    @Override

    public void createTableIfNotExists() {

        execute("CREATE TABLE IF NOT EXISTS t_order_item (order_item_id BIGINT NOT NULL AUTO_INCREMENT, order_id BIGINT NOT NULL, user_id INT NOT NULL, status VARCHAR(50), PRIMARY KEY (order_item_id))");

    }

    

    @Override

    public void dropTable() {

        execute("DROP TABLE t_order_item");

    }

    

    @Override

    public void truncateTable() {

        execute("truncate table t_order_item");

    }

    

    @Override

    public Long insert(final OrderItem orderItem) {

        long orderItemId = -1;

        try (Connection connection = dataSource.getConnection();

             Statement statement = connection.createStatement()) {

            orderItemId = insertAndGetGeneratedKey(statement, String.format("INSERT INTO t_order_item (order_id, user_id, status) VALUES (%s, %s,'%s')", orderItem.getOrderId(), orderItem.getUserId(), orderItem.getStatus()));

            orderItem.setOrderItemId(orderItemId);

        } catch (final SQLException ignored) {

        }

        return orderItemId;

    }

    

    @Override

    public void delete(final Long id) {

        execute(String.format("delete from t_order_item where order_item_id = %d", id));

    }

    

    @Override

    public List<OrderItem> selectAll() {

        List<OrderItem> result = new LinkedList<>();

        String sql = "SELECT i.* FROM t_order o, t_order_item i WHERE o.order_id = i.order_id";

        try (Connection connection = dataSource.getConnection();

             Statement statement = connection.createStatement()) {

            ResultSet resultSet = statement.executeQuery(sql);

            result.addAll(getOrderItems(resultSet));

        } catch (final SQLException ignored) {

        }

        return result;

    }

    

    private List<OrderItem> getOrderItems(final ResultSet resultSet) {

        List<OrderItem> result = new LinkedList<>();

        try {

            while (resultSet.next()) {

                OrderItem orderItem = new OrderItem();

                orderItem.setOrderItemId(resultSet.getLong(1));

                orderItem.setOrderId(resultSet.getLong(2));

                orderItem.setUserId(resultSet.getInt(3));

                orderItem.setStatus(resultSet.getString(4));

                result.add(orderItem);

            }

        } catch (final SQLException ignored) {

        }

        return result;

    }

    

    private long insertAndGetGeneratedKey(final Statement statement, final String sql) throws SQLException {

        long result = -1;

        statement.executeUpdate(sql, Statement.RETURN_GENERATED_KEYS);

        try (ResultSet resultSet = statement.getGeneratedKeys()) {

            if (resultSet.next()) {

                result = resultSet.getLong(1);

            }

        }

        return result;

    }

    

    private void close(final Connection connection, final Statement statement) {

        if (null != connection && null != statement) {

            try {

                connection.close();

                statement.close();

            } catch (final SQLException ignored) {

            }

        }

    }

    

    private void execute(final String sql) {

        try (Connection connection = dataSource.getConnection();

             Statement statement = connection.createStatement()) {

            statement.execute(sql);

        } catch (final SQLException ignored) {

        }

    }

}