/
get_dd_by_ids.go
65 lines (50 loc) · 1.88 KB
/
get_dd_by_ids.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
package queries
import (
"context"
"fmt"
"github.com/DIMO-Network/device-definitions-api/internal/core/common"
"github.com/DIMO-Network/device-definitions-api/internal/core/mediator"
"github.com/DIMO-Network/device-definitions-api/internal/core/services"
"github.com/DIMO-Network/device-definitions-api/internal/infrastructure/exceptions"
"github.com/DIMO-Network/device-definitions-api/pkg/grpc"
"github.com/pkg/errors"
"github.com/rs/zerolog"
)
type GetDeviceDefinitionByIDsQuery struct {
DeviceDefinitionID []string `json:"deviceDefinitionId" validate:"required"`
}
func (*GetDeviceDefinitionByIDsQuery) Key() string { return "GetDeviceDefinitionByIDsQuery" }
type GetDeviceDefinitionByIDsQueryHandler struct {
DDCache services.DeviceDefinitionCacheService
log *zerolog.Logger
}
func NewGetDeviceDefinitionByIDsQueryHandler(cache services.DeviceDefinitionCacheService, log *zerolog.Logger) GetDeviceDefinitionByIDsQueryHandler {
return GetDeviceDefinitionByIDsQueryHandler{
DDCache: cache,
log: log,
}
}
func (ch GetDeviceDefinitionByIDsQueryHandler) Handle(ctx context.Context, query mediator.Message) (interface{}, error) {
qry := query.(*GetDeviceDefinitionByIDsQuery)
if len(qry.DeviceDefinitionID) == 0 {
return nil, &exceptions.ValidationError{
Err: errors.New("Device Definition Ids is required"),
}
}
response := &grpc.GetDeviceDefinitionResponse{}
for _, v := range qry.DeviceDefinitionID {
dd, _ := ch.DDCache.GetDeviceDefinitionByID(ctx, v)
if dd == nil {
if len(qry.DeviceDefinitionID) > 1 {
ch.log.Warn().Str("deviceDefinitionId", v).Msg("Not found - Device Definition")
continue
}
return nil, &exceptions.NotFoundError{
Err: fmt.Errorf("could not find device definition id: %s", v),
}
}
rp := common.BuildFromQueryResultToGRPC(dd)
response.DeviceDefinitions = append(response.DeviceDefinitions, rp)
}
return response, nil
}