diff --git a/dao/src/main/java/com/xiaojukeji/kafka/manager/dao/impl/AccountDaoImpl.java b/dao/src/main/java/com/xiaojukeji/kafka/manager/dao/impl/AccountDaoImpl.java index 315762b7..9283d40b 100644 --- a/dao/src/main/java/com/xiaojukeji/kafka/manager/dao/impl/AccountDaoImpl.java +++ b/dao/src/main/java/com/xiaojukeji/kafka/manager/dao/impl/AccountDaoImpl.java @@ -18,6 +18,9 @@ public class AccountDaoImpl implements AccountDao { @Autowired private SqlSessionTemplate sqlSession; + @Autowired + private KafkaManagerProperties kafkaManagerProperties; + public void setSqlSession(SqlSessionTemplate sqlSession) { this.sqlSession = sqlSession; } @@ -25,7 +28,7 @@ public class AccountDaoImpl implements AccountDao { @Override public int addNewAccount(AccountDO accountDO) { accountDO.setStatus(DBStatusEnum.NORMAL.getStatus()); - return sqlSession.insert("AccountDao.insert", accountDO); + return updateAccount(accountDO); } @Override @@ -35,6 +38,9 @@ public class AccountDaoImpl implements AccountDao { @Override public int updateAccount(AccountDO accountDO) { + if (kafkaManagerProperties.hasPG()) { + return sqlSession.insert("AccountDao.insertOnPG", accountDO); + } return sqlSession.insert("AccountDao.insert", accountDO); } diff --git a/dao/src/main/resources/mapper/AccountDao.xml b/dao/src/main/resources/mapper/AccountDao.xml index 5d5e79a8..feed2727 100644 --- a/dao/src/main/resources/mapper/AccountDao.xml +++ b/dao/src/main/resources/mapper/AccountDao.xml @@ -22,6 +22,18 @@ ]]> + + + + DELETE FROM account WHERE username = #{username} diff --git a/dao/src/main/resources/mapper/BrokerDao.xml b/dao/src/main/resources/mapper/BrokerDao.xml index 7ef90b01..50af2ef1 100644 --- a/dao/src/main/resources/mapper/BrokerDao.xml +++ b/dao/src/main/resources/mapper/BrokerDao.xml @@ -22,12 +22,12 @@ insert into broker - (cluster_id, broker_id, host, port, timestamp, status) - values + (cluster_id, broker_id, host, port, timestamp, status) + values (#{clusterId}, #{brokerId}, #{host}, #{port}, #{timestamp}, #{status}) on conflict (cluster_id, broker_id) do update set host = excluded.host, - port = excluded.port, - timestamp = excluded.timestamp, - status = excluded.status + port = excluded.port, + timestamp = excluded.timestamp, + status = excluded.status