ryanc/haberdasher-twirp
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Rename model conversion functions

Browse Source
This commit is contained in:
Ryan Cavicchioni 2021-07-27 18:00:35 -05:00
parent a35901b976
commit e2cf17ff80
Signed by: ryanc
GPG Key ID: 877EEDAF9245103D
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
internal/haberdasherserver/server.go
View File

@ -34,7 +34,7 @@ type Store struct {
db *bolt.DB db *bolt.DB
} }
func HatToHatModel(h *pb.Hat) Hat { func PbToHatModel(h *pb.Hat) Hat {
return Hat{ return Hat{
Inches: h.Inches, Inches: h.Inches,
Color: h.Color, Color: h.Color,
@ -42,7 +42,7 @@ func HatToHatModel(h *pb.Hat) Hat {
} }
} }
func HatModelToHat(h Hat) *pb.Hat { func HatModelToPb(h Hat) *pb.Hat {
return &pb.Hat{ return &pb.Hat{
Inches: h.Inches, Inches: h.Inches,
Color: h.Color, Color: h.Color,
@ -50,16 +50,16 @@ func HatModelToHat(h Hat) *pb.Hat {
} }
} }
func HatsModelToHats(hs []Hat) (hats *pb.Hats) { func HatsModelToPb(hs []Hat) (hats *pb.Hats) {
hats = &pb.Hats{} hats = &pb.Hats{}
for _, h := range hs { for _, h := range hs {
hat := HatModelToHat(h) hat := HatModelToPb(h)
hats.Hats = append(hats.Hats, hat) hats.Hats = append(hats.Hats, hat)
} }
return hats return hats
} }
func HatQueryToHatQueryModel(q *pb.HatQuery) HatQuery { func PbToHatQueryModel(q *pb.HatQuery) HatQuery {
return HatQuery{ return HatQuery{
Limit: q.Limit, Limit: q.Limit,
} }
@ -86,16 +86,16 @@ func (s *Server) MakeHat(ctx context.Context, size *pb.Size) (hat *pb.Hat, err e
st.SaveHat(h) st.SaveHat(h)
return HatModelToHat(h), nil return HatModelToPb(h), nil
} }
func (s *Server) ListHats(ctx context.Context, q *pb.HatQuery) (hats *pb.Hats, err error) { func (s *Server) ListHats(ctx context.Context, q *pb.HatQuery) (hats *pb.Hats, err error) {
st, _ := NewStore(s.DBPath, 0600, nil) st, _ := NewStore(s.DBPath, 0600, nil)
defer st.Close() defer st.Close()
hs, err := st.ListHats(HatQueryToHatQueryModel(q)) hs, err := st.ListHats(PbToHatQueryModel(q))
return HatsModelToHats(hs), nil return HatsModelToPb(hs), nil
} }
func itob(v uint64) []byte { func itob(v uint64) []byte {