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