hashes.go

package main

import (
	"context"

	"github.com/aws/aws-sdk-go-v2/aws"
	"github.com/aws/aws-sdk-go-v2/service/dynamodb"
	v1 "github.com/dbProjectRED/redimo/v1"
	"github.com/golang/protobuf/ptypes/any"
)

func (r RedimoService) HGet(ctx context.Context, request *v1.HGetRequest) (*v1.HGetResponse, error) {
	resp, err := request.GetTable().Client().GetItemRequest(&dynamodb.GetItemInput{
		ConsistentRead: aws.Bool(true),
		Key: map[string]dynamodb.AttributeValue{
			"pk": {
				S: aws.String(request.GetKey()),
			},
			"sk": {
				S: aws.String(request.GetFieldName()),
			},
		},
		ReturnConsumedCapacity: dynamodb.ReturnConsumedCapacityTotal,
		TableName:              aws.String(request.GetTable().GetTable()),
	}).Send(ctx)
	if err != nil {
		return nil, err
	}

	return &v1.HGetResponse{
		Found: len(resp.Item) > 0,
		Value: &any.Any{
			TypeUrl: aws.StringValue(resp.Item["type"].S),
			Value:   resp.Item["data"].B,
		},
	}, nil
}

func (r RedimoService) HSet(ctx context.Context, request *v1.HSetRequest) (*v1.HSetResponse, error) {
	_, err := request.GetTable().Client().PutItemRequest(&dynamodb.PutItemInput{
		Item: map[string]dynamodb.AttributeValue{
			"pk": {
				S: aws.String(request.GetKey()),
			},
			"sk": {
				S: aws.String(request.GetFieldName()),
			},
			"type": {
				S: aws.String(request.GetValue().GetTypeUrl()),
			},
			"data": {
				B: request.GetValue().GetValue(),
			},
		},
		ReturnConsumedCapacity: dynamodb.ReturnConsumedCapacityTotal,
		ReturnValues:           dynamodb.ReturnValueNone,
		TableName:              aws.String(request.GetTable().GetTable()),
	}).Send(ctx)
	if err != nil {
		return nil, err
	}

	return &v1.HSetResponse{}, nil
}

func (r RedimoService) HDel(ctx context.Context, request *v1.HDelRequest) (*v1.HDelResponse, error) {
	_, err := request.GetTable().Client().DeleteItemRequest(&dynamodb.DeleteItemInput{
		Key: map[string]dynamodb.AttributeValue{
			"pk": {
				S: aws.String(request.GetKey()),
			},
			"sk": {
				S: aws.String(request.GetFieldName()),
			},
		},
		ReturnConsumedCapacity: dynamodb.ReturnConsumedCapacityTotal,
		ReturnValues:           dynamodb.ReturnValueNone,
		TableName:              aws.String(request.GetTable().GetTable()),
	}).Send(ctx)
	if err != nil {
		return nil, err
	}

	return &v1.HDelResponse{}, nil
}