在创建数据集时使用 @JsonProperty 重命名 spark 中的列

Question

有没有办法在创建数据集时使用 Jackson 注释重命名数据集中的列名？

我的编码器类如下：

import com.fasterxml.jackson.annotation.JsonProperty;
import lombok.*;
import scala.Serializable;
import javax.persistence.Table;

      @Builder 
      @Data 
      @AllArgsConstructor
      @EqualsAndHashCode 
      @Table(name = "sample_table")
      public class SampleRecord implements Serializable {
         @JsonProperty("sample_id")
         private Long sampleId;
         @JsonProperty("sample_name")
         private String name;
         @JsonProperty("sample_desc")
         private String description; 
      }

我的目标是根据@JsonProperty 重命名列，以便我可以重用相同的类和json 功能。

请查找模块的相关版本： - Spark：2.4.0（使用 Scala 2.11） -杰克逊模块scala_2.11：2.9.6

如果您需要更多信息，请告诉我。帮助表示赞赏。

Answer 1

public class SampleRecord implements Serializable {
         private Long sampleId;

         private String name;

         private String description; 


@JsonProperty("sample_id")
public void setSampleId(Long sampleId) {
        this.sampleId = sampleId;
    }

@JsonProperty("sample_name")
public void setName(String name) {
        this.name = name;
    }


@JsonProperty("sample_desc")
public void setDescription(String description) {
        this.description = description;
    }
}

Answer 2

有趣的想法。我会这样做的方式：

1. 在数据框中提取您的数据。
2. 编写一个采用数据框和类名的实用方法（此处为 SampleRecord）。
3. 使用自省来读取注释（如果需要定义特定属性，最终可以添加一些）。
4. 在数据框上使用 withColumnRenamed() 重命名列。
5. 返回修改后的数据框。

喂