Ambari 2.7.3接口二次开发流程

不念不忘少年蓝@ 2023-06-21 03:55 17阅读 0赞

# Ambari 2.7.3接口二次开发流程 #

## 1. 编写实体类(Entity) ##

路径： src/main/java/org/apache/ambari/server/orm/entities/自定义实体类.java

对应的数据表为 auditlog\_table，添加注解@Table、@Entity、@Id、@Column；自定义属性：id和note；添加getter/setter方法。  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70]  
还需要在 /src/main/resources/META-INF/persistence.xml里面添加entityTest的信息。

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 1]

## 2. 编写DAO层 ##

路径： src/main/java/org/apache/ambari/server/orm/dao/TestUserDao.java

sql语句要遵循JPA规范，如果要指定表的某一个属性，比如id，则id的表现形式为 test.id。  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 2]

## 3. 编写service层 ##

路径： src/main/java/org/apache/ambari/server/api/services/users/TestUserService.java  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 3]

## 4. 声明Resource类型 ##

#### 1) Resource.java ####

路径： org.apache.ambari.server.controller.spi.Resource.java  
![在这里插入图片描述][20191211103721626.png]  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 4]

#### 2) AuditlogResourceDefinition.java ####

路径： org.apache.ambari.server.api.resources.TestUserResourceDefinition.java  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 5]

#### 3) ResourceInstanceFactoryImpl.java ####

路径： org.apache.ambari.server.api.resources.ResourceInstanceFactoryImpl.java  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 6]

#### 4)ResourceProviderFactory.java ####

路径: org.apache.ambari.server.controller.ResourceProviderFactory.java  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 7]

#### 5) AbstractControllerResourceProvider.java ####

路径： org.apache.ambari.server.controller.internal.AbstractControllerResourceProvider.java  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 8]

## 5. 编写resourceProvider ##

路径： org.apache.ambari.server.controller.internal.AuditlogResourceProvider.java

#### 1) 继承AbstractControllerResourceProvider类 ####

public class TestUserResourceProvider extends AbstractControllerResourceProvider
    public static final Logger LOG = LoggerFactory.getLogger(TestUserResourceProvider.class);
    //定义属性值
    public static final String USER_RESOURCE_CATEGORY = "auditlog";
    
    public static final String AUDIT_ID_PROPERTY_ID = "id";
    public static final String AUDIT_TIME_PROPERTY_ID = "time";
    public static final String AUDIT_USER_PROPERTY_ID = "user";
    public static final String AUDIT_STATUS_PROPERTY_ID = "status";
    public static final String AUDIT_OPERATION_PROPERTY_ID = "operation";
    public static final String AUDIT_REMOTEIP_PROPERTY_ID = "remoteIp";
    public static final String AUDIT_NOTE_PROPERTY_ID = "note";
    
    public static final String USER_USERNAME_PROPERTY_ID = USER_RESOURCE_CATEGORY + "/" + AUDIT_ID_PROPERTY_ID;
    public static final String USER_DISPLAY_NAME_PROPERTY_ID = USER_RESOURCE_CATEGORY + "/" + AUDIT_TIME_PROPERTY_ID;
    public static final String USER_LOCAL_USERNAME_PROPERTY_ID = USER_RESOURCE_CATEGORY + "/" + AUDIT_USER_PROPERTY_ID;
    public static final String USER_ACTIVE_PROPERTY_ID = USER_RESOURCE_CATEGORY + "/" + AUDIT_STATUS_PROPERTY_ID;
    public static final String USER_CREATE_TIME_PROPERTY_ID = USER_RESOURCE_CATEGORY + "/" + AUDIT_OPERATION_PROPERTY_ID;
    public static final String USER_CONSECUTIVE_FAILURES_PROPERTY_ID = USER_RESOURCE_CATEGORY + "/" + AUDIT_REMOTEIP_PROPERTY_ID;
    public static final String USER_ADMIN_PROPERTY_ID = USER_RESOURCE_CATEGORY + "/" + AUDIT_NOTE_PROPERTY_ID;

/** * 用户资源的键属性id。 */
    private static Map<Resource.Type, String> keyPropertyIds = ImmutableMap.<Resource.Type, String>builder()
        .put(Resource.Type.TestUser, AUDIT_ID_PROPERTY_ID)
        .build();
    
    private static Set<String> propertyIds = Sets.newHashSet(
        USER_USERNAME_PROPERTY_ID,
        USER_DISPLAY_NAME_PROPERTY_ID,
        USER_LOCAL_USERNAME_PROPERTY_ID,
        USER_ACTIVE_PROPERTY_ID,
        USER_CREATE_TIME_PROPERTY_ID,
        USER_CONSECUTIVE_FAILURES_PROPERTY_ID,
        USER_ADMIN_PROPERTY_ID
    );

/** * 为给定的管理控制器创建新的资源提供程序。 * @param managementController */
    @AssistedInject
    TestUserResourceProvider(@Assisted AmbariManagementController managementController) { 
     super(Resource.Type.TestUser, propertyIds, keyPropertyIds, managementController);
    
     setRequiredCreateAuthorizations(EnumSet.of(RoleAuthorization.AMBARI_MANAGE_USERS));
     setRequiredDeleteAuthorizations(EnumSet.of(RoleAuthorization.AMBARI_MANAGE_USERS));
    }
    // 4) getResource()
    @Override
    public Set<Resource> getResources(Request request, Predicate predicate)
            throws SystemException, UnsupportedPropertyException, NoSuchResourceException, NoSuchParentResourceException { 
    
        final Set<TestUserRequest> requests = new HashSet<TestUserRequest>();
    
        if (predicate == null) { 
            requests.add(getRequest(null));
        } else { 
            for (Map<String, Object> propertyMap : getPropertyMaps(predicate)) { 
                requests.add(getRequest(propertyMap));
            }
        }
    
        Set<TestUserResponse> responses = getResources(new Command<Set<TestUserResponse>>() { 
            @Override
            public Set<TestUserResponse> invoke() throws AmbariException, AuthorizationException { 
                return getManagementController().getTestUser(requests);
            }
        });
    
        if (LOG.isDebugEnabled()) { 
            LOG.debug("Found clusters matching getClusters request"
                    + ", clusterResponseCount=" + responses.size());
        }
    
        Set<String>   requestedIds = getRequestPropertyIds(request, predicate);
        Set<Resource> resources    = new HashSet<Resource>();
    
        for (TestUserResponse auditlogResponse : responses) { 
    
            ResourceImpl resource = new ResourceImpl(Resource.Type.TestUser);
    
            String note =testUserResponse.getNote();
            String[] sourceNoteArray = note.split(", ");
            String time = sourceNoteArray[0].replace("T", " ").replaceAll("\\..*", "");
            String user = sourceNoteArray[1].replace("User(", "").replace(")", "");
            String status = note.split("Status\\(")[1].split("\\)")[0];
    
            boolean isOperation = note.contains("Operation");
            String operation = null;
            if (isOperation) { 
                operation = note.split("Operation\\(")[1].split("\\),")[0];
            } else { 
                operation = "";
            }
    
            boolean isRemoteIp = note.contains("RemoteIp");
            String remoteIp = null;
            if (isRemoteIp) { 
                remoteIp = note.split("RemoteIp\\(")[1].split("\\)")[0];
            } else { 
                remoteIp = "";
            }
    
            // 该方法是判断requestedIds是否包含AUDIT_ID_PROPERTY_ID，如果包含则返回true，将值存入resource中；反之则不存入。
            setResourceProperty(resource, AUDIT_ID_PROPERTY_ID,
                    testUserResponse.getId(), requestedIds);
            setResourceProperty(resource, AUDIT_TIME_PROPERTY_ID,
                    time, requestedIds);
            setResourceProperty(resource, AUDIT_USER_PROPERTY_ID,
                    user, requestedIds);
            setResourceProperty(resource, AUDIT_STATUS_PROPERTY_ID,
                    status, requestedIds);
            setResourceProperty(resource, AUDIT_OPERATION_PROPERTY_ID,
                    operation, requestedIds);
            setResourceProperty(resource, AUDIT_REMOTEIP_PROPERTY_ID,
                    remoteIp, requestedIds);
            setResourceProperty(resource, AUDIT_NOTE_PROPERTY_ID,
                    note, requestedIds);
    
            resources.add(resource);
        }
        return resources;
    }
    
    
    
    //5) 类内其他方法
     /** * ----- ResourceProvider ------------------------------------------------ */
    @Override
    protected Set<String> getPKPropertyIds() { 
        return new HashSet<>(keyPropertyIds.values());
    }
    
    private TestUserRequest getRequest(Map<String, Object> properties) { 
        if (properties == null) { 
            return new TestUserRequest(null);
        }
    
        TestUserRequest request = new TestUserRequest ((String)properties.get(USER_USERNAME_PROPERTY_ID));
    
        request.setId((String) properties.get(AUDIT_ID_PROPERTY_ID));
        request.setNote((String) properties.get(AUDIT_NOTE_PROPERTY_ID));
    
        return request;
    }

## 7. 新建TestUserRequest类 ##

路径： src/main/java/org/apache/ambari/server/controller/TestUserRequest.java

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 9]

## 8. 新建TestUserResponse类 ##

路径： src/main/java/org/apache/ambari/server/controller/TestUserResponse.java  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 10]

## 9.添加getTestUser() ##

mbariManagementController类内添加getTestUser()  
路径：org.apache.ambari.server.controller.AmbariManagementController.java  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 11]  
在AmbariManagementControllerImpl类内实现getTestUser()  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 12]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20191211103409938.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 1]: https://img-blog.csdnimg.cn/20191211103440830.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 2]: https://img-blog.csdnimg.cn/2019121110351851.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 3]: https://img-blog.csdnimg.cn/20191211103632916.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg==,size_16,color_FFFFFF,t_70
[20191211103721626.png]: https://img-blog.csdnimg.cn/20191211103721626.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 4]: https://img-blog.csdnimg.cn/20191211103729324.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 5]: https://img-blog.csdnimg.cn/2019121110390539.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 6]: https://img-blog.csdnimg.cn/20191211103935835.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 7]: https://img-blog.csdnimg.cn/20191211104007593.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 8]: https://img-blog.csdnimg.cn/20191211104035982.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 9]: https://img-blog.csdnimg.cn/20191211104555481.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 10]: https://img-blog.csdnimg.cn/20191211104622666.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 11]: https://img-blog.csdnimg.cn/20191211104709115.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg_size_16_color_FFFFFF_t_70 12]: https://img-blog.csdnimg.cn/20191211104728848.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h1ZWd1Y2hlbg==,size_16,color_FFFFFF,t_70