Add handler for the gRPC server in Golang - go

Question

I have the following server/server.go definition for my gRPC server:
package main
import (
"flag"
"github.com/golang/glog"
pb "github.com/go-grpc-tutorial/pb"
"golang.org/x/net/context"
"google.golang.org/grpc"
"net"
)
// Implements of EchoServiceServer
type echoServer struct{}
func newEchoServer() pb.EchoServiceServer {
return new(echoServer)
}
func (s *echoServer) Echo(ctx context.Context, msg *pb.Message) (*pb.Message, error) {
glog.Info(msg)
return msg, nil
}
func Run() error {
listen, err := net.Listen("tcp", ":50051")
if err != nil {
return err
}
server := grpc.NewServer()
pb.RegisterEchoServiceServer(server, newEchoServer())
server.Serve(listen)
return nil
}
func main() {
flag.Parse()
defer glog.Flush()
if err := Run(); err != nil {
glog.Fatal(err)
}
}
I want the server to listen from a Kinesis stream, so whenever a message is posted to the Kinesis stream, it will consume that message and send it to the Echo() method.
So, I'm envisioning something like:
func handleKinesis(s *echoServer) error {
// wait until a message is published to the kinesis stream
records, err := readFromKinesis() // not implemented here
if err != nil {
return err
}
for _, record := records {
message := string(record.Data)
_, _ = s.Echo(context.Background(), &pb.Message{msg: message})
}
}
And the handleKinesis would run in the background, just listening for new Kinesis messages, and then call the echoServer to process that message.
How can I accomplish that?