move gotd fork into repo. (#111)
- update to latest telegram layer - remove some references to fields in tg.Entities that don't exist in the schema - originally added here: https://github.com/beeper/td/commit/820929062a2ba0104397bc01235ab58a9cff780e - referenced here - https://github.com/mautrix/telegramgo/commit/124f0967ed195b5a380c9bd02e170ada9710dde3 - https://github.com/mautrix/telegramgo/commit/4205047aab2e0639217148b5d125bfaab668bd8e
This commit is contained in:
@@ -0,0 +1,57 @@
|
||||
package mtproto
|
||||
|
||||
import (
|
||||
"context"
|
||||
"crypto/rand"
|
||||
"io"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/stretchr/testify/require"
|
||||
"go.uber.org/zap"
|
||||
|
||||
"github.com/gotd/neo"
|
||||
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/crypto"
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/testutil"
|
||||
)
|
||||
|
||||
func benchWrite(payloadSize int) func(b *testing.B) {
|
||||
return func(b *testing.B) {
|
||||
a := require.New(b)
|
||||
logger := zap.NewNop()
|
||||
random := rand.Reader
|
||||
c := neo.NewTime(time.Now())
|
||||
|
||||
var key crypto.Key
|
||||
_, err := io.ReadFull(random, key[:])
|
||||
a.NoError(err)
|
||||
authKey := key.WithID()
|
||||
|
||||
payload := make([]byte, payloadSize)
|
||||
_, err = io.ReadFull(random, payload)
|
||||
a.NoError(err)
|
||||
data := &testPayload{Data: payload}
|
||||
|
||||
conn := Conn{
|
||||
conn: &constantConn{},
|
||||
clock: c,
|
||||
rand: random,
|
||||
cipher: crypto.NewClientCipher(random),
|
||||
log: logger,
|
||||
authKey: authKey,
|
||||
compressThreshold: -1,
|
||||
}
|
||||
b.ResetTimer()
|
||||
b.ReportAllocs()
|
||||
b.SetBytes(int64(payloadSize))
|
||||
|
||||
for i := 0; i < b.N; i++ {
|
||||
_ = conn.write(context.Background(), 1, 1, data)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkWrite(b *testing.B) {
|
||||
testutil.RunPayloads(b, benchWrite)
|
||||
}
|
||||
Reference in New Issue
Block a user