This is an automated email from the ASF dual-hosted git repository.

benjobs pushed a commit to branch dev-2.1.5
in repository https://gitbox.apache.org/repos/asf/incubator-streampark.git


The following commit(s) were added to refs/heads/dev-2.1.5 by this push:
     new ed19587c0 [Cherry-pick] default team member improvement 3953 (#3956)
ed19587c0 is described below

commit ed19587c061f1dc653f1594f43da233e96057e12
Author: ShawHee <[email protected]>
AuthorDate: Fri Aug 9 22:12:35 2024 +0800

    [Cherry-pick] default team member improvement 3953 (#3956)
    
    * set default team member
    
    * getTeamId from teamService
    
    * code format
    
    ---------
    
    Co-authored-by: hechao <[email protected]>
---
 .../org/apache/streampark/console/base/domain/Constant.java |  2 ++
 .../streampark/console/system/service/RoleService.java      |  2 ++
 .../console/system/service/impl/RoleServiceImpl.java        |  6 ++++++
 .../console/system/service/impl/UserServiceImpl.java        | 13 +++++++++++++
 4 files changed, 23 insertions(+)

diff --git 
a/streampark-console/streampark-console-service/src/main/java/org/apache/streampark/console/base/domain/Constant.java
 
b/streampark-console/streampark-console-service/src/main/java/org/apache/streampark/console/base/domain/Constant.java
index 92c760419..bfbcc369f 100644
--- 
a/streampark-console/streampark-console-service/src/main/java/org/apache/streampark/console/base/domain/Constant.java
+++ 
b/streampark-console/streampark-console-service/src/main/java/org/apache/streampark/console/base/domain/Constant.java
@@ -31,4 +31,6 @@ public class Constant {
   public static final String APP_DETAIL_MENU_ID = "100018";
 
   public static final Long DEFAULT_TEAM_ID = 100000L;
+
+  public static final Long DEFAULT_ROLE_ID = 100001L;
 }
diff --git 
a/streampark-console/streampark-console-service/src/main/java/org/apache/streampark/console/system/service/RoleService.java
 
b/streampark-console/streampark-console-service/src/main/java/org/apache/streampark/console/system/service/RoleService.java
index a509ced3a..4c1bdeebc 100644
--- 
a/streampark-console/streampark-console-service/src/main/java/org/apache/streampark/console/system/service/RoleService.java
+++ 
b/streampark-console/streampark-console-service/src/main/java/org/apache/streampark/console/system/service/RoleService.java
@@ -34,4 +34,6 @@ public interface RoleService extends IService<Role> {
   void deleteRole(Long roleId);
 
   void updateRole(Role role);
+
+  Role getSysDefaultRole();
 }
diff --git 
a/streampark-console/streampark-console-service/src/main/java/org/apache/streampark/console/system/service/impl/RoleServiceImpl.java
 
b/streampark-console/streampark-console-service/src/main/java/org/apache/streampark/console/system/service/impl/RoleServiceImpl.java
index a5a93ec54..c2447b9cf 100644
--- 
a/streampark-console/streampark-console-service/src/main/java/org/apache/streampark/console/system/service/impl/RoleServiceImpl.java
+++ 
b/streampark-console/streampark-console-service/src/main/java/org/apache/streampark/console/system/service/impl/RoleServiceImpl.java
@@ -125,4 +125,10 @@ public class RoleServiceImpl extends 
ServiceImpl<RoleMapper, Role> implements Ro
               this.roleMenuMapper.insert(rm);
             });
   }
+
+  @Override
+  public Role getSysDefaultRole() {
+    return baseMapper.selectOne(
+        new LambdaQueryWrapper<Role>().eq(Role::getRoleId, 
Constant.DEFAULT_ROLE_ID));
+  }
 }
diff --git 
a/streampark-console/streampark-console-service/src/main/java/org/apache/streampark/console/system/service/impl/UserServiceImpl.java
 
b/streampark-console/streampark-console-service/src/main/java/org/apache/streampark/console/system/service/impl/UserServiceImpl.java
index 0527d583a..584f157a1 100644
--- 
a/streampark-console/streampark-console-service/src/main/java/org/apache/streampark/console/system/service/impl/UserServiceImpl.java
+++ 
b/streampark-console/streampark-console-service/src/main/java/org/apache/streampark/console/system/service/impl/UserServiceImpl.java
@@ -23,10 +23,13 @@ import 
org.apache.streampark.console.base.exception.ApiAlertException;
 import org.apache.streampark.console.base.mybatis.pager.MybatisPager;
 import org.apache.streampark.console.base.util.ShaHashUtils;
 import org.apache.streampark.console.system.authentication.JWTToken;
+import org.apache.streampark.console.system.entity.Member;
+import org.apache.streampark.console.system.entity.Role;
 import org.apache.streampark.console.system.entity.User;
 import org.apache.streampark.console.system.mapper.UserMapper;
 import org.apache.streampark.console.system.service.MemberService;
 import org.apache.streampark.console.system.service.MenuService;
+import org.apache.streampark.console.system.service.RoleService;
 import org.apache.streampark.console.system.service.TeamService;
 import org.apache.streampark.console.system.service.UserService;
 
@@ -63,6 +66,8 @@ public class UserServiceImpl extends ServiceImpl<UserMapper, 
User> implements Us
 
   @Autowired private TeamService teamService;
 
+  @Autowired private RoleService roleService;
+
   @Override
   public User findByName(String username) {
     LambdaQueryWrapper<User> queryWrapper =
@@ -104,6 +109,14 @@ public class UserServiceImpl extends 
ServiceImpl<UserMapper, User> implements Us
     user.setLastTeamId(teamService.getSysDefaultTeam().getId());
     user.setPassword(password);
     save(user);
+    // set team member
+    Member member = new Member();
+    member.setUserName(user.getUsername());
+    member.setTeamId(teamService.getSysDefaultTeam().getId());
+    Role role = roleService.getSysDefaultRole();
+    member.setRoleId(role.getRoleId());
+    member.setRoleName(role.getRoleName());
+    memberService.createMember(member);
   }
 
   @Override

Reply via email to