8x48/KnowStreaming
1
0
Fork 0
mirror of https://github.com/didi/KnowStreaming.git synced 2026-01-12 02:54:59 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

为 AccountDao.insert 也提供 PostgreSQL 的 AccountDao.insertOnPG 版。

Browse Source
This commit is contained in:
Yang Jing
2020-07-05 00:55:55 +08:00
parent ac86f8aded
commit 27ce4d6a0d
3 changed files with 24 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
dao/src/main/java/com/xiaojukeji/kafka/manager/dao/impl/AccountDaoImpl.java
View File

@@ -18,6 +18,9 @@ public class AccountDaoImpl implements AccountDao {
@Autowired @Autowired
private SqlSessionTemplate sqlSession; private SqlSessionTemplate sqlSession;
@Autowired
private KafkaManagerProperties kafkaManagerProperties;
public void setSqlSession(SqlSessionTemplate sqlSession) { public void setSqlSession(SqlSessionTemplate sqlSession) {
this.sqlSession = sqlSession; this.sqlSession = sqlSession;
} }
@@ -25,7 +28,7 @@ public class AccountDaoImpl implements AccountDao {
@Override @Override
public int addNewAccount(AccountDO accountDO) { public int addNewAccount(AccountDO accountDO) {
accountDO.setStatus(DBStatusEnum.NORMAL.getStatus()); accountDO.setStatus(DBStatusEnum.NORMAL.getStatus());
return sqlSession.insert("AccountDao.insert", accountDO); return updateAccount(accountDO);
} }
@Override @Override
@@ -35,6 +38,9 @@ public class AccountDaoImpl implements AccountDao {
@Override @Override
public int updateAccount(AccountDO accountDO) { public int updateAccount(AccountDO accountDO) {
if (kafkaManagerProperties.hasPG()) {
return sqlSession.insert("AccountDao.insertOnPG", accountDO);
}
return sqlSession.insert("AccountDao.insert", accountDO); return sqlSession.insert("AccountDao.insert", accountDO);
} }

12
dao/src/main/resources/mapper/AccountDao.xml
View File

@@ -22,6 +22,18 @@
]]> ]]>
</insert> </insert>
<insert id="insertOnPG" parameterType="com.xiaojukeji.kafka.manager.common.entity.po.AccountDO">
<![CDATA[
insert into account
(username, password, role, status)
values
(#{username}, #{password}, #{role}, #{status})
on conflict (username) do update set password = excluded.password,
role = excluded.role,
status = excluded.status
]]>
</insert>
<delete id="deleteByName" parameterType="java.lang.String"> <delete id="deleteByName" parameterType="java.lang.String">
DELETE FROM account WHERE username = #{username} DELETE FROM account WHERE username = #{username}
</delete> </delete>

10
dao/src/main/resources/mapper/BrokerDao.xml
View File

@@ -22,12 +22,12 @@
<insert id="replaceOnPG" parameterType="BrokerDO"> <insert id="replaceOnPG" parameterType="BrokerDO">
insert into broker insert into broker
(cluster_id, broker_id, host, port, timestamp, status) (cluster_id, broker_id, host, port, timestamp, status)
values values (#{clusterId}, #{brokerId}, #{host}, #{port}, #{timestamp}, #{status})
on conflict (cluster_id, broker_id) do update set host = excluded.host, on conflict (cluster_id, broker_id) do update set host = excluded.host,
port = excluded.port, port = excluded.port,
timestamp = excluded.timestamp, timestamp = excluded.timestamp,
status = excluded.status status = excluded.status
</insert> </insert>
<delete id="deleteById" parameterType="java.util.Map"> <delete id="deleteById" parameterType="java.util.Map">