services/sentiment/subscriber/subscriber.go

package subscriber

import (
	"context"
	"fmt"
	"time"

	"github.com/micro/micro/v3/service/logger"
	pb "github.com/micro/services/posts/proto"
	"github.com/micro/services/sentiment/model"
)

var (
	// assume this is initialised by main
	PostsClient pb.PostsService
)

// EnrichPost will enrich a post with the sentiment and save it
func EnrichPost(ctx context.Context, post *pb.Post) error {
	if PostsClient == nil {
		return nil
	}

	// start by analysing the title
	// later we will look at the content
	score := model.Analyze(post.Title)
	post.Metadata["sentiment"] = fmt.Sprintf("%.1f", score)

	logger.Infof("Setting score %.1f for post '%v'", score, post.Title)

	// now save the post
	_, err := PostsClient.Save(ctx, &pb.SaveRequest{
		Id:        post.Id,
		Title:     post.Title,
		Content:   post.Content,
		Timestamp: time.Now().Unix(),
		Metadata:  post.Metadata,
		Tags:      post.Tags,
		Image:     post.Image,
		Slug:      post.Slug,
	})
	if err != nil {
		logger.Info("Error saving post: %v", err)
	}

	return nil
}