#!/usr/bin/env python
#coding=utf-8
#scribe日志接收存在小集群到大集群之间, distcp 同步失败的情况,需要手动进行补入。
#1、如果查询补入的日志量少,则可以之间用脚本处理。如果量大,则使用 hadoop 提交job。
# hadoop job 提交方式:
#   hadoop jar /usr/local/hadoop-2.4.0/share/hadoop/tools/lib/hadoop-distcp-2.4.0.jar -m 100 hdfs://scribehadoop/scribelog/common_act/2016/08/02/13/  /file/realtime/distcpv2/scribelog/common_act/2016/08/02/13  --update 
 
#  --update 参数表示如果目标地址目录存在,则更新该目录中的内容。

#手动同步脚本使用方法: python manual_check_sync.py  dst_path  
#脚本完成大集群和小集群之间的目录大小比较,目录文件比较。 输出差异文件列表。最后完成同步入库。



import sys,os,commands,re
import logging,logging.handlers 

Module=sys.argv[1].split("scribelog")[1]


logger1 = logging.getLogger('mylogger1')
logger1.setLevel(logging.DEBUG)
# 创建一个handler,用于写入日志文件
filehandle = logging.FileHandler('log/test.log',mode='a')
# 再创建一个handler,用于输出到控制台
consehandle = logging.StreamHandler()

# 定义handler的输出格式formatter
formatter = logging.Formatter('[%(asctime)s]:%(message)s',datefmt='%F %T')
#formatter = logging.Formatter('[%(asctime)s-line:%(lineno)d-%(levelname)s]:%(message)s',datefmt='%F %T')
filehandle.setFormatter(formatter)
consehandle.setFormatter(formatter)

logger1.addHandler(filehandle)
logger1.addHandler(consehandle)




little_cluster="hdfs://yz632.hadoop.com.cn/scribelog"
large_cluster="hdfs://ns1"

#logger1.info("源目录(小集群):%s%s" % ( little_cluster,Module))

logger1.info("源目录(小集群):%s%s" %(little_cluster,Module))
logger1.info("目标目录(大集群):%s%s" %(large_cluster,sys.argv[1]))



#统计目录大小等情况
du_little=commands.getoutput("hadoop fs -count " + little_cluster + Module)
du_large=commands.getoutput("hadoop fs -count " + large_cluster + sys.argv[1])

#获取的值是str类型,所以需要转为list来做比较。

logger1.info("                DIR_COUNT FILECOUNT CONTENTSIZE ")
logger1.info("小集群目录信息:%s" %(du_little))
logger1.info("大集群目录信息:%s" %(du_large))


#Python的str类有split方法,只能根据指定的某个字符分隔字符串,re模块中提供的split方法可以定义多个分隔符。这里可以用单个空格作为分隔即可。
du_little=re.split('         |           | ' ,du_little)
du_large=re.split('         |           | ' ,du_large)


#du_little=du_little.split(" ")
#du_large=du_large.split(" ")


#print du_large,du_little
#print du_large[3],du_little[3]
#print du_large[6],du_little[6]
#print du_large[7],du_little[7]
#
if du_little[3] == du_large[3]  and du_little[6] == du_large[6] and du_little[7] == du_large[7]:
 logger1.info("大小集群文件数量、大小一致,不需要同步")
 exit()



#如果大小不一致,取出目录下所有文件
little_list=commands.getoutput("hadoop fs -lsr " + little_cluster + Module + "|grep -v \"^d\"" )
large_list =commands.getoutput("hadoop fs -lsr " + large_cluster + sys.argv[1]+"|grep -v \"^d\"")

#logger1.info( "小集群情况:%s " %(little_list))
#logger1.info( "大集群情况:%s ")%(large_list ))



list1=[]
list2=[]
lost_list=[]
for i in  little_list.split("\n"):
 list1.append(i.split("scribelog")[1])

for i in  large_list.split("\n"):
 list2.append(i.split("scribelog")[1])


#logger1.info("小集群目录文件:",list1
#logger1.info("大集群目录文件" ,list2


logger1.info("对比大小集群文件---》未同步文件列表:")
for i in list1:
 if i not in list2:
 logger1.info(i) 
 lost_list.append(i)

logger1.info(lost_list)


#拉取小集群文件到本地tmp目录中
for i in lost_list: 
 s=commands.getstatusoutput("hadoop fs -get " + little_cluster + i+" /tmp/")
 logger1.info("拉取小集群文件到本地tmp目录中%s" %s )


#入库到大集群
for i in lost_list:
 logger1.info(i)
 j=commands.getstatusoutput("sudo su - datacopy -s /bin/bash -c '/usr/local/hadoop-2.4.0/bin/hadoop fs -put  /tmp/" +i.split("/")[-1] + "  /file/realtime/distcpv2/scribelog"+i+" '" )
 logger1.info(j)
 if j[0] == 0 :
 logger1.info(" %s + 同步完成" %(i))


#clear tmp file
for i  in lost_list:
 commands.getstatusoutput("rm -f /tmp/"+i.split("/")[-1] )

#当同步目录时,可能会出现put的时候提示没有目录存在。参考赵兵的 put  -  src dest  方法,是否可以避免此类问题有待验证。



du_little=commands.getoutput("hadoop fs -count " + little_cluster + Module)
du_large=commands.getoutput("hadoop fs -count " + large_cluster + sys.argv[1])


logger1.info("                  DIR_COUNT       FILECOUNT       CONTENTSIZE ")
logger1.info("小集群目录信息:%s" %(du_little))
logger1.info("大集群目录信息:%s" %(du_large))


#Python的str类有split方法,但是这个split方法只能根据指定的某个字符分隔字符串,re模块中提供的split方法可以用来做这件事情
du_little=re.split('         |           | ' ,du_little)
du_large=re.split('         |           | ' ,du_large)


if du_little[3] == du_large[3]  and du_little[6] == du_large[6] and du_little[7] == du_large[7]:
        logger1.info("小集群文件数量、大小一致,不需要同步")
        exit()


https://blog.51cto.com/devops9527/1855294