Examples with IndexPartGetRowRequest com.tencent.angel.ps.server.data.request.IndexPartGetRowRequest used on opensource projects

Example 1 with IndexPartGetRowRequest

use of com.tencent.angel.ps.server.data.request.IndexPartGetRowRequest in project angel by Tencent.

the class IndexGetRowHandler method parseRequest.

@Override
public RequestData parseRequest(ByteBuf in) {
    IndexPartGetRowRequest request = new IndexPartGetRowRequest();
    request.deserialize(in);
    return request;
}

Example 2 with IndexPartGetRowRequest

use of com.tencent.angel.ps.server.data.request.IndexPartGetRowRequest in project angel by Tencent.

the class IndexGetRowHandler method handle.

@Override
public ResponseData handle(RequestHeader header, RequestData data) throws Exception {
    IndexPartGetRowRequest request = (IndexPartGetRowRequest) data;
    KeyPart keyPart = request.getKeyPart();
    ServerRow row = MatrixUtils.getRow(context.getMatrixStorageManager(), header.matrixId, header.partId, request.getRowId());
    ValuePart result;
    StreamIndexPartGetRowResponse response;
    row.startRead();
    try {
        result = ServerRowUtils.getByKeys(row, keyPart);
        response = new StreamIndexPartGetRowResponse(result);
        return response;
    } finally {
        row.endRead();
    }
}

Example 3 with IndexPartGetRowRequest

use of com.tencent.angel.ps.server.data.request.IndexPartGetRowRequest in project angel by Tencent.

the class StreamIndexGetRowHandler method parseRequest.

@Override
public RequestData parseRequest(ByteBuf in) {
    IndexPartGetRowRequest request = new IndexPartGetRowRequest();
    request.deserializeHeader(in);
    request.setIn(in);
    return request;
}

Example 4 with IndexPartGetRowRequest

use of com.tencent.angel.ps.server.data.request.IndexPartGetRowRequest in project angel by Tencent.

the class StreamIndexGetRowHandler method handle.

@Override
public ResponseData handle(RequestHeader requestHeader, RequestData data) throws Exception {
    IndexPartGetRowRequest request = (IndexPartGetRowRequest) data;
    StreamIndexPartGetRowResponse response = new StreamIndexPartGetRowResponse();
    ByteBuf in = request.getInputBuffer();
    ServerBasicTypeRow row = (ServerBasicTypeRow) context.getMatrixStorageManager().getRow(requestHeader.matrixId, request.getRowId(), requestHeader.partId);
    ValueType valueType = MatrixUtils.getValueType(row.getRowType());
    ResponseHeader responseHeader = new ResponseHeader(requestHeader.seqId, requestHeader.methodId, context.getRunningContext().getState(), ResponseType.SUCCESS);
    // Read router type
    RouterType routerType = RouterType.valueOf(in.readInt());
    // Key type
    KeyType keyType = KeyType.valueOf(in.readInt());
    // Row id
    int rowId = in.readInt();
    // Key number
    int size = in.readInt();
    // Calculate final output buffer len
    // Response header
    int buffLen = responseHeader.bufferLen();
    // Data flag
    buffLen += ByteBufSerdeUtils.BOOLEN_LENGTH;
    // Value type
    buffLen += ByteBufSerdeUtils.INT_LENGTH;
    // Row id
    buffLen += ByteBufSerdeUtils.INT_LENGTH;
    // Size
    buffLen += ByteBufSerdeUtils.INT_LENGTH;
    // Data
    buffLen += size * ByteBufSerdeUtils.serializedValueLen(valueType);
    // Allocate final result byte buffer
    ByteBuf resultBuf = ByteBufUtils.allocResultBuf(buffLen, context.isUseDirectBuffer());
    // Write response header
    responseHeader.serialize(resultBuf);
    // Data
    // Value part flag
    ByteBufSerdeUtils.serializeBoolean(resultBuf, true);
    // Value type
    ByteBufSerdeUtils.serializeInt(resultBuf, valueType.getTypeId());
    // Row id
    ByteBufSerdeUtils.serializeInt(resultBuf, rowId);
    // Values number
    ByteBufSerdeUtils.serializeInt(resultBuf, size);
    // Result data
    if (request.getFunc() == null) {
        row.startRead();
        try {
            row.indexGet(keyType, size, in, resultBuf, null);
        } finally {
            row.endRead();
        }
    } else {
        row.startWrite();
        try {
            row.indexGet(keyType, size, in, resultBuf, request.getFunc());
        } finally {
            row.endWrite();
        }
    }
    response.setOutputBuffer(resultBuf);
    return response;
}

Example 5 with IndexPartGetRowRequest

use of com.tencent.angel.ps.server.data.request.IndexPartGetRowRequest in project angel by Tencent.

the class UserRequestAdapter method sendIndexGetRowRequest.

private void sendIndexGetRowRequest(MatrixTransportClient matrixClient, int userRequestId, int matrixId, int rowId, int partId, KeyPart keyPart, InitFunc func) {
    // Request header
    RequestHeader header = createRequestHeader(userRequestId, TransportMethod.INDEX_GET_ROW, matrixId, partId);
    // Request body
    IndexPartGetRowRequest requestData = new IndexPartGetRowRequest(rowId, keyPart, func);
    // Request
    Request request = new Request(header, requestData);
    // Send the request
    matrixClient.sendGetRequest(request);
}