Use struct tags when generating via parquet

parsyl · Sep 8, 2019 · c2aaf24 · c2aaf24
1 parent 5f9dca5
commit c2aaf24
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/examples/via_parquet/main.go b/examples/via_parquet/main.go
@@ -26,8 +26,4 @@ func main() {
 		r.Scan(&p)
 		enc.Encode(p)
 	}
-
-	if err := r.Error(); err != nil {
-		log.Fatal(err)
-	}
 }
diff --git a/internal/structs/structs.go b/internal/structs/structs.go
@@ -53,7 +53,7 @@ func field(elem *sch.SchemaElement) string {
 	if elem.RepetitionType != nil && *elem.RepetitionType == sch.FieldRepetitionType_OPTIONAL {
 		ptr = "*"
 	}
-	return fmt.Sprintf("%s %s%s", n, ptr, t)
+	return fmt.Sprintf("%s %s%s `parquet:\"%s\"`", n, ptr, t, elem.Name)
 }
 
 func getType(t string) string {