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
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user