server.go

server.go

package main

import (
	"context"
	"fmt"
	"google.golang.org/grpc"
	"net"
	agentPb "github.com/squzy/squzy_generated/generated/agent/proto/v1"
	"time"
)

type server struct {

}

func (s server) Register(_ context.Context,rq *agentPb.RegisterRequest) (*agentPb.RegisterResponse, error) {
	fmt.Println(rq.HostInfo.HostName)
	fmt.Println(rq.HostInfo.Os)
	fmt.Println(rq.HostInfo.PlatformInfo.Name)
	return &agentPb.RegisterResponse{
		Id: "123",
	}, nil
}

func (s server) UnRegister(_ context.Context, rq *agentPb.UnRegisterRequest) (*agentPb.UnRegisterResponse, error) {
	fmt.Println(rq.Id)

	return &agentPb.UnRegisterResponse{
		Id: "123",
	}, nil
}

func (s server) SendStat(r agentPb.AgentServer_SendStatServer) error {
	for {
		stat, err := r.Recv()
		if err !=nil {
			continue
		}
		fmt.Println(stat)
	}
}

func (s server) GetList(context.Context, *agentPb.GetListRequest) (*agentPb.GetListResponse, error) {
	panic("implement me")
}

func main() {
	fmt.Println(time.Second)
	lis, _ := net.Listen("tcp", fmt.Sprintf(":%d", 9091))
	grpcServer := grpc.NewServer()
	agentPb.RegisterAgentServerServer(grpcServer, &server{})
	grpcServer.Serve(lis)
}