/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.ignite.internal.processors.cache.distributed;
import java.io.Externalizable;
import java.nio.ByteBuffer;
import java.util.ArrayList;
import java.util.List;
import org.apache.ignite.IgniteCheckedException;
import org.apache.ignite.IgniteLogger;
import org.apache.ignite.internal.GridDirectCollection;
import org.apache.ignite.internal.GridDirectTransient;
import org.apache.ignite.internal.processors.cache.CacheObject;
import org.apache.ignite.internal.processors.cache.GridCacheSharedContext;
import org.apache.ignite.internal.processors.cache.version.GridCacheVersion;
import org.apache.ignite.internal.util.tostring.GridToStringInclude;
import org.apache.ignite.internal.util.typedef.F;
import org.apache.ignite.internal.util.typedef.internal.S;
import org.apache.ignite.internal.util.typedef.internal.U;
import org.apache.ignite.lang.IgniteUuid;
import org.apache.ignite.plugin.extensions.communication.MessageCollectionItemType;
import org.apache.ignite.plugin.extensions.communication.MessageReader;
import org.apache.ignite.plugin.extensions.communication.MessageWriter;
import org.jetbrains.annotations.Nullable;
/**
* Lock response message.
*/
public class GridDistributedLockResponse extends GridDistributedBaseMessage {
/** */
private static final long serialVersionUID = 0L;
/** Future ID. */
private IgniteUuid futId;
/** Error. */
@GridDirectTransient
private Throwable err;
/** Serialized error. */
private byte[] errBytes;
/** Values. */
@GridToStringInclude
@GridDirectCollection(CacheObject.class)
private List<CacheObject> vals;
/**
* Empty constructor (required by {@link Externalizable}).
*/
public GridDistributedLockResponse() {
/* No-op. */
}
/**
* @param cacheId Cache ID.
* @param lockVer Lock version.
* @param futId Future ID.
* @param cnt Key count.
* @param addDepInfo Deployment info.
*/
public GridDistributedLockResponse(int cacheId,
GridCacheVersion lockVer,
IgniteUuid futId,
int cnt,
boolean addDepInfo) {
super(lockVer, cnt, addDepInfo);
assert futId != null;
this.cacheId = cacheId;
this.futId = futId;
vals = new ArrayList<>(cnt);
}
/**
* @param cacheId Cache ID.
* @param lockVer Lock ID.
* @param futId Future ID.
* @param err Error.
* @param addDepInfo Deployment info.
*/
public GridDistributedLockResponse(int cacheId,
GridCacheVersion lockVer,
IgniteUuid futId,
Throwable err,
boolean addDepInfo) {
super(lockVer, 0, addDepInfo);
assert futId != null;
this.cacheId = cacheId;
this.futId = futId;
this.err = err;
}
/**
* @param cacheId Cache ID.
* @param lockVer Lock ID.
* @param futId Future ID.
* @param cnt Count.
* @param err Error.
* @param addDepInfo Deployment info.
*/
public GridDistributedLockResponse(int cacheId,
GridCacheVersion lockVer,
IgniteUuid futId,
int cnt,
Throwable err,
boolean addDepInfo) {
super(lockVer, cnt, addDepInfo);
assert futId != null;
this.cacheId = cacheId;
this.futId = futId;
this.err = err;
vals = new ArrayList<>(cnt);
}
/**
*
* @return Future ID.
*/
public IgniteUuid futureId() {
return futId;
}
/** {@inheritDoc} */
@Override public Throwable error() {
return err;
}
/**
* @param err Error to set.
*/
public void error(Throwable err) {
this.err = err;
}
/**
* @param val Value.
*/
public void addValue(CacheObject val) {
vals.add(val);
}
/**
* @return Values size.
*/
protected int valuesSize() {
return vals.size();
}
/**
* @param idx Index.
* @return Value for given index.
*/
@Nullable public CacheObject value(int idx) {
if (!F.isEmpty(vals))
return vals.get(idx);
return null;
}
/** {@inheritDoc} */
@Override public IgniteLogger messageLogger(GridCacheSharedContext ctx) {
return ctx.txLockMessageLogger();
}
/** {@inheritDoc}
* @param ctx*/
@Override public void prepareMarshal(GridCacheSharedContext ctx) throws IgniteCheckedException {
super.prepareMarshal(ctx);
prepareMarshalCacheObjects(vals, ctx.cacheContext(cacheId));
if (err != null && errBytes == null)
errBytes = U.marshal(ctx.marshaller(), err);
}
/** {@inheritDoc} */
@Override public void finishUnmarshal(GridCacheSharedContext ctx, ClassLoader ldr) throws IgniteCheckedException {
super.finishUnmarshal(ctx, ldr);
finishUnmarshalCacheObjects(vals, ctx.cacheContext(cacheId), ldr);
if (errBytes != null)
err = U.unmarshal(ctx, errBytes, U.resolveClassLoader(ldr, ctx.gridConfig()));
}
/** {@inheritDoc} */
@Override public boolean writeTo(ByteBuffer buf, MessageWriter writer) {
writer.setBuffer(buf);
if (!super.writeTo(buf, writer))
return false;
if (!writer.isHeaderWritten()) {
if (!writer.writeHeader(directType(), fieldsCount()))
return false;
writer.onHeaderWritten();
}
switch (writer.state()) {
case 7:
if (!writer.writeByteArray("errBytes", errBytes))
return false;
writer.incrementState();
case 8:
if (!writer.writeIgniteUuid("futId", futId))
return false;
writer.incrementState();
case 9:
if (!writer.writeCollection("vals", vals, MessageCollectionItemType.MSG))
return false;
writer.incrementState();
}
return true;
}
/** {@inheritDoc} */
@Override public boolean readFrom(ByteBuffer buf, MessageReader reader) {
reader.setBuffer(buf);
if (!reader.beforeMessageRead())
return false;
if (!super.readFrom(buf, reader))
return false;
switch (reader.state()) {
case 7:
errBytes = reader.readByteArray("errBytes");
if (!reader.isLastRead())
return false;
reader.incrementState();
case 8:
futId = reader.readIgniteUuid("futId");
if (!reader.isLastRead())
return false;
reader.incrementState();
case 9:
vals = reader.readCollection("vals", MessageCollectionItemType.MSG);
if (!reader.isLastRead())
return false;
reader.incrementState();
}
return reader.afterMessageRead(GridDistributedLockResponse.class);
}
/** {@inheritDoc} */
@Override public short directType() {
return 22;
}
/** {@inheritDoc} */
@Override public byte fieldsCount() {
return 10;
}
/** {@inheritDoc} */
@Override public String toString() {
return S.toString(GridDistributedLockResponse.class, this,
"super", super.toString());
}
}