FilesExpand file tree

master

/

main.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

55 lines (46 loc) · 1.14 KB

master

/

main.go

Copy path

Top

File metadata and controls

• Code
• Blame

55 lines (46 loc) · 1.14 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

// server/main.go

package main

import (

"log"

"net"

"strings"

"golang.org/x/net/context"

"google.golang.org/grpc"

pb "./customer"

)

const (

port = ":50051"

)

// server is used to implement customer.CustomerServer.

type server struct {

savedCustomers []*pb.CustomerRequest

}

// CreateCustomer creates a new Customer

func (s *server) CreateCustomer(ctx context.Context, in *pb.CustomerRequest) (*pb.CustomerResponse, error) {

s.savedCustomers = append(s.savedCustomers, in)

return &pb.CustomerResponse{Id: in.Id, Success: true}, nil

}

// GetCustomers returns all customers by given filter

func (s *server) GetCustomers(filter *pb.CustomerFilter, stream pb.Customer_GetCustomersServer) error {

for _, customer := range s.savedCustomers {

if filter.Keyword != "" {

if !strings.Contains(customer.Name, filter.Keyword) {

continue

}

}

if err := stream.Send(customer); err != nil {

return err

}

}

return nil

}

func main() {

lis, err := net.Listen("tcp", port)

if err != nil {

log.Fatal("failed to listen: %v", err)

}

//Create a new grpc server

s := grpc.NewServer()

pb.RegisterCustomerServer(s, &server{})

s.Serve(lis)

}