Find the nearest rent company

internal/db/db.go

@@ -20,11 +20,12 @@ func InitDb(dsn string) (*gorm.DB, error) {
 
  if err == nil {
  db.AutoMigrate(
- &models.User{},
- &models.Interest{},
  &models.AvailableFlight{},
- &models.Offer{},
+ &models.Interest{},
  &models.Journey{},
+ &models.Offer{},
+ &models.Rent{},
+ &models.User{},
  )
  }
 

internal/handlers/acmesky/tm_compute_distance_user_airport.go

@@ -84,7 +84,7 @@ func TMComputeDistanceUserAirport(client worker.JobClient, job entities.Job) {
  return
  }
  variables["distance"] = distance.GetDistance() / 1000
- log.Info("[%s] [%d] Found a distance of: %d km", job.Type, jobKey, variables["distance"])
+ log.Infof("[%s] [%d] Found a distance of: %d km", job.Type, jobKey, variables["distance"])
 
  request, err := client.NewCompleteJobCommand().JobKey(jobKey).VariablesFromMap(variables)
  if err != nil {

internal/handlers/acmesky/tm_find_nearest_available_rent_company.go

@@ -2,13 +2,22 @@ package handlers
 
 import (
  "context"
+ "time"
+
  "github.com/charmbracelet/log"
 
+ pb "github.com/acme-sky/geodistance-api/pkg/distance/proto"
+ "github.com/acme-sky/workers/internal/config"
+ "github.com/acme-sky/workers/internal/db"
  acmejob "github.com/acme-sky/workers/internal/job"
+ "github.com/acme-sky/workers/internal/models"
  "github.com/camunda/zeebe/clients/go/v8/pkg/entities"
  "github.com/camunda/zeebe/clients/go/v8/pkg/worker"
+ "google.golang.org/grpc"
+ "google.golang.org/grpc/credentials/insecure"
 )
 
+// Task used to find the nearest rent company to the user' address
 func TMFindNearestAvailableRentCompany(client worker.JobClient, job entities.Job) {
  jobKey := job.GetKey()
 
@@ -18,15 +27,88 @@ func TMFindNearestAvailableRentCompany(client worker.JobClient, job entities.Job
  return
  }
 
- request, err := client.NewCompleteJobCommand().JobKey(jobKey).VariablesFromMap(variables)
+ db, _ := db.GetDb()
+ var offer models.Offer
+
+ if err := db.Where("id = ?", int(variables["offer_id"].(float64))).Preload("User").Preload("Journey").Preload("Journey.Flight1").First(&offer).Error; err != nil {
+ log.Errorf("[%s] [%d] Error on getting offer %s", job.Type, jobKey, err.Error())
+ }
+
+ if offer.User.Address == nil {
+ log.Warnf("[%s] [%d] User does not have an address", job.Type, jobKey)
+ acmejob.FailJob(client, job)
+ return
+ }
+
+ conf, _ := config.GetConfig()
+
+ conn, err := grpc.Dial(conf.String("geodistance.api"), grpc.WithTransportCredentials(insecure.NewCredentials()))
  if err != nil {
+ log.Errorf("[%s] [%d] Can't connect to Geodistance url: %s", job.Type, jobKey, err.Error())
  acmejob.FailJob(client, job)
  return
  }
 
- log.Debug("Processing data:", variables)
+ defer conn.Close()
+ c := pb.NewDistanceClient(conn)
+
+ ctx, cancel := context.WithTimeout(context.Background(), time.Second)
+ defer cancel()
+
+ userGeometry, err := c.FindGeometry(ctx, &pb.AddressRequest{
+ Address: *offer.User.Address,
+ })
+ if err != nil {
+ log.Errorf("[%s] [%d] Can't find geometry for user: %s", job.Type, jobKey, err.Error())
+ acmejob.FailJob(client, job)
+ return
+ }
+
+ var rents []models.Rent
+ if err := db.Find(&rents).Error; err != nil {
+ log.Errorf("[%s] [%d] Rents not found %s", job.Type, jobKey, err.Error())
+ acmejob.FailJob(client, job)
+ return
+ }
+ var distances []int
+
+ for _, rent := range rents {
+ distance, err := c.FindDistance(ctx, &pb.DistanceRequest{
+ Origin: &pb.MapPosition{Latitude: userGeometry.Latitude, Longitude: userGeometry.Longitude},
+ Destination: &pb.MapPosition{Latitude: rent.Latitude, Longitude: rent.Longitude},
+ })
+ if err != nil {
+ log.Warn("[%s] [%d] Can't find distance for %s: %s", job.Type, jobKey, rent.Name, err.Error())
+ distances = append(distances, 9999999)
+ continue
+ }
+ distances = append(distances, int(distance.GetDistance()))
+ }
+
+ if len(distances) == 0 {
+ log.Errorf("[%s] [%d] There is no available rent company: %s", job.Type, jobKey, err.Error())
+ acmejob.FailJob(client, job)
+ return
+ }
+
+ var selectRentIndex = 0
+ var minRentDistance = distances[0]
+ for i := 1; i < len(distances); {
+ if distances[i] < minRentDistance {
+ minRentDistance = distances[i]
+ selectRentIndex = i
+ }
+ }
+
+ variables["rent_company"] = rents[selectRentIndex]
+
+ request, err := client.NewCompleteJobCommand().JobKey(jobKey).VariablesFromMap(variables)
+ if err != nil {
+ acmejob.FailJob(client, job)
+ return
+ }
 
- ctx := context.Background()
+ ctx = context.Background()
  _, err = request.Send(ctx)
  if err != nil {
  acmejob.FailJob(client, job)

internal/models/rent.go

@@ -0,0 +1,15 @@
+package models
+
+import (
+ "time"
+)
+
+// Rent model
+type Rent struct {
+ Id uint `gorm:"column:id" json:"id"`
+ CreatedAt time.Time `gorm:"column:created_at" json:"created_at"`
+ Name string `gorm:"name" json:"name"`
+ Latitude float32 `gorm:"latitude" json:"latitude"`
+ Longitude float32 `gorm:"longitude" json:"longitude"`
+ Endpoint string `gorm:"endpoint" json:"endpoint"`
+}