106 lines
2.2 KiB
Go
106 lines
2.2 KiB
Go
package shreder_client
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/samlior/libsam/pkg/logger"
|
|
"github.com/samlior/libsam/pkg/txparser"
|
|
"github.com/samlior/libsam/pkg/types"
|
|
"github.com/samlior/libsam/third_party/shreder_protos"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/credentials/insecure"
|
|
)
|
|
|
|
type ShrederClient struct {
|
|
log logger.Logger
|
|
|
|
conn *grpc.ClientConn
|
|
client shreder_protos.ShrederServiceClient
|
|
}
|
|
|
|
func NewShrederClient(logger logger.Logger, url string) (*ShrederClient, func(), error) {
|
|
conn, err := grpc.NewClient(url, grpc.WithTransportCredentials(insecure.NewCredentials()))
|
|
if err != nil {
|
|
return nil, func() {}, err
|
|
}
|
|
|
|
s := &ShrederClient{
|
|
log: logger,
|
|
conn: conn,
|
|
client: shreder_protos.NewShrederServiceClient(conn),
|
|
}
|
|
|
|
return s, func() {
|
|
s.Wait()
|
|
}, nil
|
|
}
|
|
|
|
func (c *ShrederClient) Wait() {
|
|
c.log.Debug("waiting for shreder client to stop")
|
|
|
|
err := c.conn.Close()
|
|
if err != nil {
|
|
c.log.Errorf("failed to close connection: %v", err)
|
|
}
|
|
|
|
c.log.Debug("shreder client stopped")
|
|
}
|
|
|
|
func (c *ShrederClient) ReadSync(ctx context.Context, txCh chan<- types.TxSignalBatch) error {
|
|
stream, err := c.client.SubscribeTransactions(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = stream.Send(&shreder_protos.SubscribeTransactionsRequest{
|
|
Transactions: map[string]*shreder_protos.SubscribeRequestFilterTransactions{
|
|
"pumpfun": {
|
|
AccountRequired: []string{
|
|
"6EF8rrecthR5Dkzon8Nwu78hRvfCKubJ14M5uBEwF6P",
|
|
},
|
|
},
|
|
"axiom": {
|
|
AccountRequired: []string{
|
|
"F5tfvbLog9VdGUPqBDTT8rgXvTTcq7e5UiGnupL1zvBq",
|
|
},
|
|
},
|
|
"photon": {
|
|
AccountRequired: []string{
|
|
"BSfD6SHZigAfDWSjzD5Q41jw8LmKwtmjskPH9XW1mrRW",
|
|
},
|
|
},
|
|
"fjsz": {
|
|
AccountRequired: []string{
|
|
"FJsZbftBqRLfF7uqUKpm4s2goDr6xsQ5Q3mN7AFJB6hK",
|
|
},
|
|
},
|
|
},
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for {
|
|
response, err := stream.Recv()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
txBatch := txparser.ParseTransaction(response.Transaction)
|
|
if len(txBatch) == 0 {
|
|
continue
|
|
}
|
|
|
|
// set fixed source for tx signals
|
|
for _, tx := range txBatch {
|
|
tx.Source = "shreder"
|
|
}
|
|
|
|
select {
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
case txCh <- txBatch:
|
|
}
|
|
}
|
|
}
|