@@ -11,11 +11,12 @@ import (
1111 "strings"
1212 "time"
1313
14- "github.com/golang/protobuf/jsonpb"
15- "github.com/golang/protobuf/ptypes"
14+ "github.com/golang/protobuf/jsonpb" //nolint: staticcheck
1615 structpb "github.com/golang/protobuf/ptypes/struct"
1716 "github.com/pkg/errors"
1817 "github.com/shamaton/msgpack/v2"
18+ "google.golang.org/protobuf/types/known/durationpb"
19+ "google.golang.org/protobuf/types/known/timestamppb"
1920
2021 "github.com/flyteorg/flyte/v2/flytestdlib/storage"
2122 "github.com/flyteorg/flyte/v2/gen/go/flyteidl2/core"
@@ -45,17 +46,14 @@ func MakePrimitive(v interface{}) (*core.Primitive, error) {
4546 },
4647 }, nil
4748 case time.Time :
48- t , err := ptypes .TimestampProto (p )
49- if err != nil {
50- return nil , err
51- }
49+ t := timestamppb .New (p )
5250 return & core.Primitive {
5351 Value : & core.Primitive_Datetime {
5452 Datetime : t ,
5553 },
5654 }, nil
5755 case time.Duration :
58- d := ptypes . DurationProto (p )
56+ d := durationpb . New (p )
5957 return & core.Primitive {
6058 Value : & core.Primitive_Duration {
6159 Duration : d ,
@@ -360,16 +358,13 @@ func MakePrimitiveForType(t core.SimpleType, s string) (*core.Primitive, error)
360358 if err != nil {
361359 return nil , errors .Wrap (err , "failed to parse Duration, valid formats: e.g. 300ms, -1.5h, 2h45m" )
362360 }
363- p .Value = & core.Primitive_Duration {Duration : ptypes . DurationProto (v )}
361+ p .Value = & core.Primitive_Duration {Duration : durationpb . New (v )}
364362 case core .SimpleType_DATETIME :
365363 v , err := time .Parse (time .RFC3339 , s )
366364 if err != nil {
367365 return nil , errors .Wrap (err , "failed to parse Datetime in RFC3339 format" )
368366 }
369- ts , err := ptypes .TimestampProto (v )
370- if err != nil {
371- return nil , errors .Wrap (err , "failed to convert datetime to proto" )
372- }
367+ ts := timestamppb .New (v )
373368 p .Value = & core.Primitive_Datetime {Datetime : ts }
374369 default :
375370 return nil , fmt .Errorf ("unsupported type %s" , t .String ())
0 commit comments