feat:#36 updated feed service with more protobuf endpoints

This commit is contained in:
Elmar Kresse
2024-06-18 23:46:23 +02:00
parent 4792b07e8c
commit acc048a80a
10 changed files with 197 additions and 69 deletions

View File

@ -9,7 +9,7 @@ build-ical:
gen: gen:
@protoc \ @protoc \
--proto_path=protobuf "protobuf/modules.proto" \ --proto_path=protobuf "protobuf/*" \
--go_out="common/genproto/modules" \ --go_out="common/genproto/modules" \
--go_opt=paths=source_relative \ --go_opt=paths=source_relative \
--go-grpc_out="common/genproto/modules" \ --go-grpc_out="common/genproto/modules" \

View File

@ -37,7 +37,7 @@ func SaveFeed(feed model.Feed, collection *models.Collection, app *pocketbase.Po
return record, nil return record, nil
} }
func FindFeedByToken(token string, app *pocketbase.PocketBase) (*model.Feed, error) { func FindFeedByToken(app *pocketbase.PocketBase, token string) (*model.Feed, error) {
record, err := app.Dao().FindRecordById("feeds", token) record, err := app.Dao().FindRecordById("feeds", token)

View File

@ -0,0 +1,33 @@
package grpc
import (
"context"
"github.com/pocketbase/pocketbase"
pb "htwkalender/common/genproto/modules"
"htwkalender/data-manager/service/db"
)
type FeedServiceHandler struct {
app *pocketbase.PocketBase
pb.UnimplementedFeedServiceServer
}
func (s *FeedServiceHandler) GetFeed(ctx context.Context, in *pb.GetFeedRequest) (*pb.GetFeedResponse, error) {
s.app.Logger().Info(
"Protobuf - GetFeed",
"uuid", in.Id,
)
// get feed from database by UUID
feed, err := db.FindFeedByToken(s.app, in.Id)
if err != nil {
return nil, err
}
// Implement your logic here to fetch feed data based on the UUID
// Example response
return &pb.GetFeedResponse{
Feed: feedToProto(feed),
}, nil
}

View File

@ -31,3 +31,13 @@ func eventsToProto(events model.Events) []*pb.Event {
} }
return protoEvents return protoEvents
} }
func feedToProto(feed *model.Feed) *pb.Feed {
return &pb.Feed{
Id: feed.Id,
Created: feed.Created.String(),
Updated: feed.Updated.String(),
Retrieved: feed.Retrieved.String(),
Modules: feed.Modules,
}
}

View File

@ -0,0 +1,69 @@
package grpc
import (
"context"
"github.com/pocketbase/pocketbase"
pb "htwkalender/common/genproto/modules"
"htwkalender/data-manager/service/db"
)
type ModuleServiceHandler struct {
app *pocketbase.PocketBase
pb.UnimplementedModuleServiceServer
}
func (s *ModuleServiceHandler) GetModule(ctx context.Context, in *pb.GetModuleRequest) (*pb.GetModuleResponse, error) {
s.app.Logger().Info(
"Protobuf - GetModule",
"uuid", in.Uuid,
)
// get module from database by UUID
module, err := db.FindModuleByUUID(s.app, in.Uuid)
if err != nil {
return nil, err
}
events, err := db.FindAllEventsByModule(s.app, module)
if err != nil {
return nil, err
}
//map module Events to proto struct Events
protoEvents := make([]*pb.Event, 0)
for _, event := range events {
protoEvents = append(protoEvents, eventToProto(&event))
}
//map module to proto struct
protoModule := &pb.Module{
Uuid: module.UUID,
Name: module.Name,
Prof: module.Prof,
Course: module.Course,
Semester: module.Semester,
Events: protoEvents,
}
// Implement your logic here to fetch module data based on the UUID
// Example response
return &pb.GetModuleResponse{
Module: protoModule,
}, nil
}
func (s *ModuleServiceHandler) GetEventsForModules(ctx context.Context, in *pb.GetModulesRequest) (*pb.GetEventsResponse, error) {
s.app.Logger().Info(
"Protobuf - GetEventsForModules",
"uuids", in.Uuids,
)
events, err := db.GetPlanForModules(s.app, in.Uuids)
if err != nil {
return nil, err
}
return &pb.GetEventsResponse{
Events: eventsToProto(events),
}, nil
}

View File

@ -1,9 +1,7 @@
package grpc package grpc
import ( import (
"context"
"github.com/pocketbase/pocketbase" "github.com/pocketbase/pocketbase"
"htwkalender/data-manager/service/db"
"log" "log"
"net" "net"
@ -11,78 +9,23 @@ import (
pb "htwkalender/common/genproto/modules" pb "htwkalender/common/genproto/modules"
) )
type ModulesGrpcHandler struct {
app *pocketbase.PocketBase
pb.UnimplementedModuleServiceServer
}
func (s *ModulesGrpcHandler) GetModule(ctx context.Context, in *pb.GetModuleRequest) (*pb.GetModuleResponse, error) {
s.app.Logger().Info(
"Protobuf - GetModule",
"uuid", in.Uuid,
)
// get module from database by UUID
module, err := db.FindModuleByUUID(s.app, in.Uuid)
if err != nil {
return nil, err
}
events, err := db.FindAllEventsByModule(s.app, module)
if err != nil {
return nil, err
}
//map module Events to proto struct Events
protoEvents := make([]*pb.Event, 0)
for _, event := range events {
protoEvents = append(protoEvents, eventToProto(&event))
}
//map module to proto struct
protoModule := &pb.Module{
Uuid: module.UUID,
Name: module.Name,
Prof: module.Prof,
Course: module.Course,
Semester: module.Semester,
Events: protoEvents,
}
// Implement your logic here to fetch module data based on the UUID
// Example response
return &pb.GetModuleResponse{
Module: protoModule,
}, nil
}
func StartGRPCServer(app *pocketbase.PocketBase) { func StartGRPCServer(app *pocketbase.PocketBase) {
lis, err := net.Listen("tcp", ":50051") lis, err := net.Listen("tcp", ":50051")
if err != nil { if err != nil {
log.Fatalf("failed to listen: %v", err) log.Fatalf("failed to listen: %v", err)
} }
s := grpc.NewServer() s := grpc.NewServer()
pb.RegisterModuleServiceServer(s, &ModulesGrpcHandler{
pb.RegisterModuleServiceServer(s, &ModuleServiceHandler{
app: app, app: app,
}) })
pb.RegisterFeedServiceServer(s, &FeedServiceHandler{
app: app,
})
log.Printf("server listening at %v", lis.Addr()) log.Printf("server listening at %v", lis.Addr())
if err := s.Serve(lis); err != nil { if err := s.Serve(lis); err != nil {
log.Fatalf("failed to serve: %v", err) log.Fatalf("failed to serve: %v", err)
} }
} }
func (s *ModulesGrpcHandler) GetEventsForModules(ctx context.Context, in *pb.GetModulesRequest) (*pb.GetEventsResponse, error) {
s.app.Logger().Info(
"Protobuf - GetEventsForModules",
"uuids", in.Uuids,
)
events, err := db.GetPlanForModules(s.app, in.Uuids)
if err != nil {
return nil, err
}
return &pb.GetEventsResponse{
Events: eventsToProto(events),
}, nil
}

View File

@ -19,7 +19,6 @@ package model
import ( import (
"encoding/json" "encoding/json"
"fmt" "fmt"
"log/slog"
"strings" "strings"
"time" "time"
) )
@ -88,7 +87,6 @@ func (jt *JSONTime) UnmarshalJSON(b []byte) error {
func ToJSONTime(timeString string) JSONTime { func ToJSONTime(timeString string) JSONTime {
t, err := time.Parse(DefaultDateLayout, timeString) t, err := time.Parse(DefaultDateLayout, timeString)
if err != nil { if err != nil {
slog.Error("error parsing time string: ", "error", err)
return JSONTime(time.Time{}) return JSONTime(time.Time{})
} }
return JSONTime(t) return JSONTime(t)

View File

@ -0,0 +1,52 @@
package grpc
import (
"context"
"github.com/goccy/go-json"
"google.golang.org/grpc"
pb "htwkalender/common/genproto/modules"
"htwkalender/ical/model"
"log/slog"
"time"
)
func GetFeed(feedId string, conn *grpc.ClientConn) (model.FeedRecord, error) {
c := pb.NewFeedServiceClient(conn)
ctx, cancel := context.WithTimeout(context.Background(), time.Second)
defer cancel()
r, err := c.GetFeed(ctx, &pb.GetFeedRequest{Id: feedId})
if err != nil {
slog.Error("could not get modules: %v", "error", err)
return model.FeedRecord{}, err
}
feed, err := protoToFeed(r.GetFeed())
if err != nil {
slog.Error("could not convert feed: %v", "error", err)
return model.FeedRecord{}, err
}
return feed, nil
}
func protoToFeed(feed *pb.Feed) (model.FeedRecord, error) {
// unmarshal the []model.FeedModule from json
var modules []model.FeedModule
err := json.Unmarshal([]byte(feed.Modules), &modules)
if err != nil {
slog.Error("could not unmarshal modules: %v", "error", err)
return model.FeedRecord{}, err
}
return model.FeedRecord{
BaseModel: model.BaseModel{
Id: feed.Id,
Created: model.ToJSONTime(feed.Created),
Updated: model.ToJSONTime(feed.Updated),
},
Retrieved: model.ToJSONTime(feed.Retrieved),
Modules: modules,
}, nil
}

View File

@ -14,7 +14,7 @@ const expirationTime = 5 * time.Minute
func Feed(app model.AppType, token string) (string, error) { func Feed(app model.AppType, token string) (string, error) {
// get feed by token // get feed by token
feed, err := connector.GetFeedByToken(app.DataManagerURL, token) feed, err := htwkalenderGrpc.GetFeed(token, app.GrpcClient)
if err != nil { if err != nil {
return "", err return "", err
} }

View File

@ -0,0 +1,23 @@
syntax = "proto3";
option go_package = "htwkalender/common/modules";
service FeedService {
rpc GetFeed(GetFeedRequest) returns (GetFeedResponse) {}
}
message Feed {
string id = 1;
string modules = 2;
string retrieved = 3;
string created = 4;
string updated = 5;
}
message GetFeedRequest {
string id = 1;
}
message GetFeedResponse {
Feed feed = 1;
}