public static void readHiveResult(String uuid, OutputStreamWriter outStream)
throws SQLException, IOException {
Map stormconf = Utils.readStormConfig();
String hdpConf = (String) stormconf.get("hadoop.conf.dir");
String connstr = (String) stormconf.get("higo.download.offline.conn");
String uname = (String) stormconf.get("higo.download.offline.username");
String passwd = (String) stormconf.get("higo.download.offline.passwd");
MySqlConn conn = new MySqlConn(connstr, uname, passwd);
Configuration conf = new Configuration();
HadoopBaseUtils.grabConfiguration(hdpConf, conf);
MysqlInfo info = new MysqlInfo(conn);
HashMap<String, String> result = info.get(uuid);
String exehql = "";
if (result != null && result.containsKey("storedir")) {
exehql = result.get("_exehql");
}
if (result != null && result.containsKey("storedir") && result.containsKey("cols")) {
outStream.write(result.get("cols").replaceAll("\001", ",").replaceAll("\t", ",") + "\n");
AdhocOfflineService.readHiveResult(result.get("storedir"), outStream, conf);
}
// outStream.write(exehql.replaceAll("\001", ",").replaceAll("\t",
// "\t")+"\n");
}
public static void readJoinResult(String uuid, OutputStreamWriter outStream)
throws SQLException, IOException, JSONException {
final HashMap<String, String> tableInfo = getTableInfo(uuid);
JSONObject jsonObj = new JSONObject();
if (tableInfo == null || tableInfo.isEmpty()) {
jsonObj.put("code", "0");
jsonObj.put("message", "该表不存在");
outStream.append(jsonObj.toString());
return;
}
if (tableInfo.get("status").equals("INDEXING")) {
jsonObj.put("code", "0");
jsonObj.put("message", "正在创建索引中请稍后");
outStream.append(jsonObj.toString());
return;
}
Map stormconf = Utils.readStormConfig();
Configuration conf = getConf(stormconf);
com.alimama.mdrill.ui.service.AdhocOfflineService.readHiveResult(
tableInfo.get("txtStorePath"), outStream, conf);
}
public static String offline(
String projectName,
String jsoncallback,
String queryStr,
String fl,
String groupby,
String mailto,
String username,
String jobname,
String params,
String leftjoin,
String fq2,
String limit2,
String orderby2,
String desc2)
throws Exception {
int limit = 0;
if (limit2 != null) {
limit = Integer.parseInt(limit2);
}
long t1 = System.currentTimeMillis();
boolean isPartionByPt = false;
if (projectName.equals("rpt_hitfake_auctionall_d")) {
projectName = "rpt_p4padhoc_auction";
}
String hpart = "dt";
if (projectName.equals("rpt_b2bad_hoc_memb_sum_d")) {
isPartionByPt = true;
hpart = "pt";
}
if (projectName.equals("r_rpt_tanx_adzone_total")) {
hpart = "ds";
}
queryStr = WebServiceParams.query(queryStr);
// String sqlSort = WebServiceParams.sortHive(sort, order);
TablePartion part = GetPartions.partion(projectName);
String[] cores = GetShards.get(part.name, false, 10000);
String[] ms = GetShards.get(part.name, true, 10000);
String[] partionsAll = GetPartions.get(queryStr, part.parttype);
GetPartions.Shards shard = GetPartions.getshard(part, partionsAll, cores, ms, 10000, 0);
HashMap<String, String> filetypeMap = MdrillService.readFieldsFromSchemaXml(part.name);
ArrayList<String> fqList =
WebServiceParams.fqListHive(
hpart, queryStr, shard, isPartionByPt, filetypeMap, null, null, null);
StringBuffer sqlWhere = new StringBuffer();
String join = " where ";
for (String fq : fqList) {
sqlWhere.append(join);
sqlWhere.append(fq);
join = " and ";
}
ArrayList<String> groupFields = WebServiceParams.groupFields(groupby);
StringBuffer sqlGroup = new StringBuffer();
String daycols = "";
join = " group by ";
for (String field : groupFields) {
sqlGroup.append(join);
sqlGroup.append(field);
if (field.equals("thedate")) {
daycols = "日期,";
}
join = ",";
}
HashMap<String, String> colMap = new HashMap<String, String>();
ArrayList<String> showFields = WebServiceParams.showHiveFields(fl);
StringBuffer cols = new StringBuffer();
join = "";
int nameindex = 0;
for (String field : groupFields) {
cols.append(join);
cols.append(field);
String alias = "tmp_" + nameindex++;
cols.append(" as " + alias);
colMap.put(field, alias);
join = ",";
}
for (String field : showFields) {
if (field.equals("thedate")) {
daycols = "日期,";
}
if (!groupFields.contains(field)) {
cols.append(join);
cols.append(field);
String alias = "tmp_" + nameindex++;
cols.append(" as " + alias);
colMap.put(field, alias);
join = ",";
}
}
HigoAdhocJoinParams[] joins = AdhocWebServiceParams.parseJoinsHive(leftjoin, shard);
for (int i = 0; i < joins.length; i++) {
HigoAdhocJoinParams jp = joins[i];
if (!groupFields.contains(jp.leftkey) && !showFields.contains(jp.leftkey)) {
cols.append(join);
cols.append(jp.leftkey);
String alias = "tmp_" + nameindex++;
cols.append(" as " + alias);
colMap.put(jp.leftkey, alias);
join = ",";
}
}
String hql =
"select "
+ cols.toString()
+ " from "
+ projectName
+ " "
+ sqlWhere.toString()
+ " "
+ sqlGroup.toString()
+ " ";
HashMap<String, String> colMap2 = new HashMap<String, String>();
if (joins.length > 0) {
StringBuffer buffer = new StringBuffer();
buffer.append("select ");
join = "";
for (String field : groupFields) {
buffer.append(join);
buffer.append("jl1.");
buffer.append(colMap.get(field));
String alias = "tmp_" + nameindex++;
buffer.append(" as " + alias);
colMap2.put(field, alias);
join = ",";
}
for (String field : showFields) {
if (!groupFields.contains(field)) {
buffer.append(join);
buffer.append("jl1.");
buffer.append(colMap.get(field));
String alias = "tmp_" + nameindex++;
buffer.append(" as " + alias);
colMap2.put(field, alias);
join = ",";
}
}
for (int i = 0; i < joins.length; i++) {
HigoAdhocJoinParams jp = joins[i];
for (String field : jp.fl) {
buffer.append(join);
buffer.append("jr" + i + ".");
buffer.append(field);
String alias = "tmp_" + nameindex++;
buffer.append(" as " + alias);
join = ",";
}
}
buffer.append(" from ");
buffer.append(" (" + hql + ") jl1 ");
for (int i = 0; i < joins.length; i++) {
HigoAdhocJoinParams jp = joins[i];
buffer.append(
" join ("
+ jp.frQuer
+ ") jr"
+ i
+ " on jl1."
+ colMap.get(jp.leftkey)
+ "==jr"
+ i
+ "."
+ jp.rightkey
+ " ");
}
hql = buffer.toString();
}
// String fq2,int limit
ArrayList<String> fq2list =
WebServiceParams.fqListHive(
hpart, fq2, shard, isPartionByPt, filetypeMap, colMap, colMap2, "fq2");
if (fq2list.size() > 0) {
StringBuffer buffer = new StringBuffer();
buffer.append("select * from ");
buffer.append("(" + hql + ") fq2");
String join2 = " where ";
for (String fq : fq2list) {
buffer.append(join2);
buffer.append(fq);
join2 = " and ";
}
hql = buffer.toString();
}
if (orderby2 != null) {
hql = hql + " order by fq2." + colMap2.get(orderby2) + " " + desc2;
}
if (limit > 1000000) {
limit = 1000000;
}
if (limit > 0) {
hql = hql + " limit " + limit;
}
String md5 = MD5.getMD5(hql);
SimpleDateFormat fmt = new SimpleDateFormat("yyyyMMdd");
String day = fmt.format(new Date());
Map stormconf = Utils.readStormConfig();
String hdpConf = (String) stormconf.get("hadoop.conf.dir");
String connstr = (String) stormconf.get("higo.download.offline.conn");
String uname = (String) stormconf.get("higo.download.offline.username");
String passwd = (String) stormconf.get("higo.download.offline.passwd");
String store =
(String) stormconf.get("higo.download.offline.store")
+ "/"
+ day
+ "/"
+ java.util.UUID.randomUUID().toString();
MySqlConn conn = new MySqlConn(connstr, uname, passwd);
MysqlInfo info = new MysqlInfo(conn);
if (username == null || username.length() <= 0) {
username = "******";
}
StringBuffer sqlbuff = new StringBuffer();
// int jobsize=info.getUserJobname(username, jobname, sqlbuff).size();
sqlbuff.append(";");
int size = info.getUser(username, true, sqlbuff).size();
JSONObject jsonObj = new JSONObject();
jsonObj.put("sqlbuff", sqlbuff.toString());
jsonObj.put("size", size);
// if(jobsize>0)
// {
// jsonObj.put("code", "0");
// jsonObj.put("message", "之前已经有叫"+jobname+"的任务,请换个名字");
// }else
//
if (size < 5) {
MysqlCallback callback = new MysqlCallback(conn);
String[] pcols =
params == null
? new String[0]
: new String(daycols + params.replaceAll("维度指标：", "").replaceAll("。.*$", ""))
.split(",");
StringBuffer liststat = new StringBuffer();
StringBuffer listgroup = new StringBuffer();
for (String s : pcols) {
if (AdhocOfflineService.isStatFields(s)) {
liststat.append(s);
liststat.append(",");
} else {
listgroup.append(s);
listgroup.append(",");
}
}
callback.setCols(
(params == null || params.isEmpty())
? cols.toString()
: listgroup.toString() + liststat.toString());
OfflineDownload download = new OfflineDownload();
download.setOffline(callback);
if (mailto == null || mailto.length() <= 0) {
mailto = "*****@*****.**";
}
StringBuffer setSql = new StringBuffer();
StringBuffer cleanSql = new StringBuffer();
for (int i = 0; i < joins.length; i++) {
HigoAdhocJoinParams jp = joins[i];
setSql.append(jp.createSql);
setSql.append(";");
setSql.append(jp.addData);
setSql.append(";");
cleanSql.append(";");
cleanSql.append(jp.DropSql);
}
download.setMailto(mailto);
download.setHql(
setSql.toString()
+ " INSERT OVERWRITE DIRECTORY '"
+ store
+ "' "
+ hql
+ " "
+ cleanSql.toString());
download.setUseName(username);
if (jobname == null || jobname.length() <= 0) {
jobname = day + "_" + md5;
}
download.setJobName(jobname);
download.setDisplayParams((params == null || params.isEmpty()) ? hql : params);
download.setStoreDir(store);
download.setConfdir(hdpConf);
download.setSqlMd5(md5);
download.run();
long t2 = System.currentTimeMillis();
jsonObj.put("code", "1");
jsonObj.put(
"message",
"数据下载中...完成后将会通过<b style=\"color:red\">旺旺</b>和<b style=\"color:red\">邮件</b>通知");
jsonObj.put("uuid", callback.getUuid());
jsonObj.put("debug", callback.toString());
jsonObj.put("timedebug", String.valueOf(t2 - t1));
} else {
jsonObj.put("code", "0");
jsonObj.put("message", "还有" + size + "个任务没有完成数据下载，请稍后提交");
}
if (jsoncallback != null && jsoncallback.length() > 0) {
return jsoncallback + "(" + jsonObj.toString() + ")";
} else {
return jsonObj.toString();
}
}
