码迷,mamicode.com
首页 > 其他好文 > 详细

storm源码剖析(1):storm脚本

时间:2015-04-15 21:03:25      阅读:194      评论:0      收藏:0      [点我收藏+]

标签:

今天看了一下storm的命令行脚本${STORM_HOME}/bin/storm,现在将剖析过程整理一下,作为记录。注:使用的storm版本为0.8.0。

${STORM_HOME}/bin/storm文件是用python写的,该文件写的还是相当精简和清晰的。

首先,命令的运行从main()方法开始,main()方法主要是解析输入的命令和命令携带的参数以及读取默认配置和配置文件配置。

if __name__ == "__main__":
  main()

def main():

  if len(sys.argv) <= 1:

     print_usage()

    sys.exit(-1)
    global CONFIG_OPTS
    config_list, args = parse_config_opts(sys.argv[1:]) //读取配置项和参数(因为命令行以storm 
                 //
开头,故这里出去argv[0]),"-c"指定配置项,其他为参数
  parse_config(config_list)
  COMMAND
= args[0] //提取执行命令(第一项为命令)
  ARGS
= args[1:] //提取命令携带的参数
  (COMMANDS.get(COMMAND,
"help"))(*ARGS) //查询COMMAND字典,获取要执行的命令对应的方法,调用该方法

以命令storm jar xxx.jar MAINCLASS arg1 arg2 为例,则上面的命令执行jar(xxx.jar, MAINCLASS, arg1, arg2)方法

下面重点分析一下命令执行方法,还是以jar()方法为例吧

def jar(jarfile, klass, *args):
    """Syntax: [storm jar topology-jar-path class ...]

    Runs the main method of class with the specified arguments. 
    The storm jars and configs in ~/.storm are put on the classpath. 
    The process is configured so that StormSubmitter 
    (http://nathanmarz.github.com/storm/doc/backtype/storm/StormSubmitter.html)
    will upload the jar at topology-jar-path when the topology is submitted.
    """
    exec_storm_class(
        klass,
        jvmtype="-client",
        extrajars=[jarfile, CONF_DIR, STORM_DIR + "/bin"],
        args=args,
        jvmopts=["-Dstorm.jar=" + jarfile])
        
def exec_storm_class(klass, jvmtype="-server", jvmopts=[], extrajars=[], args=[], fork=False):
    all_args = [
        "java", jvmtype, get_config_opts(),
        "-Dstorm.home=" + STORM_DIR,
        "-Djava.library.path=" + confvalue("java.library.path", extrajars),
        "-cp", get_classpath(extrajars),
    ] + jvmopts + [klass] + list(args)
    print "Running: " + " ".join(all_args)
    if fork:
        os.spawnvp(os.P_WAIT, "java", all_args)
    else:
        os.execvp("java", all_args) # replaces the current process and never returns

 

storm源码剖析(1):storm脚本

标签:

原文地址:http://www.cnblogs.com/jerryshao2015/p/4429790.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!