wyh
/
grpcservice
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

格式化输出grpc

This commit is contained in:
wyh 2022-03-03 10:38:10 +08:00
parent 34a2ded916
commit 112ae25709
2 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
data.go → data/data.go
View File

@ -1,8 +1,8 @@
package main package data
import "myschools.me/wyh/grpcservice/pb" import "myschools.me/wyh/grpcservice/pb"
var employees = []pb.Employee{ var Employees = []pb.Employee{
{ {
Id: 1, Id: 1,
No: 1994, No: 1994,

9
server/employee-server.go
View File

@ -7,6 +7,7 @@ import (
"time" "time"
"google.golang.org/grpc/metadata" "google.golang.org/grpc/metadata"
"myschools.me/wyh/grpcservice/data"
"myschools.me/wyh/grpcservice/pb" "myschools.me/wyh/grpcservice/pb"
) )
@ -15,7 +16,7 @@ type EmployeeService struct {
} }
func (msg *EmployeeService) GetByNo(ctx context.Context, req *pb.GetByNoRequset) (*pb.EmployeeResponse, error) { func (msg *EmployeeService) GetByNo(ctx context.Context, req *pb.GetByNoRequset) (*pb.EmployeeResponse, error) {
for _, e := range employees { for _, e := range data.Employees {
if req.No == e.No { if req.No == e.No {
return &pb.EmployeeResponse{ return &pb.EmployeeResponse{
Employee: &e, Employee: &e,
@ -26,7 +27,7 @@ func (msg *EmployeeService) GetByNo(ctx context.Context, req *pb.GetByNoRequset)
} }
func (msg *EmployeeService) GetAll(req *pb.GetAllRequest, steam pb.EmployeeService_GetAllServer) error { func (msg *EmployeeService) GetAll(req *pb.GetAllRequest, steam pb.EmployeeService_GetAllServer) error {
for _, e := range employees { for _, e := range data.Employees {
steam.Send(&pb.EmployeeResponse{ steam.Send(&pb.EmployeeResponse{
Employee: &e, Employee: &e,
@ -67,10 +68,10 @@ func (msg *EmployeeService) SaveAll(stream pb.EmployeeService_SaveAllServer) err
if err != nil { if err != nil {
return err return err
} }
employees = append(employees, *emp.Employee) data.Employees = append(data.Employees, *emp.Employee)
stream.Send(&pb.EmployeeResponse{Employee: emp.Employee}) stream.Send(&pb.EmployeeResponse{Employee: emp.Employee})
} }
for _, emp := range employees { for _, emp := range data.Employees {
fmt.Println(emp) fmt.Println(emp)
} }
return nil return nil