Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
38 commits
Select commit Hold shift + click to select a range
7136ee9
remove priority multiplier code
d80tb7 Mar 3, 2025
fbbc3a5
Remove Market Based Scheduling
d80tb7 Mar 3, 2025
5762b91
Merge branch 'master' of github.com:armadaproject/armada into f/chris…
d80tb7 Mar 4, 2025
518972f
wip
d80tb7 Mar 4, 2025
c832f91
remove some stuff
d80tb7 Mar 4, 2025
45a7764
Merge branch 'master' of github.com:armadaproject/armada into f/chris…
d80tb7 Mar 4, 2025
19cc8cc
add back the price
d80tb7 Mar 4, 2025
00244ab
wip
d80tb7 Mar 7, 2025
f8c72a0
add security back
d80tb7 Mar 8, 2025
b1c5a6a
merge master
d80tb7 Mar 8, 2025
3f55c04
merge master
d80tb7 Mar 8, 2025
c451b31
merge master
d80tb7 Mar 8, 2025
ec836cc
merge master
d80tb7 Mar 8, 2025
231a174
Merge branch 'f/chrisma/remove-market-based-scheduling' of github.com…
d80tb7 Mar 8, 2025
daca0a3
merged from branch
d80tb7 Mar 8, 2025
50ac2c6
merge master
d80tb7 Mar 8, 2025
00017ff
merge master
d80tb7 Mar 8, 2025
e8b45e4
merge master
d80tb7 Mar 8, 2025
7c2ef94
merge master
d80tb7 Mar 8, 2025
58335d4
merge master
d80tb7 Mar 8, 2025
e3c3d35
lint
d80tb7 Mar 9, 2025
b14dfb3
go mod tidy
d80tb7 Mar 9, 2025
b5b3ef6
add more
d80tb7 Mar 9, 2025
fe5a3de
add more
d80tb7 Mar 9, 2025
e1c7c8d
lint
d80tb7 Mar 10, 2025
66eacbc
merge master
d80tb7 Mar 11, 2025
1a40c01
fix
d80tb7 Mar 11, 2025
e358b50
merge master
d80tb7 Mar 16, 2025
71d1f3d
fix proto package
d80tb7 Mar 16, 2025
a294afa
fix proto package part 2
d80tb7 Mar 16, 2025
f1b9b4f
lit
d80tb7 Mar 16, 2025
0f0ebc5
wip
d80tb7 Mar 17, 2025
ea2f768
wip
d80tb7 Mar 17, 2025
01524b1
lint
d80tb7 Mar 17, 2025
1f2b699
lint
d80tb7 Mar 17, 2025
673bfef
merge master
d80tb7 Mar 17, 2025
296d948
add protoc-gen-go-grpc to tools
d80tb7 Mar 17, 2025
f7df57d
remove out of date comment
d80tb7 Mar 17, 2025
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,6 @@
../../../proto/k8s.io/apimachinery/pkg/util/intstr/generated.proto;
../../../proto/k8s.io/api/networking/v1/generated.proto;
../../../proto/k8s.io/api/core/v1/generated.proto;
../../../proto/github.com/gogo/protobuf/gogoproto/gogo.proto;
../../../proto/google/api/*.proto
"
ProtoRoot="../../../proto/"
Expand Down
9 changes: 4 additions & 5 deletions go.mod
Original file line number Diff line number Diff line change
Expand Up @@ -17,9 +17,8 @@ require (
github.com/go-openapi/loads v0.22.0
github.com/go-openapi/runtime v0.28.0
github.com/go-openapi/spec v0.21.0
github.com/gogo/protobuf v1.3.2
github.com/golang/protobuf v1.5.4
github.com/google/go-cmp v0.7.0 // indirect
github.com/google/go-cmp v0.7.0
github.com/google/uuid v1.6.0
github.com/grpc-ecosystem/grpc-gateway v1.16.0
github.com/hashicorp/go-memdb v1.3.5
Expand Down Expand Up @@ -66,7 +65,6 @@ require (
github.com/go-openapi/swag v0.23.1
github.com/go-openapi/validate v0.24.0
github.com/go-playground/validator/v10 v10.25.0
github.com/gogo/status v1.1.1
github.com/goreleaser/goreleaser/v2 v2.8.1
github.com/grpc-ecosystem/go-grpc-middleware/providers/prometheus v1.0.1
github.com/grpc-ecosystem/go-grpc-middleware/v2 v2.3.1
Expand All @@ -85,6 +83,7 @@ require (
golang.org/x/term v0.30.0
golang.org/x/time v0.11.0
google.golang.org/genproto/googleapis/api v0.0.0-20250311190419-81fb87f6b8bf
google.golang.org/protobuf v1.36.5
gopkg.in/inf.v0 v0.9.1
gopkg.in/natefinch/lumberjack.v2 v2.2.1
gopkg.in/yaml.v3 v3.0.1
Expand Down Expand Up @@ -143,7 +142,7 @@ require (
github.com/go-playground/universal-translator v0.18.1 // indirect
github.com/gobwas/glob v0.2.3 // indirect
github.com/godbus/dbus v0.0.0-20190726142602-4481cbc300e2 // indirect
github.com/gogo/googleapis v0.0.0-20180223154316-0cd9801be74a // indirect
github.com/gogo/protobuf v1.3.2 // indirect
github.com/golang-jwt/jwt/v5 v5.2.1 // indirect
github.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da // indirect
github.com/golang/snappy v0.0.4 // indirect
Expand Down Expand Up @@ -194,6 +193,7 @@ require (
github.com/muesli/reflow v0.3.0 // indirect
github.com/muesli/termenv v0.16.0 // indirect
github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822 // indirect
github.com/onsi/ginkgo/v2 v2.19.0 // indirect
github.com/pelletier/go-toml/v2 v2.2.3 // indirect
github.com/peterbourgon/diskv v2.0.1+incompatible // indirect
github.com/pierrec/lz4 v2.0.5+incompatible // indirect
Expand Down Expand Up @@ -232,7 +232,6 @@ require (
golang.org/x/text v0.23.0 // indirect
golang.org/x/xerrors v0.0.0-20240716161551-93cc26a95ae9 // indirect
google.golang.org/genproto/googleapis/rpc v0.0.0-20250227231956-55c901821b1e // indirect
google.golang.org/protobuf v1.36.5 // indirect
gopkg.in/ini.v1 v1.67.0 // indirect
gopkg.in/warnings.v0 v0.1.2 // indirect
k8s.io/cli-runtime v0.26.15 // indirect
Expand Down
16 changes: 8 additions & 8 deletions go.sum
Original file line number Diff line number Diff line change
Expand Up @@ -248,17 +248,15 @@ github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91
github.com/go-playground/validator/v10 v10.25.0 h1:5Dh7cjvzR7BRZadnsVOzPhWsrwUr0nmsZJxEAnFLNO8=
github.com/go-playground/validator/v10 v10.25.0/go.mod h1:GGzBIJMuE98Ic/kJsBXbz1x/7cByt++cQ+YOuDM5wus=
github.com/go-sql-driver/mysql v1.5.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg=
github.com/go-task/slim-sprig/v3 v3.0.0 h1:sUs3vkvUymDpBKi3qH1YSqBQk9+9D/8M2mN1vB6EwHI=
github.com/go-task/slim-sprig/v3 v3.0.0/go.mod h1:W848ghGpv3Qj3dhTPRyJypKRiqCdHZiAzKg9hl15HA8=
github.com/gobwas/glob v0.2.3 h1:A4xDbljILXROh+kObIiy5kIaPYD8e96x1tgBhUI5J+Y=
github.com/gobwas/glob v0.2.3/go.mod h1:d3Ez4x06l9bZtSvzIay5+Yzi0fmZzPgnTbPcKjJAkT8=
github.com/godbus/dbus v0.0.0-20190726142602-4481cbc300e2 h1:ZpnhV/YsD2/4cESfV5+Hoeu/iUR3ruzNvZ+yQfO03a0=
github.com/godbus/dbus v0.0.0-20190726142602-4481cbc300e2/go.mod h1:bBOAhwG1umN6/6ZUMtDFBMQR8jRg9O75tm9K00oMsK4=
github.com/godbus/dbus/v5 v5.0.4/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA=
github.com/gogo/googleapis v0.0.0-20180223154316-0cd9801be74a h1:dR8+Q0uO5S2ZBcs2IH6VBKYwSxPo2vYCYq0ot0mu7xA=
github.com/gogo/googleapis v0.0.0-20180223154316-0cd9801be74a/go.mod h1:gf4bu3Q80BeJ6H1S1vYPm8/ELATdvryBaNFGgqEef3s=
github.com/gogo/protobuf v1.3.2 h1:Ov1cvc58UF3b5XjBnZv7+opcTcQFZebYjWzi34vdm4Q=
github.com/gogo/protobuf v1.3.2/go.mod h1:P1XiOD3dCwIKUDQYPy72D8LYyHL2YPYrpS2s69NZV8Q=
github.com/gogo/status v1.1.1 h1:DuHXlSFHNKqTQ+/ACf5Vs6r4X/dH2EgIzR9Vr+H65kg=
github.com/gogo/status v1.1.1/go.mod h1:jpG3dM5QPcqu19Hg8lkUhBFBa3TcLs1DG7+2Jqci7oU=
github.com/golang-jwt/jwt/v5 v5.2.1 h1:OuVbFODueb089Lh128TAcimifWaLhJwVflnrgM17wHk=
github.com/golang-jwt/jwt/v5 v5.2.1/go.mod h1:pqrtFR0X4osieyHYxtmOUWsAWrfe1Q5UVIyoH402zdk=
github.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q=
Expand Down Expand Up @@ -316,6 +314,8 @@ github.com/google/pprof v0.0.0-20181206194817-3ea8567a2e57/go.mod h1:zfwlbNMJ+OI
github.com/google/pprof v0.0.0-20190515194954-54271f7e092f/go.mod h1:zfwlbNMJ+OItoe0UupaVj+oy1omPYYDuagoSzA8v9mc=
github.com/google/pprof v0.0.0-20191218002539-d4f498aebedc/go.mod h1:ZgVRPoUq/hfqzAqh7sHMqb3I9Rq5C59dIz2SbBwJ4eM=
github.com/google/pprof v0.0.0-20200212024743-f11f1df84d12/go.mod h1:ZgVRPoUq/hfqzAqh7sHMqb3I9Rq5C59dIz2SbBwJ4eM=
github.com/google/pprof v0.0.0-20240424215950-a892ee059fd6 h1:k7nVchz72niMH6YLQNvHSdIE7iqsQxK1P41mySCvssg=
github.com/google/pprof v0.0.0-20240424215950-a892ee059fd6/go.mod h1:kf6iHlnVGwgKolg33glAes7Yg/8iWP8ukqeldJSO7jw=
github.com/google/renameio v0.1.0/go.mod h1:KWCgfxg9yswjAJkECMjeO8J8rahYeXnNhOm40UhjYkI=
github.com/google/rpmpack v0.6.1-0.20240329070804-c2247cbb881a h1:JJBdjSfqSy3mnDT0940ASQFghwcZ4y4cb6ttjAoXqwE=
github.com/google/rpmpack v0.6.1-0.20240329070804-c2247cbb881a/go.mod h1:uqVAUVQLq8UY2hCDfmJ/+rtO3aw7qyhc90rCVEabEfI=
Expand Down Expand Up @@ -505,8 +505,8 @@ github.com/oklog/ulid v1.3.1 h1:EGfNDEx6MqHz8B3uNV6QAib1UR2Lm97sHi3ocA6ESJ4=
github.com/oklog/ulid v1.3.1/go.mod h1:CirwcVhetQ6Lv90oh/F+FBtV6XMibvdAFo93nm5qn4U=
github.com/onsi/ginkgo v1.16.5 h1:8xi0RTUf59SOSfEtZMvwTvXYMzG4gV23XVHOZiXNtnE=
github.com/onsi/ginkgo v1.16.5/go.mod h1:+E8gABHa3K6zRBolWtd+ROzc/U5bkGt0FwiG042wbpU=
github.com/onsi/ginkgo/v2 v2.4.0 h1:+Ig9nvqgS5OBSACXNk15PLdp0U9XPYROt9CFzVdFGIs=
github.com/onsi/ginkgo/v2 v2.4.0/go.mod h1:iHkDK1fKGcBoEHT5W7YBq4RFWaQulw+caOMkAt4OrFo=
github.com/onsi/ginkgo/v2 v2.19.0 h1:9Cnnf7UHo57Hy3k6/m5k3dRfGTMXGvxhHFvkDTCTpvA=
github.com/onsi/ginkgo/v2 v2.19.0/go.mod h1:rlwLi9PilAFJ8jCg9UE1QP6VBpd6/xj3SRC0d6TU0To=
github.com/onsi/gomega v1.34.1 h1:EUMJIKUjM8sKjYbtxQI9A4z2o+rruxnzNvpknOXie6k=
github.com/onsi/gomega v1.34.1/go.mod h1:kU1QgUvBDLXBJq618Xvm2LUX6rSAfRaFRTcdOeDLwwY=
github.com/openconfig/goyang v1.6.2 h1:LVwwlVIIt4nmwacW67yBsxzP5DhDM94SOEMWod1hEA0=
Expand Down Expand Up @@ -835,6 +835,8 @@ golang.org/x/tools v0.0.0-20200212150539-ea181f53ac56/go.mod h1:TB2adYChydJhpapK
golang.org/x/tools v0.0.0-20200224181240-023911ca70b2/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=
golang.org/x/tools v0.0.0-20200619180055-7c47624df98f/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=
golang.org/x/tools v0.0.0-20210106214847-113979e3529a/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=
golang.org/x/tools v0.31.0 h1:0EedkvKDbh+qistFTd0Bcwe/YLh4vHwWEkiI0toFIBU=
golang.org/x/tools v0.31.0/go.mod h1:naFTU+Cev749tSJRXJlna0T3WxKvb1kWEx15xA4SdmQ=
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
Expand All @@ -855,7 +857,6 @@ google.golang.org/appengine v1.4.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7
google.golang.org/appengine v1.5.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4=
google.golang.org/appengine v1.6.1/go.mod h1:i06prIuMbXzDqacNJfV5OdTW448YApPu5ww/cMBSeb0=
google.golang.org/appengine v1.6.5/go.mod h1:8WjMMxjGQR8xUklV/ARdw2HLXBOI7O7uCIDZVag1xfc=
google.golang.org/genproto v0.0.0-20180518175338-11a468237815/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc=
google.golang.org/genproto v0.0.0-20180817151627-c66870c02cf8/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc=
google.golang.org/genproto v0.0.0-20190307195333-5fe7a883aa19/go.mod h1:VzzqZJRnGkLBvHegQrXjBqPurQTc5/KpmUdxsrq26oE=
google.golang.org/genproto v0.0.0-20190418145605-e7d98fc518a7/go.mod h1:VzzqZJRnGkLBvHegQrXjBqPurQTc5/KpmUdxsrq26oE=
Expand All @@ -882,7 +883,6 @@ google.golang.org/genproto/googleapis/api v0.0.0-20250311190419-81fb87f6b8bf h1:
google.golang.org/genproto/googleapis/api v0.0.0-20250311190419-81fb87f6b8bf/go.mod h1:jbe3Bkdp+Dh2IrslsFCklNhweNTBgSYanP1UXhJDhKg=
google.golang.org/genproto/googleapis/rpc v0.0.0-20250227231956-55c901821b1e h1:YA5lmSs3zc/5w+xsRcHqpETkaYyK63ivEPzNTcUUlSA=
google.golang.org/genproto/googleapis/rpc v0.0.0-20250227231956-55c901821b1e/go.mod h1:LuRYeWDFV6WOn90g357N17oMCaxpgCnbi/44qJvDn2I=
google.golang.org/grpc v1.12.0/go.mod h1:yo6s7OP7yaDglbqo1J04qKzAhqBH6lvTonzMVmEdcZw=
google.golang.org/grpc v1.19.0/go.mod h1:mqu4LbDTu4XGKhr4mRzUsmM4RtVoemTSY81AxZiDr8c=
google.golang.org/grpc v1.20.1/go.mod h1:10oTOabMzJvdu6/UiuZezV6QK5dSlG84ov/aaiqXj38=
google.golang.org/grpc v1.21.1/go.mod h1:oYelfM1adQP15Ek0mdvEgi9Df8B9CZIaU1084ijfRaM=
Expand Down
7 changes: 4 additions & 3 deletions internal/binoculars/server/binoculars.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ import (
"context"
"strconv"

"github.com/gogo/protobuf/types"
"google.golang.org/protobuf/types/known/emptypb"

"github.com/armadaproject/armada/internal/binoculars/service"
"github.com/armadaproject/armada/internal/common"
Expand All @@ -16,6 +16,7 @@ import (
type BinocularsServer struct {
logService service.LogService
cordonService service.CordonService
binoculars.UnimplementedBinocularsServer
}

func NewBinocularsServer(logService service.LogService, cordonService service.CordonService) *BinocularsServer {
Expand All @@ -42,11 +43,11 @@ func (b *BinocularsServer) Logs(ctx context.Context, request *binoculars.LogRequ
return &binoculars.LogResponse{Log: logLines}, nil
}

func (b *BinocularsServer) Cordon(ctx context.Context, request *binoculars.CordonRequest) (*types.Empty, error) {
func (b *BinocularsServer) Cordon(ctx context.Context, request *binoculars.CordonRequest) (*emptypb.Empty, error) {
err := b.cordonService.CordonNode(armadacontext.FromGrpcCtx(ctx), request)
if err != nil {
return nil, err
}

return &types.Empty{}, nil
return &emptypb.Empty{}, nil
}
2 changes: 1 addition & 1 deletion internal/common/eventutil/eventutil.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,8 @@ import (
"fmt"
"time"

"github.com/gogo/protobuf/proto"
"github.com/pkg/errors"
"google.golang.org/protobuf/proto"
v1 "k8s.io/api/core/v1"
networking "k8s.io/api/networking/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
Expand Down
12 changes: 6 additions & 6 deletions internal/common/eventutil/eventutil_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,9 +3,11 @@ package eventutil
import (
"testing"

"github.com/gogo/protobuf/proto"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"google.golang.org/protobuf/proto"

"github.com/armadaproject/armada/internal/common/testutil"
"github.com/armadaproject/armada/internal/common/util"
"github.com/armadaproject/armada/pkg/armadaevents"
)
Expand Down Expand Up @@ -368,10 +370,8 @@ func TestLimitSequenceByteSize(t *testing.T) {
}

actual, err := LimitSequenceByteSize(sequence, 1000, true)
if !assert.NoError(t, err) {
return
}
assert.Equal(t, []*armadaevents.EventSequence{sequence}, actual)
require.NoError(t, err)
testutil.AssertProtoEqual(t, []*armadaevents.EventSequence{sequence}, actual)

_, err = LimitSequenceByteSize(sequence, 1, true)
assert.Error(t, err)
Expand Down Expand Up @@ -401,7 +401,7 @@ func TestLimitSequenceByteSize(t *testing.T) {
if !assert.NoError(t, err) {
return
}
assert.Equal(t, expected, actual)
testutil.AssertProtoEqual(t, expected, actual)
}

func TestLimitSequencesByteSize(t *testing.T) {
Expand Down
6 changes: 3 additions & 3 deletions internal/common/ingest/ingestion_pipeline_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,8 +6,8 @@ import (
"time"

"github.com/apache/pulsar-client-go/pulsar"
"github.com/gogo/protobuf/proto"
"github.com/stretchr/testify/assert"
"google.golang.org/protobuf/proto"

"github.com/armadaproject/armada/internal/common/armadacontext"
commonconfig "github.com/armadaproject/armada/internal/common/config"
Expand Down Expand Up @@ -209,7 +209,7 @@ func (s *simpleEventSequenceConverter) Convert(_ *armadacontext.Context, msg *ut
for i, sequence := range msg.Events {
converted = append(converted, &simpleMessage{
id: msg.MessageIds[i],
size: sequence.Size(),
size: proto.Size(sequence),
})
}
return &simpleMessages{
Expand Down Expand Up @@ -237,7 +237,7 @@ func (s *simpleControlPlaneEventConverter) Convert(_ *armadacontext.Context, msg
for i, sequence := range msg.Events {
converted = append(converted, &simpleMessage{
id: msg.MessageIds[i],
size: sequence.Size(),
size: proto.Size(sequence),
})
}
return &simpleMessages{
Expand Down
2 changes: 1 addition & 1 deletion internal/common/ingest/testfixtures/event.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,8 @@ package testfixtures
import (
"time"

"github.com/gogo/protobuf/proto"
"github.com/google/uuid"
"google.golang.org/protobuf/proto"
v1 "k8s.io/api/core/v1"
"k8s.io/apimachinery/pkg/api/resource"

Expand Down
6 changes: 3 additions & 3 deletions internal/common/mocks/mock_executorapi.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

28 changes: 20 additions & 8 deletions internal/common/proto/protoutil.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,10 +3,14 @@ package protoutil
import (
"time"

"github.com/gogo/protobuf/proto"
"github.com/gogo/protobuf/types"
legacyproto "github.com/golang/protobuf/proto"
"google.golang.org/protobuf/proto"

"github.com/pkg/errors"

"google.golang.org/protobuf/types/known/durationpb"
"google.golang.org/protobuf/types/known/timestamppb"

"github.com/armadaproject/armada/internal/common/compress"
)

Expand Down Expand Up @@ -65,6 +69,14 @@ func MustMarshall(msg proto.Message) []byte {
return b
}

func MustMarshallLegacyProto(msg legacyproto.Message) []byte {
b, err := legacyproto.Marshal(msg)
if err != nil {
panic(errors.Wrap(err, "Error marshalling object"))
}
return b
}

// MustMarshallAndCompress first marshalls the supplied proto message and then compresses it.
// If either of these steps fail then it will panic. The main use case here is for unit tests.
// Think carefully if you intend to use this elsewhere
Expand All @@ -76,30 +88,30 @@ func MustMarshallAndCompress(msg proto.Message, compressor compress.Compressor)
return b
}

func ToStdTime(ts *types.Timestamp) time.Time {
func ToStdTime(ts *timestamppb.Timestamp) time.Time {
if ts == nil {
return time.Time{}.UTC()
}

return time.Unix(ts.Seconds, int64(ts.Nanos)).UTC()
}

func ToTimestamp(t time.Time) *types.Timestamp {
return &types.Timestamp{
func ToTimestamp(t time.Time) *timestamppb.Timestamp {
return &timestamppb.Timestamp{
Seconds: t.Unix(),
Nanos: int32(t.Nanosecond()),
}
}

func ToStdDuration(pd *types.Duration) time.Duration {
func ToStdDuration(pd *durationpb.Duration) time.Duration {
if pd == nil {
return 0
}
return time.Duration(pd.Seconds)*time.Second + time.Duration(pd.Nanos)*time.Nanosecond
}

func ToDuration(d time.Duration) *types.Duration {
return &types.Duration{
func ToDuration(d time.Duration) *durationpb.Duration {
return &durationpb.Duration{
Seconds: d.Nanoseconds() / 1000000000,
Nanos: int32(d.Nanoseconds() % 1000000000),
}
Expand Down
Loading
Loading