hive java streaming_利用SparkSQL(java版)将离线数据或实时流数据写入hive的用法及坑点...

1. 通常利用SparkSQL将离线或实时流数据的SparkRDD数据写入Hive,一般有两种方法。第一种是利用org.apache.spark.sql.types.StructType和org.apache.spark.sql.types.DataTypes来映射拆分RDD的值；第二种方法是利用rdd和Java bean来反射的机制。下面对两种方法做代码举例

2. 利用org.apache.spark.sql.types.StructType和org.apache.spark.sql.types.DataTypes来映射拆分RDD的值

JavaRDD resultRdd = rdd.map(new Function() {

@Override

public Row call(String[] line) throws Exception {

if (line != null && line.length > 0) {

return helper.createRow(line);

}

return null;

}

});

StructType structType = helper.createSchame();

Dataset dataFrame = session.createDataFrame(resultRdd, structType);

DataFrameWriter writer = dataFrame.coalesce(1).write().format(TableHelperInter.TABLE_FORMAT_TYPE).mode(SaveMode.Append);

String tableName = hiveDataBaseName + "." + helper.getTableName();

writer.insertInto(tableName);

这种方法的有点是写入简单，不必去考虑字段映射有误，但缺点是需要去写一个TableHelperInter，而且这种方式对字段的类型要求严格，在做字段类型和字段校验时比对时一旦字段过多会及其复杂，所以不推崇这种写法

3. 利用rdd和Java bean来反射

来一个完整的程序

public class SparkSQLTest {

public static void main(String[] args) {

SparkConf conf = new SparkConf().setMaster("yarn").setAppName("SparkSQL_test");

JavaSparkContext sc = new JavaSparkContext(conf);

String line = "1102,jason,20,male,15927384023,developer,col7,col8,col9,col10,col11,col12,col13,col14,col15,col16,col17,col18,col19,col20";

String line2 = "1103,jason1,21,male,15927352023,developer1,col7,col8,col9,col10,col11,col12,col13,col14,col15,col16,col17,col18,col19,col20";

List list = new ArrayList();

list.add(line);

list.add(line2);

JavaRDD rdd = sc.parallelize(list);

JavaRDD rddResult = rdd.map(new Function() {

@Override

public Person call(String s) throws Exception {

String[] message = s.split(",");

Person person = new Person();

person.setNo(message[0]);

person.setName(message[1]);

person.setAge(message[2]);

person.setGender(message[3]);

person.setPhone(message[4]);

person.setJob(message[5]);

person.setCol7(message[6]);

person.setCol8(message[7]);

person.setCol9(message[8]);

person.setCol10(message[9]);

person.setCol11(message[10]);

person.setCol12(message[11]);

person.setCol13(message[12]);

person.setCol14(message[13]);

person.setCol15(message[14]);

person.setCol16(message[15]);

person.setCol17(message[16]);

person.setCol18(message[17]);

person.setCol19(message[18]);

person.setCol20(message[19]);

person.setCreate_time_p(DateTimeFormatter.ofPattern("yyyyMMdd").format(LocalDate.now()));

return person;

}

});

//这行代码必须在实例SparkSession不然会出错

SparkSession.clearDefaultSession();

SparkSession session = SparkSession.builder()

.config("hive.metastore.uris", "localhost:9083")

.config("spark.sql.warehouse.dir", "/apps/hive/warehouse")

.config("hive.exec.dynamic.partition", true)

.config("spark.sql.sources.partitionColumnTypeInference.enabled", false)

.config("hive.exec.dynamic.partition.mode", "nonstrict")

.enableHiveSupport()

.getOrCreate();

Dataset dataset = session.createDataFrame(rddResult,Person.class);

dataset.registerTempTable("person_temp_table");

session.sql("insert into qwrenzixing.person_table20 partition (create_time_p="+DateTimeFormatter.ofPattern("yyyyMMdd")

.format(LocalDate.now())+") select no,name,age,gender,phone,job,col7,col8,col9,col10,col11,col12,col13,col14,col15,col16,col17,col18,col19,col20 from person_temp_table");

}

}

这种方法比较简洁，为了避免去做繁琐的字段比对和校验。可以将字段类型以string写入hive。同时通过SparkSession操作SQL的方法是spark2.0后的。这里是将dataset写成一张临时表，再将临时表的值查询出来insert into到hive表中。但将DataSet通过SparkSQL写成一张临时表的操作，Spark原生提供了四个关于这种操作API

dataset.registerTempTable("temp_table");

dataset.createGlobalTempView("temp_table");

dataset.createOrReplaceTempView("temp_table");

dataset.createTempView("temp_table");

4. 关于这四个将DataSet写成一张临时表的作用和坑点

1>. dataset.registerTempTable("temp_table")这个方法建议在离线，批处理中使用，在实时流式计算中会导致后续写入hive值与字段不匹配乱序的问题

2>. dataset.createGlobalTempView("temp_table")这个方法是创建一个全局临时表，意思就是别的spark-submit也可以用，这种场景很少,而且无法用在实时流式计算中,因为创建一次表后不能再创建会包表已经存在的错误

3>. dataset.createOrReplaceTempView("temp_table");这个其实比较好理解，如果存在就覆盖

4>. dataset.createTempView("temp_table"); 这个方法当spark程序没有结束时不能重复创建

这里的创建临时表在spark程序结束后临时表不存在，所以spark streaming程序要特别注意用法

5. 关于Spark SQL的一个坑点

在mysql中insert into有两种方式

INSERT INTO table_name VALUES (value1, value2,....)

INSERT INTO table_name (column1, column2,...) VALUES (value1, value2,....)

要注意第二种写法在SparkSQL会报错，SparkSQL不支持这种写法，只支持第一种写法。这个是为什么其实也很好理解，每个人想法不一样。第一次使用要避免这个坑点

最后附上我在利用SparkSQL将kafka数据写入hive的重要环节的代码：

String tableName = hiveDataBaseName + ".test_data";

Dataset dataFrame = session.createDataFrame(resultRdd, SJGJEntity.class);

// createOrReplaceTempView API方式将数据写入hive 不存在值与字段名错乱的问题

dataFrame.createOrReplaceTempView("temp_table");

session.sql("insert into " + tableName + " partition(create_time_p=" + DateTimeFormatter.ofPattern("yyyyMMdd").format(LocalDate.now())

+ ") select base_name,base_num,serviceCode,phoneno,called_phoneno,call_time,call_longth,lac,ci,xpoint,ypoint,imei,imsi,insert_time,call_address," + "source_table,mark_type,companyId,type,createKafkaTime from temp_table");