69 lines
2.3 KiB
Go
69 lines
2.3 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
"caatsm/internal/domain"
|
|
|
|
"github.com/hasura/go-graphql-client"
|
|
"golang.org/x/oauth2"
|
|
)
|
|
|
|
type HasuraRepository struct {
|
|
hasuraClient *graphql.Client
|
|
}
|
|
|
|
// NewHasuraClient creates a new HasuraClient
|
|
func NewHasuraRepo(endpoint, secret string) *HasuraRepository {
|
|
src := oauth2.StaticTokenSource(
|
|
&oauth2.Token{AccessToken: os.Getenv("GRAPHQL_TOKEN")},
|
|
)
|
|
httpClient := oauth2.NewClient(context.Background(), src)
|
|
return &HasuraRepository{
|
|
hasuraClient: graphql.NewClient(endpoint, httpClient),
|
|
}
|
|
}
|
|
|
|
// InsertParsedMessage inserts a new ParsedMessage into the Hasura GraphQL API
|
|
func (hr *HasuraRepository) InsertParsedMessage(pm *domain.ParsedMessage) error {
|
|
if hr.hasuraClient == nil {
|
|
return fmt.Errorf("hasuraClient is nil")
|
|
}
|
|
|
|
var mutation struct {
|
|
InsertParsedMessages struct {
|
|
Returning []domain.ParsedMessage `json:"returning"`
|
|
} `graphql:"insert_messages_one(object: {message_id: $messageId, date_time: $dateTime, priority_indicator: $priorityIndicator, primary_address: $primaryAddress, secondary_addresses: $secondaryAddresses, originator: $originator, originator_date_time: $originatorDateTime, category: $category, body_and_footer: $bodyAndFooter, body_data: $bodyData, received_at: $receivedAt, parsed_at: $parsedAt, need_dispatch: $needDispatch})"`
|
|
}
|
|
|
|
// Define the variables with their types
|
|
variables := map[string]interface{}{
|
|
"messageId": graphql.String(pm.MessageID),
|
|
"dateTime": graphql.String(pm.DateTime),
|
|
"priorityIndicator": graphql.String(pm.PriorityIndicator),
|
|
"primaryAddress": graphql.String(pm.PrimaryAddress),
|
|
"secondaryAddresses": graphql.String("second"),
|
|
"originator": graphql.String(pm.Originator),
|
|
"originatorDateTime": graphql.String(pm.OriginatorDateTime),
|
|
"category": graphql.String(pm.Category),
|
|
"bodyAndFooter": graphql.String(pm.BodyAndFooter),
|
|
"bodyData": graphql.String("{}"),
|
|
"receivedAt": pm.ReceivedAt.Format(time.RFC3339),
|
|
"parsedAt": pm.ParsedAt.Format(time.RFC3339),
|
|
// "dispatchedAt": pm.DispatchedAt.Format(time.RFC3339),
|
|
"needDispatch": graphql.Boolean(false),
|
|
}
|
|
|
|
ctx := context.Background()
|
|
// Execute the mutation with the variables
|
|
err := hr.hasuraClient.Mutate(ctx, &mutation, variables)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|