| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package util |
| |
| import ( |
| "fmt" |
| "github.com/apache/servicecomb-service-center/server/plugin/pkg/discovery" |
| . "github.com/onsi/ginkgo" |
| . "github.com/onsi/gomega" |
| "sort" |
| "testing" |
| ) |
| |
| const VERSIONRULE_BASE = 5000 |
| |
| func BenchmarkVersionRule_Latest_GetServicesIds(b *testing.B) { |
| var kvs = make([]*discovery.KeyValue, VERSIONRULE_BASE) |
| for i := 1; i <= VERSIONRULE_BASE; i++ { |
| kvs[i-1] = &discovery.KeyValue{ |
| Key: []byte(fmt.Sprintf("/service/ver/1.%d", i)), |
| Value: []byte(fmt.Sprintf("%d", i)), |
| } |
| } |
| b.N = VERSIONRULE_BASE |
| b.ResetTimer() |
| for i := 0; i < b.N; i++ { |
| VersionRule(Latest).Match(kvs) |
| } |
| b.ReportAllocs() |
| // 5000 7105020 ns/op 2180198 B/op 39068 allocs/op |
| } |
| |
| func BenchmarkVersionRule_Range_GetServicesIds(b *testing.B) { |
| var kvs = make([]*discovery.KeyValue, VERSIONRULE_BASE) |
| for i := 1; i <= VERSIONRULE_BASE; i++ { |
| kvs[i-1] = &discovery.KeyValue{ |
| Key: []byte(fmt.Sprintf("/service/ver/1.%d", i)), |
| Value: []byte(fmt.Sprintf("%d", i)), |
| } |
| } |
| b.N = VERSIONRULE_BASE |
| b.ResetTimer() |
| for i := 0; i < b.N; i++ { |
| VersionRule(Range).Match(kvs, fmt.Sprintf("1.%d", i), fmt.Sprintf("1.%d", i+VERSIONRULE_BASE/10)) |
| } |
| b.ReportAllocs() |
| // 5000 7244029 ns/op 2287389 B/op 39584 allocs/op |
| } |
| |
| func BenchmarkVersionRule_AtLess_GetServicesIds(b *testing.B) { |
| var kvs = make([]*discovery.KeyValue, VERSIONRULE_BASE) |
| for i := 1; i <= VERSIONRULE_BASE; i++ { |
| kvs[i-1] = &discovery.KeyValue{ |
| Key: []byte(fmt.Sprintf("/service/ver/1.%d", i)), |
| Value: []byte(fmt.Sprintf("%d", i)), |
| } |
| } |
| b.N = VERSIONRULE_BASE |
| b.ResetTimer() |
| for i := 0; i < b.N; i++ { |
| VersionRule(AtLess).Match(kvs, fmt.Sprintf("1.%d", i)) |
| } |
| b.ReportAllocs() |
| // 5000 11221098 ns/op 3174720 B/op 58064 allocs/op |
| } |
| |
| func BenchmarkParseVersionRule(b *testing.B) { |
| f := ParseVersionRule("latest") |
| kvs := []*discovery.KeyValue{ |
| { |
| Key: []byte("/service/ver/1.0.300"), |
| Value: "1.0.300", |
| }, |
| { |
| Key: []byte("/service/ver/1.0.303"), |
| Value: "1.0.303", |
| }, |
| { |
| Key: []byte("/service/ver/1.0.304"), |
| Value: "1.0.304", |
| }, |
| } |
| b.ResetTimer() |
| b.RunParallel(func(pb *testing.PB) { |
| for pb.Next() { |
| f(kvs) |
| } |
| }) |
| b.ReportAllocs() |
| } |
| |
| var _ = Describe("Version Rule sorter", func() { |
| Describe("Sorter", func() { |
| Context("Normal", func() { |
| It("version asc", func() { |
| kvs := []string{"1.0.0", "1.0.1"} |
| sort.Sort(&serviceKeySorter{ |
| sortArr: kvs, |
| kvs: make(map[string]*discovery.KeyValue), |
| cmp: Larger, |
| }) |
| Expect(kvs[0]).To(Equal("1.0.1")) |
| Expect(kvs[1]).To(Equal("1.0.0")) |
| }) |
| It("version desc", func() { |
| kvs := []string{"1.0.1", "1.0.0"} |
| sort.Sort(&serviceKeySorter{ |
| sortArr: kvs, |
| kvs: make(map[string]*discovery.KeyValue), |
| cmp: Larger, |
| }) |
| Expect(kvs[0]).To(Equal("1.0.1")) |
| Expect(kvs[1]).To(Equal("1.0.0")) |
| }) |
| It("len(v1) != len(v2)", func() { |
| kvs := []string{"1.0.0.0", "1.0.1"} |
| sort.Sort(&serviceKeySorter{ |
| sortArr: kvs, |
| kvs: make(map[string]*discovery.KeyValue), |
| cmp: Larger, |
| }) |
| Expect(kvs[0]).To(Equal("1.0.1")) |
| Expect(kvs[1]).To(Equal("1.0.0.0")) |
| }) |
| It("1.0.9 vs 1.0.10", func() { |
| kvs := []string{"1.0.9", "1.0.10"} |
| sort.Sort(&serviceKeySorter{ |
| sortArr: kvs, |
| kvs: make(map[string]*discovery.KeyValue), |
| cmp: Larger, |
| }) |
| Expect(kvs[0]).To(Equal("1.0.10")) |
| Expect(kvs[1]).To(Equal("1.0.9")) |
| }) |
| It("1.10 vs 4", func() { |
| kvs := []string{"1.10", "4"} |
| sort.Sort(&serviceKeySorter{ |
| sortArr: kvs, |
| kvs: make(map[string]*discovery.KeyValue), |
| cmp: Larger, |
| }) |
| Expect(kvs[0]).To(Equal("4")) |
| Expect(kvs[1]).To(Equal("1.10")) |
| }) |
| }) |
| Context("Exception", func() { |
| It("invalid version1", func() { |
| kvs := []string{"1.a", "1.0.1.a", ""} |
| sort.Sort(&serviceKeySorter{ |
| sortArr: kvs, |
| kvs: make(map[string]*discovery.KeyValue), |
| cmp: Larger, |
| }) |
| Expect(kvs[0]).To(Equal("1.a")) |
| Expect(kvs[1]).To(Equal("1.0.1.a")) |
| Expect(kvs[2]).To(Equal("")) |
| }) |
| It("invalid version2 > 32767", func() { |
| kvs := []string{"1.0", "1.0.1.32768"} |
| sort.Sort(&serviceKeySorter{ |
| sortArr: kvs, |
| kvs: make(map[string]*discovery.KeyValue), |
| cmp: Larger, |
| }) |
| Expect(kvs[0]).To(Equal("1.0")) |
| Expect(kvs[1]).To(Equal("1.0.1.32768")) |
| kvs = []string{"1.0", "1.0.1.32767"} |
| sort.Sort(&serviceKeySorter{ |
| sortArr: kvs, |
| kvs: make(map[string]*discovery.KeyValue), |
| cmp: Larger, |
| }) |
| Expect(kvs[0]).To(Equal("1.0.1.32767")) |
| Expect(kvs[1]).To(Equal("1.0")) |
| }) |
| }) |
| }) |
| Describe("VersionRule", func() { |
| const count = 10 |
| var kvs = [count]*discovery.KeyValue{} |
| BeforeEach(func() { |
| for i := 1; i <= count; i++ { |
| kvs[i-1] = &discovery.KeyValue{ |
| Key: []byte(fmt.Sprintf("/service/ver/1.%d", i)), |
| Value: fmt.Sprintf("%d", i), |
| } |
| } |
| }) |
| Context("Normal", func() { |
| It("Latest", func() { |
| results := VersionRule(Latest).Match(kvs[:]) |
| Expect(len(results)).To(Equal(1)) |
| Expect(results[0]).To(Equal(fmt.Sprintf("%d", count))) |
| }) |
| It("Range1.1 ver in [1.4, 1.8)", func() { |
| results := VersionRule(Range).Match(kvs[:], "1.4", "1.8") |
| Expect(len(results)).To(Equal(4)) |
| Expect(results[0]).To(Equal("7")) |
| Expect(results[3]).To(Equal("4")) |
| }) |
| It("Range1.2 ver in (1.8, 1.4]", func() { |
| results := VersionRule(Range).Match(kvs[:], "1.8", "1.4") |
| Expect(len(results)).To(Equal(4)) |
| Expect(results[0]).To(Equal("7")) |
| Expect(results[3]).To(Equal("4")) |
| }) |
| It("Range2 ver in [1, 2]", func() { |
| results := VersionRule(Range).Match(kvs[:], "1", "2") |
| Expect(len(results)).To(Equal(10)) |
| Expect(results[0]).To(Equal("10")) |
| Expect(results[9]).To(Equal("1")) |
| }) |
| It("Range3 ver in [1.4.1, 1.9.1]", func() { |
| results := VersionRule(Range).Match(kvs[:], "1.4.1", "1.9.1") |
| Expect(len(results)).To(Equal(5)) |
| Expect(results[0]).To(Equal("9")) |
| Expect(results[4]).To(Equal("5")) |
| }) |
| It("Range4 ver in [2, 4]", func() { |
| results := VersionRule(Range).Match(kvs[:], "2", "4") |
| Expect(len(results)).To(Equal(0)) |
| }) |
| It("AtLess1 ver >= 1.6", func() { |
| results := VersionRule(AtLess).Match(kvs[:], "1.6") |
| Expect(len(results)).To(Equal(5)) |
| Expect(results[0]).To(Equal("10")) |
| Expect(results[4]).To(Equal("6")) |
| }) |
| It("AtLess2 ver >= 1", func() { |
| results := VersionRule(AtLess).Match(kvs[:], "1") |
| Expect(len(results)).To(Equal(10)) |
| Expect(results[0]).To(Equal("10")) |
| Expect(results[9]).To(Equal("1")) |
| }) |
| It("AtLess3 ver >= 1.5.1", func() { |
| results := VersionRule(AtLess).Match(kvs[:], "1.5.1") |
| Expect(len(results)).To(Equal(5)) |
| Expect(results[0]).To(Equal("10")) |
| Expect(results[4]).To(Equal("6")) |
| }) |
| It("AtLess4 ver >= 2", func() { |
| results := VersionRule(AtLess).Match(kvs[:], "2") |
| Expect(len(results)).To(Equal(0)) |
| }) |
| }) |
| Context("Exception", func() { |
| It("nil", func() { |
| results := VersionRule(Latest).Match(nil) |
| Expect(len(results)).To(Equal(0)) |
| results = VersionRule(AtLess).Match(nil) |
| Expect(len(results)).To(Equal(0)) |
| results = VersionRule(Range).Match(nil) |
| Expect(len(results)).To(Equal(0)) |
| Expect(ParseVersionRule("")).To(BeNil()) |
| Expect(ParseVersionRule("abc")).To(BeNil()) |
| Expect(VersionMatchRule("1.0", "1.0")).To(BeTrue()) |
| Expect(VersionMatchRule("1.0", "1.2")).To(BeFalse()) |
| }) |
| }) |
| Context("Parse", func() { |
| It("Latest", func() { |
| match := ParseVersionRule("latest") |
| results := match(kvs[:]) |
| Expect(len(results)).To(Equal(1)) |
| Expect(results[0]).To(Equal(fmt.Sprintf("%d", count))) |
| }) |
| It("Range ver in [1.4, 1.8)", func() { |
| match := ParseVersionRule("1.4-1.8") |
| results := match(kvs[:]) |
| Expect(len(results)).To(Equal(4)) |
| Expect(results[0]).To(Equal("7")) |
| Expect(results[3]).To(Equal("4")) |
| }) |
| It("AtLess ver >= 1.6", func() { |
| match := ParseVersionRule("1.6+") |
| results := match(kvs[:]) |
| Expect(len(results)).To(Equal(5)) |
| Expect(results[0]).To(Equal("10")) |
| Expect(results[4]).To(Equal("6")) |
| }) |
| }) |
| Context("VersionMatchRule", func() { |
| It("Latest", func() { |
| Expect(VersionMatchRule("1.0", "latest")).To(BeTrue()) |
| }) |
| It("Range ver in [1.4, 1.8]", func() { |
| Expect(VersionMatchRule("1.4", "1.4-1.8")).To(BeTrue()) |
| Expect(VersionMatchRule("1.6", "1.4-1.8")).To(BeTrue()) |
| Expect(VersionMatchRule("1.8", "1.4-1.8")).To(BeFalse()) |
| Expect(VersionMatchRule("1.0", "1.4-1.8")).To(BeFalse()) |
| Expect(VersionMatchRule("1.9", "1.4-1.8")).To(BeFalse()) |
| }) |
| It("AtLess ver >= 1.6", func() { |
| Expect(VersionMatchRule("1.6", "1.6+")).To(BeTrue()) |
| Expect(VersionMatchRule("1.9", "1.6+")).To(BeTrue()) |
| Expect(VersionMatchRule("1.0", "1.6+")).To(BeFalse()) |
| }) |
| }) |
| }) |
| Describe("NewVersionRegexp", func() { |
| Context("Normal", func() { |
| It("Latest", func() { |
| vr := NewVersionRegexp(false) |
| Expect(vr.MatchString("latest")).To(BeFalse()) |
| vr = NewVersionRegexp(true) |
| Expect(vr.MatchString("latest")).To(BeTrue()) |
| }) |
| It("Range", func() { |
| vr := NewVersionRegexp(false) |
| Expect(vr.MatchString("1.1-2.2")).To(BeFalse()) |
| vr = NewVersionRegexp(true) |
| Expect(vr.MatchString("-")).To(BeFalse()) |
| Expect(vr.MatchString("1.1-")).To(BeFalse()) |
| Expect(vr.MatchString("-1.1")).To(BeFalse()) |
| Expect(vr.MatchString("1.a-2.b")).To(BeFalse()) |
| Expect(vr.MatchString("1.-.2")).To(BeFalse()) |
| Expect(vr.MatchString("60000-1")).To(BeFalse()) |
| Expect(vr.MatchString("1.1-2.2")).To(BeTrue()) |
| Expect(vr.MatchString("1.1.1.1-2.2.2.2")).To(BeTrue()) |
| Expect(vr.MatchString("1.1.1.1.1-2.2.2.2")).To(BeFalse()) |
| Expect(vr.MatchString("1.1.1.1-2.2.2.2.2")).To(BeFalse()) |
| }) |
| It("AtLess", func() { |
| vr := NewVersionRegexp(false) |
| Expect(vr.MatchString("1.0+")).To(BeFalse()) |
| vr = NewVersionRegexp(true) |
| Expect(vr.MatchString("+")).To(BeFalse()) |
| Expect(vr.MatchString("+1.0")).To(BeFalse()) |
| Expect(vr.MatchString("1.a+")).To(BeFalse()) |
| Expect(vr.MatchString(".1+")).To(BeFalse()) |
| Expect(vr.MatchString("1.+")).To(BeFalse()) |
| Expect(vr.MatchString(".+")).To(BeFalse()) |
| Expect(vr.MatchString("60000+")).To(BeFalse()) |
| Expect(vr.MatchString("1.0+")).To(BeTrue()) |
| Expect(vr.MatchString("1.0.0.0+")).To(BeTrue()) |
| Expect(vr.MatchString("1.0.0.0.0+")).To(BeFalse()) |
| }) |
| It("Explicit", func() { |
| vr := NewVersionRegexp(false) |
| Expect(vr.MatchString("")).To(BeFalse()) |
| Expect(vr.MatchString("a")).To(BeFalse()) |
| Expect(vr.MatchString("60000")).To(BeFalse()) |
| Expect(vr.MatchString(".")).To(BeFalse()) |
| Expect(vr.MatchString("1.")).To(BeFalse()) |
| Expect(vr.MatchString(".1")).To(BeFalse()) |
| Expect(vr.MatchString("1.4")).To(BeTrue()) |
| vr = NewVersionRegexp(true) |
| Expect(vr.MatchString("")).To(BeFalse()) |
| Expect(vr.MatchString("a")).To(BeFalse()) |
| Expect(vr.MatchString("60000")).To(BeFalse()) |
| Expect(vr.MatchString(".")).To(BeFalse()) |
| Expect(vr.MatchString("1.")).To(BeFalse()) |
| Expect(vr.MatchString(".1")).To(BeFalse()) |
| Expect(vr.MatchString("1.4")).To(BeTrue()) |
| Expect(vr.MatchString("1.4.0.0")).To(BeTrue()) |
| Expect(vr.MatchString("1.4.0.0.0")).To(BeFalse()) |
| }) |
| }) |
| Context("Exception", func() { |
| It("MatchString & String", func() { |
| vr := VersionRegexp{} |
| Expect(vr.MatchString("")).To(BeTrue()) |
| Expect(vr.String()).NotTo(BeEmpty()) |
| vr = VersionRegexp{Fuzzy: true} |
| Expect(vr.MatchString("")).To(BeTrue()) |
| Expect(vr.String()).NotTo(BeEmpty()) |
| }) |
| }) |
| }) |
| }) |
