github.com/matrixorigin/matrixone@v1.2.0/pkg/vm/engine/tae/logstore/driver/entry/mock.go (about) 1 // Copyright 2021 Matrix Origin 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); 4 // you may not use this file except in compliance with the License. 5 // You may obtain a copy of the License at 6 // 7 // http://www.apache.org/licenses/LICENSE-2.0 8 // 9 // Unless required by applicable law or agreed to in writing, software 10 // distributed under the License is distributed on an "AS IS" BASIS, 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 // See the License for the specific language governing permissions and 13 // limitations under the License. 14 15 package entry 16 17 import ( 18 "bytes" 19 "math/rand" 20 21 "github.com/matrixorigin/matrixone/pkg/vm/engine/tae/logstore/entry" 22 ) 23 24 var buf []byte 25 26 func init() { 27 var bs bytes.Buffer 28 for i := 0; i < 3000; i++ { 29 bs.WriteString("helloyou") 30 } 31 buf = bs.Bytes() 32 } 33 34 func MockEntry() *Entry { 35 payloadSize := 100 36 37 e := entry.GetBase() 38 e.SetType(entry.IOET_WALEntry_Test) 39 info := &entry.Info{GroupLSN: uint64(rand.Intn(1000))} 40 e.SetInfo(info) 41 payload := make([]byte, payloadSize) 42 copy(payload, buf) 43 err := e.SetPayload(payload) 44 if err != nil { 45 panic(err) 46 } 47 e.PrepareWrite() 48 return NewEntry(e) 49 } 50 51 func MockEntryWithPayload(payload []byte) *Entry { 52 e := entry.GetBase() 53 e.SetType(entry.IOET_WALEntry_Test) 54 info := &entry.Info{GroupLSN: uint64(rand.Intn(1000))} 55 e.SetInfo(info) 56 err := e.SetPayload(payload) 57 if err != nil { 58 panic(err) 59 } 60 e.PrepareWrite() 61 return NewEntry(e) 62 63 }