DRAuthServer/service/pb_study_client.go

package service

import (
	"context"
	"log/slog"
	"time"
)
import (
	"google.golang.org/grpc"
	"google.golang.org/grpc/credentials/insecure"
)

import (
	"auth-server/common"
	pb "auth-server/rpc_idl/dr_study_pb"
)

var StudyClient *Study

func init() {
	StudyClient = new(Study)
}

type Study struct {
	basicClient pb.BasicClient
	studyClient pb.StudyClient
}

func (s *Study) Setup() {
	conn, err := grpc.NewClient(common.ServerConfig.Study, grpc.WithTransportCredentials(insecure.NewCredentials()))
	if err != nil {
		slog.Error("NewClient failed:", err)
		panic(err)
	}
	slog.Info("study conn success", "conn", conn.GetState())

	s.basicClient = pb.NewBasicClient(conn)
	s.studyClient = pb.NewStudyClient(conn)
}

func (s *Study) GetSoftwareInfo() *pb.SoftwareInfoReply {
	slog.Info("[rpc]SoftwareInfo...")
	ctx, cancel := context.WithTimeout(context.Background(), 200*time.Millisecond)
	defer cancel()
	r, err := s.basicClient.SoftwareInfo(ctx, &pb.EmptyRequest{})
	if err != nil {
		slog.Error("[rpc]SoftwareInfo failed", "err", err)
		return &pb.SoftwareInfoReply{
			Module:  "",
			Desc:    "",
			Build:   "",
			Version: "",
		}
	}
	return r
}

func (s *Study) CreateStudy(ctx context.Context, in *pb.StudyRequest) (*pb.StudyReply, error) {
	slog.Info("[rpc]CreateStudy...")
	r, err := s.studyClient.CreateStudy(ctx, in)
	if err != nil {
		slog.Error("[rpc]CreateStudy failed", "err", err)
		return r, err
	}
	return r, nil
}

func (s *Study) GetStudy(ctx context.Context, id *string) (*pb.StudyReply, error) {
	slog.Info("[rpc]GetStudy...")
	r, err := s.studyClient.GetStudy(ctx, &pb.IDRequest{
		InstanceId: id,
	})
	if err != nil {
		slog.Error("[rpc]GetStudy failed", "err", err)
		return r, err
	}
	return r, nil
}