grpc-api-example/server/notes.go

package server

import (
	"context"
	"github.com/gofrs/uuid"
	notesv1 "github.com/sundowndev/grpc-api-example/proto/notes/v1"
	"sync"
)

type NotesService struct {
	notesv1.UnimplementedNotesServiceServer
	mu    *sync.RWMutex
	notes []*notesv1.Note
}

func NewNotesService() *NotesService {
	return &NotesService{}
}

func (s *NotesService) ListNotes(_ *notesv1.ListNotesRequest, srv notesv1.NotesService_ListNotesServer) error {
	s.mu.RLock()
	defer s.mu.RUnlock()

	for _, note := range s.notes {
		err := srv.Send(&notesv1.ListNotesResponse{Note: note})
		if err != nil {
			return err
		}
	}

	return nil
}

func (s *NotesService) AddNote(_ context.Context, req *notesv1.AddNoteRequest) (*notesv1.AddNoteResponse, error) {
	n := &notesv1.Note{
		Id:       uuid.Must(uuid.NewV4()).String(),
		Title:    req.Title,
		Archived: false,
	}

	s.notes = append(s.notes, n)

	return &notesv1.AddNoteResponse{
		Note: n,
	}, nil
}