2018-10-17 09:35:24 -07:00
|
|
|
// @flow
|
|
|
|
|
|
|
|
import * as BufferLayout from 'buffer-layout';
|
|
|
|
|
2018-10-26 13:19:47 -07:00
|
|
|
import {Account} from './account';
|
|
|
|
import {PublicKey} from './publickey';
|
|
|
|
import {SystemProgram} from './system-program';
|
|
|
|
import {Transaction} from './transaction';
|
2018-10-17 09:35:24 -07:00
|
|
|
import {sendAndConfirmTransaction} from './util/send-and-confirm-transaction';
|
2018-10-26 13:19:47 -07:00
|
|
|
import type {Connection} from './connection';
|
2018-10-17 09:35:24 -07:00
|
|
|
|
|
|
|
/**
|
|
|
|
* Program loader interface
|
|
|
|
*/
|
|
|
|
export class Loader {
|
|
|
|
/**
|
|
|
|
* @private
|
|
|
|
*/
|
|
|
|
connection: Connection;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @private
|
|
|
|
*/
|
|
|
|
programId: PublicKey;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @param connection The connection to use
|
|
|
|
* @param programId Public key that identifies the loader
|
|
|
|
*/
|
|
|
|
constructor(connection: Connection, programId: PublicKey) {
|
|
|
|
Object.assign(this, {connection, programId});
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Load program data
|
|
|
|
*
|
|
|
|
* @param program Account to load the program info
|
2018-10-23 20:56:54 -07:00
|
|
|
* @param data Program data
|
2018-10-17 09:35:24 -07:00
|
|
|
*/
|
2018-10-23 20:56:54 -07:00
|
|
|
async load(program: Account, data: Array<number>) {
|
2018-10-17 09:35:24 -07:00
|
|
|
const userdataLayout = BufferLayout.struct([
|
|
|
|
BufferLayout.u32('instruction'),
|
|
|
|
BufferLayout.u32('offset'),
|
|
|
|
BufferLayout.u32('bytesLength'),
|
|
|
|
BufferLayout.u32('bytesLengthPadding'),
|
|
|
|
BufferLayout.seq(
|
|
|
|
BufferLayout.u8('byte'),
|
2018-10-23 20:56:54 -07:00
|
|
|
BufferLayout.offset(BufferLayout.u32(), -8), 'bytes'),
|
2018-10-17 09:35:24 -07:00
|
|
|
]);
|
|
|
|
|
2018-10-23 20:56:54 -07:00
|
|
|
const chunkSize = 256;
|
|
|
|
let offset = 0;
|
|
|
|
let array = data;
|
2018-11-01 20:35:19 -07:00
|
|
|
let transactions = [];
|
2018-10-23 20:56:54 -07:00
|
|
|
while (array.length > 0) {
|
|
|
|
const bytes = array.slice(0, chunkSize);
|
2018-10-25 10:41:18 -07:00
|
|
|
const userdata = Buffer.alloc(chunkSize + 16);
|
2018-10-23 20:56:54 -07:00
|
|
|
userdataLayout.encode(
|
|
|
|
{
|
|
|
|
instruction: 0, // Load instruction
|
|
|
|
offset,
|
|
|
|
bytes,
|
|
|
|
},
|
|
|
|
userdata,
|
|
|
|
);
|
|
|
|
|
|
|
|
const transaction = new Transaction().add({
|
|
|
|
keys: [program.publicKey],
|
|
|
|
programId: this.programId,
|
|
|
|
userdata,
|
|
|
|
});
|
2018-10-24 09:21:49 -07:00
|
|
|
transactions.push(sendAndConfirmTransaction(this.connection, program, transaction));
|
2018-10-23 20:56:54 -07:00
|
|
|
|
2018-11-01 20:35:19 -07:00
|
|
|
// Run up to 8 Loads in parallel to prevent too many parallel transactions from
|
|
|
|
// getting rejected with AccountInUse.
|
|
|
|
//
|
|
|
|
// TODO: 8 was selected empirically and should probably be revisited
|
|
|
|
if (transactions.length === 8) {
|
|
|
|
await Promise.all(transactions);
|
|
|
|
transactions = [];
|
|
|
|
}
|
|
|
|
|
2018-10-23 20:56:54 -07:00
|
|
|
offset += chunkSize;
|
|
|
|
array = array.slice(chunkSize);
|
|
|
|
}
|
2018-10-24 09:21:49 -07:00
|
|
|
await Promise.all(transactions);
|
2018-10-17 09:35:24 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Finalize an account loaded with program data for execution
|
|
|
|
*
|
|
|
|
* @param program `load()`ed Account
|
|
|
|
*/
|
|
|
|
async finalize(program: Account) {
|
|
|
|
const userdataLayout = BufferLayout.struct([
|
|
|
|
BufferLayout.u32('instruction'),
|
|
|
|
]);
|
|
|
|
|
|
|
|
const userdata = Buffer.alloc(userdataLayout.span);
|
|
|
|
userdataLayout.encode(
|
|
|
|
{
|
|
|
|
instruction: 1, // Finalize instruction
|
|
|
|
},
|
|
|
|
userdata,
|
|
|
|
);
|
|
|
|
|
2018-10-30 11:09:48 -07:00
|
|
|
const transaction = new Transaction();
|
|
|
|
|
|
|
|
transaction.add({
|
2018-10-17 09:35:24 -07:00
|
|
|
keys: [program.publicKey],
|
|
|
|
programId: this.programId,
|
|
|
|
userdata,
|
|
|
|
});
|
2018-10-30 11:09:48 -07:00
|
|
|
transaction.add(SystemProgram.spawn(program.publicKey));
|
2018-10-18 14:29:57 -07:00
|
|
|
await sendAndConfirmTransaction(this.connection, program, transaction);
|
2018-10-17 09:35:24 -07:00
|
|
|
}
|
|
|
|
}
|