-
Notifications
You must be signed in to change notification settings - Fork 1
/
get_template_by_name.go
78 lines (64 loc) · 2.06 KB
/
get_template_by_name.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
66
67
68
69
70
71
72
73
74
75
76
77
78
package queries
import (
"context"
"database/sql"
"errors"
"fmt"
"github.com/DIMO-Network/shared/db"
"github.com/DIMO-Network/vehicle-signal-decoding/internal/infrastructure/db/models"
"github.com/DIMO-Network/vehicle-signal-decoding/pkg/grpc"
"github.com/rs/zerolog"
"github.com/volatiletech/sqlboiler/v4/queries/qm"
"google.golang.org/protobuf/types/known/timestamppb"
)
type GetTemplateByNameQueryHandler struct {
DBS func() *db.ReaderWriter
logger *zerolog.Logger
}
func NewGetTemplateByNameQueryHandler(dbs func() *db.ReaderWriter, logger *zerolog.Logger) GetTemplateByNameQueryHandler {
return GetTemplateByNameQueryHandler{
DBS: dbs,
logger: logger,
}
}
type GetTemplateByNameQueryRequest struct {
Name string
}
func (h GetTemplateByNameQueryHandler) Handle(ctx context.Context, query *GetTemplateByNameQueryRequest) (*grpc.GetTemplateByNameResponse, error) {
item, err := models.Templates(
models.TemplateWhere.TemplateName.EQ(query.Name),
qm.Load(models.TemplateRels.TemplateNameDBCFile),
qm.Load(models.TemplateRels.TemplateNamePidConfigs),
).One(ctx, h.DBS().Reader)
if err != nil {
if errors.Is(err, sql.ErrNoRows) {
return nil, fmt.Errorf("template not found name: %s", query.Name)
}
return nil, fmt.Errorf("failed to get template")
}
pidsCount := 0
if item.R.TemplateNamePidConfigs != nil {
pidsCount = len(item.R.GetTemplateNamePidConfigs())
}
hasDbc := false
if item.R.TemplateNameDBCFile != nil {
hasDbc = true
}
result := &grpc.GetTemplateByNameResponse{
Template: &grpc.Template{
Name: item.TemplateName,
ParentTemplateName: item.ParentTemplateName.String,
Version: item.Version,
Protocol: item.Protocol,
Powertrain: item.Powertrain,
HasDbc: hasDbc,
PidsCount: int32(pidsCount),
CreatedAt: timestamppb.New(item.CreatedAt),
UpdatedAt: timestamppb.New(item.UpdatedAt),
},
}
if item.ParentTemplateName.Valid {
result.Template.ParentTemplateName = item.ParentTemplateName.String
}
return result, nil
}