Parsing and bridging between the CSV struct and proto done
This commit is contained in:
cdricms
@@ -1,54 +1,13 @@
|
||||
package parsing
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"encoding/csv"
|
||||
"reflect"
|
||||
"strconv"
|
||||
"encoding/csv"
|
||||
"errors"
|
||||
"reflect"
|
||||
"strconv"
|
||||
"git.cems.dev/cdricms/bdooc/proto"
|
||||
)
|
||||
|
||||
type Employee struct {
|
||||
EmpID uint `csv:"Emp ID"`
|
||||
NamePrefix string `csv:"Name Prefix"`
|
||||
FirstName string `csv:"First Name"`
|
||||
MiddleInitial string `csv:"Middle Initial"`
|
||||
LastName string `csv:"Last Name"`
|
||||
Gender Gender `csv:"Gender"`
|
||||
Email string `csv:"E Mail"`
|
||||
FathersName string `csv:"Father's Name"`
|
||||
MothersName string `csv:"Mothers's Name"`
|
||||
MothersMaidenName string `csv:"Mother's Maiden Name"`
|
||||
Birthdate string `csv:"Date of Birth"`
|
||||
BirthTime string `csv:"Time of Birth"`
|
||||
WeightKg uint16 `csv:"Weight in Kgs."`
|
||||
JoiningDate string `csv:"Date of Joining"`
|
||||
JoiningQuarter Quarter `csv:"Quarter of Joining"`
|
||||
JoiningHalf Half `csv:"Half of Joining"`
|
||||
JoiningYear uint16 `csv:"Year of Joining"`
|
||||
JoiningMonth uint8 `csv:"Month of Joining"`
|
||||
JoiningMonthName Month `csv:"Month Name of Joining"`
|
||||
JoiningMonthNameShort MonthShort `csv:"Short Month"`
|
||||
JoiningMonthDay uint8 `csv:"Day of Joining"`
|
||||
JoiningWeekDay WeekDay `csv:"DOW of Joining"`
|
||||
JoiningWeekDayShort WeekDayShort `csv:"Short DOW"`
|
||||
YearsOfService float32 `csv:"Age in Company (Years)"`
|
||||
Salary uint32 `csv:"Salary"`
|
||||
LatestHikePercentage string `csv:"Last % Hike"`
|
||||
SSN string `csv:"SSN"`
|
||||
PhoneNumber string `csv:"Phone No."`
|
||||
PlaceName string `csv:"Place Name"`
|
||||
County string `csv:"County"`
|
||||
City string `csv:"City"`
|
||||
State string `csv:"State"`
|
||||
Zip uint32 `csv:"Zip"`
|
||||
Region string `csv:"Region"`
|
||||
Username string `csv:"User Name"`
|
||||
Password string `csv:"Password"`
|
||||
// Empty slot
|
||||
// Empty slot
|
||||
// Empty slot
|
||||
}
|
||||
|
||||
func UnmarshalEmployees(reader *csv.Reader) ([]Employee, error) {
|
||||
reader.Comma = ','
|
||||
headers, err := reader.Read()
|
||||
@@ -56,19 +15,18 @@ func UnmarshalEmployees(reader *csv.Reader) ([]Employee, error) {
|
||||
return nil, errors.New("Couldn't read the headers.")
|
||||
}
|
||||
|
||||
e := Employee{}
|
||||
refet := reflect.TypeOf(e)
|
||||
e := Employee{}
|
||||
refet := reflect.TypeOf(e)
|
||||
tagMap := make(map[string]int)
|
||||
headersLength := len(headers)
|
||||
numFields := refet.NumField()
|
||||
for i := 0; i < numFields && i < headersLength; i++ {
|
||||
field := refet.Field(i)
|
||||
if field.Tag.Get("csv") == headers[i] {
|
||||
tagMap[field.Name] = i
|
||||
}
|
||||
headersLength := len(headers)
|
||||
numFields := refet.NumField()
|
||||
for i := 0; i < numFields && i < headersLength; i++ {
|
||||
field := refet.Field(i)
|
||||
if field.Tag.Get("csv") == headers[i] {
|
||||
tagMap[field.Name] = i
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
var employees []Employee
|
||||
for {
|
||||
record, err := reader.Read()
|
||||
@@ -85,10 +43,6 @@ func UnmarshalEmployees(reader *csv.Reader) ([]Employee, error) {
|
||||
switch field.Type().Name() {
|
||||
case "string":
|
||||
field.SetString(record[index])
|
||||
case "uint":
|
||||
if intValue, err := strconv.ParseUint(record[index], 10, 64); err == nil {
|
||||
field.SetUint(intValue)
|
||||
}
|
||||
case "uint8":
|
||||
if intValue, err := strconv.ParseUint(record[index], 10, 8); err == nil {
|
||||
field.SetUint(intValue)
|
||||
@@ -120,10 +74,117 @@ func UnmarshalEmployees(reader *csv.Reader) ([]Employee, error) {
|
||||
case "WeekDayShort":
|
||||
field.Set(reflect.ValueOf(WeekDayShort(string(record[index]))))
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
employees = append(employees, employee)
|
||||
}
|
||||
|
||||
return employees, nil
|
||||
}
|
||||
|
||||
func (e Employee) MapToProto() *proto.Employee {
|
||||
|
||||
employee := &proto.Employee{}
|
||||
|
||||
employee.EmpId = e.EmpID
|
||||
employee.NamePrefix = e.NamePrefix
|
||||
employee.FirstName = e.FirstName
|
||||
employee.MiddleInitial = e.MiddleInitial
|
||||
employee.LastName = e.LastName
|
||||
employee.Gender = proto.Gender(e.Gender)
|
||||
employee.Email = e.Email
|
||||
employee.FathersName = e.FathersName
|
||||
employee.MothersName = e.MothersName
|
||||
employee.MothersMaidenName = e.MothersMaidenName
|
||||
employee.Birthdate = e.Birthdate
|
||||
employee.BirthTime = e.BirthTime
|
||||
employee.WeightKg = uint32(e.WeightKg)
|
||||
employee.JoiningDate = e.JoiningDate
|
||||
employee.JoiningQuarter = proto.Quarter(e.JoiningQuarter.Uint8())
|
||||
employee.JoiningHalf = proto.Half(e.JoiningHalf.Uint8())
|
||||
employee.JoiningYear = uint32(e.JoiningYear)
|
||||
employee.JoiningMonth = uint32(e.JoiningMonth)
|
||||
employee.JoiningMonthName = proto.Month(e.JoiningMonthName.Uint8())
|
||||
employee.JoiningMonthNameShort = proto.MonthShort(e.JoiningMonthNameShort.Uint8())
|
||||
employee.JoiningMonthDay = uint32(e.JoiningMonthDay)
|
||||
employee.JoiningWeekDay = proto.WeekDay(e.JoiningWeekDay.Uint8())
|
||||
employee.JoiningWeekDayShort = proto.WeekDayShort(e.JoiningWeekDayShort.Uint8())
|
||||
employee.YearsOfService = float32(e.YearsOfService)
|
||||
employee.Salary = uint32(e.Salary)
|
||||
employee.LatestHikePercentage = e.LatestHikePercentage
|
||||
employee.Ssn = e.SSN
|
||||
employee.PhoneNumber = e.PhoneNumber
|
||||
employee.PlaceName = e.PlaceName
|
||||
employee.County = e.County
|
||||
employee.City = e.City
|
||||
employee.State = e.State
|
||||
employee.Zip = uint32(e.Zip)
|
||||
employee.Region = e.Region
|
||||
employee.Username = e.Username
|
||||
employee.Password = e.Password
|
||||
|
||||
return employee
|
||||
}
|
||||
|
||||
func MapToProto(es []Employee) *proto.EmployeeList {
|
||||
|
||||
employeeList := &proto.EmployeeList{}
|
||||
|
||||
for _, employee := range es {
|
||||
employeeList.Employees = append(employeeList.Employees, employee.MapToProto())
|
||||
}
|
||||
|
||||
return employeeList
|
||||
}
|
||||
|
||||
func MapFromProto(e *proto.Employee) Employee {
|
||||
employee := Employee{}
|
||||
employee.EmpID = e.EmpId
|
||||
employee.NamePrefix = e.NamePrefix
|
||||
employee.FirstName = e.FirstName
|
||||
employee.MiddleInitial = e.MiddleInitial
|
||||
employee.LastName = e.LastName
|
||||
employee.Gender = Gender(e.Gender)
|
||||
employee.Email = e.Email
|
||||
employee.FathersName = e.FathersName
|
||||
employee.MothersName = e.MothersName
|
||||
employee.MothersMaidenName = e.MothersMaidenName
|
||||
employee.Birthdate = e.Birthdate
|
||||
employee.BirthTime = e.BirthTime
|
||||
employee.WeightKg = uint16(e.WeightKg)
|
||||
employee.JoiningDate = e.JoiningDate
|
||||
employee.JoiningQuarter = Quarter(e.JoiningQuarter.String())
|
||||
employee.JoiningHalf = Half(e.JoiningHalf.String())
|
||||
employee.JoiningYear = uint16(e.JoiningYear)
|
||||
employee.JoiningMonth = uint8(e.JoiningMonth)
|
||||
employee.JoiningMonthName = Month(e.JoiningMonthName.String())
|
||||
employee.JoiningMonthNameShort = MonthShort(e.JoiningMonthNameShort.String())
|
||||
employee.JoiningMonthDay = uint8(e.JoiningMonthDay)
|
||||
employee.JoiningWeekDay = WeekDay(e.JoiningWeekDay.String())
|
||||
employee.JoiningWeekDayShort = WeekDayShort(e.JoiningWeekDayShort.String())
|
||||
employee.YearsOfService = float32(e.YearsOfService)
|
||||
employee.Salary = uint32(e.Salary)
|
||||
employee.LatestHikePercentage = e.LatestHikePercentage
|
||||
employee.SSN = e.Ssn
|
||||
employee.PhoneNumber = e.PhoneNumber
|
||||
employee.PlaceName = e.PlaceName
|
||||
employee.County = e.County
|
||||
employee.City = e.City
|
||||
employee.State = e.State
|
||||
employee.Zip = uint32(e.Zip)
|
||||
employee.Region = e.Region
|
||||
employee.Username = e.Username
|
||||
employee.Password = e.Password
|
||||
return employee
|
||||
}
|
||||
|
||||
func MapFromProtoList(el *proto.EmployeeList) []Employee {
|
||||
|
||||
employees := []Employee{}
|
||||
|
||||
for _, pemployee := range el.GetEmployees() {
|
||||
employees = append(employees, MapFromProto(pemployee))
|
||||
}
|
||||
|
||||
return employees
|
||||
}
|
||||
|
||||
@@ -1,82 +0,0 @@
|
||||
package parsing
|
||||
|
||||
type Gender rune
|
||||
|
||||
const (
|
||||
Male Gender = 'M'
|
||||
Female Gender = 'F'
|
||||
)
|
||||
|
||||
type Quarter string
|
||||
|
||||
const (
|
||||
QuarterOne Quarter = "Q1"
|
||||
QuarterTwo Quarter = "Q2"
|
||||
QuarterThree Quarter = "Q3"
|
||||
QuarterFour Quarter = "Q4"
|
||||
)
|
||||
|
||||
type Half string
|
||||
|
||||
const (
|
||||
HalfOne Half = "H1"
|
||||
HalfTwo Half = "H2"
|
||||
)
|
||||
|
||||
type Month string
|
||||
|
||||
const (
|
||||
January Month = "January"
|
||||
February Month = "February"
|
||||
March Month = "March"
|
||||
April Month = "April"
|
||||
May Month = "May"
|
||||
June Month = "June"
|
||||
July Month = "July"
|
||||
August Month = "August"
|
||||
September Month = "September"
|
||||
October Month = "October"
|
||||
November Month = "November"
|
||||
December Month = "December"
|
||||
)
|
||||
|
||||
type MonthShort string
|
||||
|
||||
const (
|
||||
Jan MonthShort = "Jan"
|
||||
Feb MonthShort = "Feb"
|
||||
Mar MonthShort = "Mar"
|
||||
Apr MonthShort = "Apr"
|
||||
MayShort MonthShort = "May"
|
||||
Jun MonthShort = "Jun"
|
||||
Jul MonthShort = "Jul"
|
||||
Aug MonthShort = "Aug"
|
||||
Sep MonthShort = "Sep"
|
||||
Oct MonthShort = "Oct"
|
||||
Nov MonthShort = "Nov"
|
||||
Dec MonthShort = "Dec"
|
||||
)
|
||||
|
||||
type WeekDay string
|
||||
|
||||
const (
|
||||
Monday WeekDay = "Monday"
|
||||
Tuesday WeekDay = "Tuesday"
|
||||
Wednesday WeekDay = "Wednesday"
|
||||
Thursday WeekDay = "Thursday"
|
||||
Friday WeekDay = "Friday"
|
||||
Saturday WeekDay = "Saturday"
|
||||
Sunday WeekDay = "Sunday"
|
||||
)
|
||||
|
||||
type WeekDayShort string
|
||||
|
||||
const (
|
||||
Mon WeekDayShort = "Mon"
|
||||
Tue WeekDayShort = "Tue"
|
||||
Wed WeekDayShort = "Wed"
|
||||
Thu WeekDayShort = "Thu"
|
||||
Fri WeekDayShort = "Fri"
|
||||
Sat WeekDayShort = "Sat"
|
||||
Sun WeekDayShort = "Sun"
|
||||
)
|
||||
250
parsing/types.go
Normal file
250
parsing/types.go
Normal file
@@ -0,0 +1,250 @@
|
||||
package parsing
|
||||
|
||||
type Gender rune
|
||||
|
||||
const (
|
||||
Male Gender = 'M'
|
||||
Female Gender = 'F'
|
||||
)
|
||||
|
||||
type Quarter string
|
||||
|
||||
const (
|
||||
QuarterOne Quarter = "Q1"
|
||||
QuarterTwo Quarter = "Q2"
|
||||
QuarterThree Quarter = "Q3"
|
||||
QuarterFour Quarter = "Q4"
|
||||
)
|
||||
|
||||
func (x Quarter) Uint8() uint8 {
|
||||
switch x {
|
||||
case QuarterOne:
|
||||
return 1
|
||||
case QuarterTwo:
|
||||
return 2
|
||||
case QuarterThree:
|
||||
return 3
|
||||
case QuarterFour:
|
||||
return 4
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
type Half string
|
||||
|
||||
const (
|
||||
HalfOne Half = "H1"
|
||||
HalfTwo Half = "H2"
|
||||
)
|
||||
|
||||
func (x Half) Uint8() uint8 {
|
||||
switch x {
|
||||
case HalfOne:
|
||||
return 1
|
||||
case HalfTwo:
|
||||
return 2
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
type Month string
|
||||
|
||||
const (
|
||||
January Month = "January"
|
||||
February Month = "February"
|
||||
March Month = "March"
|
||||
April Month = "April"
|
||||
May Month = "May"
|
||||
June Month = "June"
|
||||
July Month = "July"
|
||||
August Month = "August"
|
||||
September Month = "September"
|
||||
October Month = "October"
|
||||
November Month = "November"
|
||||
December Month = "December"
|
||||
)
|
||||
|
||||
func (x Month) Uint8() uint8 {
|
||||
switch x {
|
||||
case January:
|
||||
return 1
|
||||
case February:
|
||||
return 2
|
||||
case March:
|
||||
return 3
|
||||
case April:
|
||||
return 4
|
||||
case May:
|
||||
return 5
|
||||
case June:
|
||||
return 6
|
||||
case July:
|
||||
return 7
|
||||
case August:
|
||||
return 8
|
||||
case September:
|
||||
return 9
|
||||
case October:
|
||||
return 10
|
||||
case November:
|
||||
return 11
|
||||
case December:
|
||||
return 12
|
||||
default:
|
||||
return 0
|
||||
}
|
||||
}
|
||||
|
||||
type MonthShort string
|
||||
|
||||
const (
|
||||
Jan MonthShort = "Jan"
|
||||
Feb MonthShort = "Feb"
|
||||
Mar MonthShort = "Mar"
|
||||
Apr MonthShort = "Apr"
|
||||
MayShort MonthShort = "May"
|
||||
Jun MonthShort = "Jun"
|
||||
Jul MonthShort = "Jul"
|
||||
Aug MonthShort = "Aug"
|
||||
Sep MonthShort = "Sep"
|
||||
Oct MonthShort = "Oct"
|
||||
Nov MonthShort = "Nov"
|
||||
Dec MonthShort = "Dec"
|
||||
)
|
||||
|
||||
func (x MonthShort) Uint8() uint8 {
|
||||
switch x {
|
||||
case Jan:
|
||||
return 1
|
||||
case Feb:
|
||||
return 2
|
||||
case Mar:
|
||||
return 3
|
||||
case Apr:
|
||||
return 4
|
||||
case MayShort:
|
||||
return 5
|
||||
case Jun:
|
||||
return 6
|
||||
case Jul:
|
||||
return 7
|
||||
case Aug:
|
||||
return 8
|
||||
case Sep:
|
||||
return 9
|
||||
case Oct:
|
||||
return 10
|
||||
case Nov:
|
||||
return 11
|
||||
case Dec:
|
||||
return 12
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
type WeekDay string
|
||||
|
||||
const (
|
||||
Monday WeekDay = "Monday"
|
||||
Tuesday WeekDay = "Tuesday"
|
||||
Wednesday WeekDay = "Wednesday"
|
||||
Thursday WeekDay = "Thursday"
|
||||
Friday WeekDay = "Friday"
|
||||
Saturday WeekDay = "Saturday"
|
||||
Sunday WeekDay = "Sunday"
|
||||
)
|
||||
|
||||
func (x WeekDay) Uint8() uint8 {
|
||||
switch x {
|
||||
case Monday:
|
||||
return 1
|
||||
case Tuesday:
|
||||
return 2
|
||||
case Wednesday:
|
||||
return 3
|
||||
case Thursday:
|
||||
return 4
|
||||
case Friday:
|
||||
return 5
|
||||
case Saturday:
|
||||
return 6
|
||||
case Sunday:
|
||||
return 7
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
type WeekDayShort string
|
||||
|
||||
const (
|
||||
Mon WeekDayShort = "Mon"
|
||||
Tue WeekDayShort = "Tue"
|
||||
Wed WeekDayShort = "Wed"
|
||||
Thu WeekDayShort = "Thu"
|
||||
Fri WeekDayShort = "Fri"
|
||||
Sat WeekDayShort = "Sat"
|
||||
Sun WeekDayShort = "Sun"
|
||||
)
|
||||
|
||||
func (x WeekDayShort) Uint8() uint8 {
|
||||
switch x {
|
||||
case Mon:
|
||||
return 1
|
||||
case Tue:
|
||||
return 2
|
||||
case Wed:
|
||||
return 3
|
||||
case Thu:
|
||||
return 4
|
||||
case Fri:
|
||||
return 5
|
||||
case Sat:
|
||||
return 6
|
||||
case Sun:
|
||||
return 7
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
|
||||
type Employee struct {
|
||||
EmpID uint32 `csv:"Emp ID"`
|
||||
NamePrefix string `csv:"Name Prefix"`
|
||||
FirstName string `csv:"First Name"`
|
||||
MiddleInitial string `csv:"Middle Initial"`
|
||||
LastName string `csv:"Last Name"`
|
||||
Gender Gender `csv:"Gender"`
|
||||
Email string `csv:"E Mail"`
|
||||
FathersName string `csv:"Father's Name"`
|
||||
MothersName string `csv:"Mothers's Name"`
|
||||
MothersMaidenName string `csv:"Mother's Maiden Name"`
|
||||
Birthdate string `csv:"Date of Birth"`
|
||||
BirthTime string `csv:"Time of Birth"`
|
||||
WeightKg uint16 `csv:"Weight in Kgs."`
|
||||
JoiningDate string `csv:"Date of Joining"`
|
||||
JoiningQuarter Quarter `csv:"Quarter of Joining"`
|
||||
JoiningHalf Half `csv:"Half of Joining"`
|
||||
JoiningYear uint16 `csv:"Year of Joining"`
|
||||
JoiningMonth uint8 `csv:"Month of Joining"`
|
||||
JoiningMonthName Month `csv:"Month Name of Joining"`
|
||||
JoiningMonthNameShort MonthShort `csv:"Short Month"`
|
||||
JoiningMonthDay uint8 `csv:"Day of Joining"`
|
||||
JoiningWeekDay WeekDay `csv:"DOW of Joining"`
|
||||
JoiningWeekDayShort WeekDayShort `csv:"Short DOW"`
|
||||
YearsOfService float32 `csv:"Age in Company (Years)"`
|
||||
Salary uint32 `csv:"Salary"`
|
||||
LatestHikePercentage string `csv:"Last % Hike"`
|
||||
SSN string `csv:"SSN"`
|
||||
PhoneNumber string `csv:"Phone No."`
|
||||
PlaceName string `csv:"Place Name"`
|
||||
County string `csv:"County"`
|
||||
City string `csv:"City"`
|
||||
State string `csv:"State"`
|
||||
Zip uint32 `csv:"Zip"`
|
||||
Region string `csv:"Region"`
|
||||
Username string `csv:"User Name"`
|
||||
Password string `csv:"Password"`
|
||||
// Empty slot
|
||||
// Empty slot
|
||||
// Empty slot
|
||||
}
|
||||
Reference in New Issue
Block a user