Ambari 插件开发笔记

import sys
from resource_management import *
class Master(Script):
  def install(self, env):
    print 'Install the ES Master';
  def stop(self, env):
    print 'Stop the ES Master';
  def start(self, env):
    print 'Start the ES Master';
     
  def status(self, env):
    print 'Status of the ES Master';
  def configure(self, env):
    print 'Configure the ES Master';
if __name__ == "__main__":
  Master().execute()

# coding=utf8
import random
import sys
from resource_management import *
from resource_management.libraries.script.script import Script
from resource_management.libraries.functions import get_unique_id_and_date
from resource_management.libraries.functions import conf_select
from resource_management.libraries.functions import stack_select
from resource_management.libraries.functions import StackFeature
from resource_management.libraries.functions.version import compare_versions, format_stack_version
from resource_management.libraries.functions.stack_features import check_stack_feature
from resource_management.libraries.functions.security_commons import build_expectations, \
  cached_kinit_executor, get_params_from_filesystem, validate_security_config_properties, \
  FILE_TYPE_JAAS_CONF
from resource_management import *
from resource_management.core.logger import Logger
from resource_management.core.resources.system import Execute
from resource_management.libraries.functions.check_process_status import check_process_status
from resource_management.libraries.functions.format import format
from resource_management.libraries.functions.validate import call_and_match_output
from pickle import STOP
from time import sleep
import ambari_simplejson as json

class Master(Script):
    
    def install(self, env):
        import params
        self.install_packages(env)
        Execute("chown -R {0}:{1} {2}".format(params.app_user,params.app_group,params.install_dir))
        

    def configure(self, env):
    	  # params 见下一个文件
        import params
        env.set_params(params)

        Directory(params.app_conf_dir,
            mode=0755,
            owner=params.app_user,
            group=params.app_group,
            create_parents = True
        )
        
        # 这里根据Templates目录下的.j2文件生成最终程序所需的配置文件
        File(params.app_conf_dir + '/application.properties',
            content=Template('application.properties.j2'),
            owner=params.app_user,
            group=params.app_group,
            mode=0755
        )

        #init system dir
        Directory(params.install_dir,
            mode=0755,
            owner=params.app_user,
            group=params.app_group,
            create_parents = True
        )
        # 省略其他权限、文件夹等配置
   
    def start(self ,env):    
        import params
        env.set_params(params)
        self.configure(env)
        
        no_op_test = format("ls {app_pid_path} >/dev/null 2>&1 && ps `cat {app_pid_path}` | grep `cat {app_pid_path}` >/dev/null 2>&1")
        
        #startup.sh 是打在rpm包中的
        Execute("sh {0}/startup.sh".format(params.app_bin_dir),
              user=params.app_user,
              not_if=no_op_test
        )
        
    def stop(self ,env):    
        import params
        env.set_params(params)
        self.configure(env)
        
        no_op_test = ""
        
        #shutdown.sh 是打在rpm包中的
        Execute("sh {0}/shutdown.sh".format(params.app_bin_dir),
              user=params.app_user
        )
        
    def status(self, env):
        import status_params
        env.set_params(status_params)
        #pid文件 是启停脚本维护的
        check_process_status(params.app_pid_path)     
        
if __name__ == "__main__":
    Master().execute()

#!/usr/bin/env python
from resource_management.core.logger import Logger
from resource_management.libraries.functions import conf_select
from resource_management.libraries.functions import stack_select
from resource_management import *
from resource_management.libraries.functions.get_not_managed_resources import get_not_managed_resources
from resource_management.libraries.functions.expect import expect
from ambari_commons.ambari_metrics_helper import select_metric_collector_hosts_from_hostnames
from ambari_commons import OSCheck
from resource_management.libraries.resources.hdfs_resource import HdfsResource
import os


config = Script.get_config()
exec_tmp_dir = Script.get_tmp_dir()

install_dir = '/data/micro-services/8082-data-api-jar'

app_conf_dir = install_dir + '/conf'
app_bin_dir = install_dir + '/bin'

app_user = config['configurations']['ark-data-api-env']['ark-dataapi-user']
app_group = config['configurations']['ark-data-api-env']['ark-dataapi-group']

app_pid = 'lryhis.pid'
app_pid_path = app_bin_dir + "/" + app_pid

# ---------- server properties ------------
app_server_port = config['configurations']['ark-data-api-server']['dataapi.app.server.port']
app_log_file = install_dir + '/logs/application.log'

presto_host = config['configurations']['ark-data-api-server']['dataapi.presto.host']
presto_query_timeout = config['configurations']['ark-data-api-server']['dataapi.presto.query.timeout']

<!-- 省略部分 -->

db_host=config['configurations']['ark-data-api-server']['dataapi.db.host']
db_admin_user=config['configurations']['ark-data-api-server']['dataapi.db.admin.user']
db_admin_password=config['configurations']['ark-data-api-server']['dataapi.db.admin.password']
db_pool_size=config['configurations']['ark-data-api-server']['dataapi.db.pool.size']

# 这里根据Templates目录下的.j2文件生成最终程序所需的配置文件
        File(params.app_conf_dir + '/application.properties',
            content=Template('application.properties.j2'),
            owner=params.app_user,
            group=params.app_group,
            mode=0755
        )