提交 76777583 编写于 作者: L lizhanhui 提交者: yukon

[ROCKETMQ-74] Fix DataVersion equals defect, closes apache/incubator-rocketmq#50

上级 44ef40ab
...@@ -20,25 +20,25 @@ import java.util.concurrent.atomic.AtomicLong; ...@@ -20,25 +20,25 @@ import java.util.concurrent.atomic.AtomicLong;
import org.apache.rocketmq.remoting.protocol.RemotingSerializable; import org.apache.rocketmq.remoting.protocol.RemotingSerializable;
public class DataVersion extends RemotingSerializable { public class DataVersion extends RemotingSerializable {
private long timestatmp = System.currentTimeMillis(); private long timestamp = System.currentTimeMillis();
private AtomicLong counter = new AtomicLong(0); private AtomicLong counter = new AtomicLong(0);
public void assignNewOne(final DataVersion dataVersion) { public void assignNewOne(final DataVersion dataVersion) {
this.timestatmp = dataVersion.timestatmp; this.timestamp = dataVersion.timestamp;
this.counter.set(dataVersion.counter.get()); this.counter.set(dataVersion.counter.get());
} }
public void nextVersion() { public void nextVersion() {
this.timestatmp = System.currentTimeMillis(); this.timestamp = System.currentTimeMillis();
this.counter.incrementAndGet(); this.counter.incrementAndGet();
} }
public long getTimestatmp() { public long getTimestamp() {
return timestatmp; return timestamp;
} }
public void setTimestatmp(long timestatmp) { public void setTimestamp(long timestamp) {
this.timestatmp = timestatmp; this.timestamp = timestamp;
} }
public AtomicLong getCounter() { public AtomicLong getCounter() {
...@@ -58,16 +58,24 @@ public class DataVersion extends RemotingSerializable { ...@@ -58,16 +58,24 @@ public class DataVersion extends RemotingSerializable {
final DataVersion that = (DataVersion) o; final DataVersion that = (DataVersion) o;
if (timestatmp != that.timestatmp) if (timestamp != that.timestamp) {
return false; return false;
return counter != null ? counter.equals(that.counter) : that.counter == null; }
if (counter != null && that.counter != null) {
return counter.longValue() == that.counter.longValue();
}
return (null == counter) && (null == that.counter);
} }
@Override @Override
public int hashCode() { public int hashCode() {
int result = (int) (timestatmp ^ (timestatmp >>> 32)); int result = (int) (timestamp ^ (timestamp >>> 32));
result = 31 * result + (counter != null ? counter.hashCode() : 0); if (null != counter) {
long l = counter.get();
result = 31 * result + (int)(l ^ (l >>> 32));
}
return result; return result;
} }
} }
/*
* 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.rocketmq.common;
import java.util.concurrent.atomic.AtomicLong;
import org.junit.Assert;
import org.junit.Test;
public class DataVersionTest {
@Test
public void testEquals() {
DataVersion dataVersion = new DataVersion();
DataVersion other = new DataVersion();
other.setTimestamp(dataVersion.getTimestamp());
Assert.assertTrue(dataVersion.equals(other));
}
@Test
public void testEquals_falseWhenCounterDifferent() {
DataVersion dataVersion = new DataVersion();
DataVersion other = new DataVersion();
other.setCounter(new AtomicLong(1L));
other.setTimestamp(dataVersion.getTimestamp());
Assert.assertFalse(dataVersion.equals(other));
}
@Test
public void testEquals_falseWhenCounterDifferent2() {
DataVersion dataVersion = new DataVersion();
DataVersion other = new DataVersion();
other.setCounter(null);
other.setTimestamp(dataVersion.getTimestamp());
Assert.assertFalse(dataVersion.equals(other));
}
@Test
public void testEquals_falseWhenCounterDifferent3() {
DataVersion dataVersion = new DataVersion();
dataVersion.setCounter(null);
DataVersion other = new DataVersion();
other.setTimestamp(dataVersion.getTimestamp());
Assert.assertFalse(dataVersion.equals(other));
}
@Test
public void testEquals_trueWhenCountersBothNull() {
DataVersion dataVersion = new DataVersion();
dataVersion.setCounter(null);
DataVersion other = new DataVersion();
other.setCounter(null);
other.setTimestamp(dataVersion.getTimestamp());
Assert.assertTrue(dataVersion.equals(other));
}
}
\ No newline at end of file
...@@ -191,7 +191,7 @@ public class DefaultRequestProcessor implements NettyRequestProcessor { ...@@ -191,7 +191,7 @@ public class DefaultRequestProcessor implements NettyRequestProcessor {
registerBrokerBody = RegisterBrokerBody.decode(request.getBody(), RegisterBrokerBody.class); registerBrokerBody = RegisterBrokerBody.decode(request.getBody(), RegisterBrokerBody.class);
} else { } else {
registerBrokerBody.getTopicConfigSerializeWrapper().getDataVersion().setCounter(new AtomicLong(0)); registerBrokerBody.getTopicConfigSerializeWrapper().getDataVersion().setCounter(new AtomicLong(0));
registerBrokerBody.getTopicConfigSerializeWrapper().getDataVersion().setTimestatmp(0); registerBrokerBody.getTopicConfigSerializeWrapper().getDataVersion().setTimestamp(0);
} }
RegisterBrokerResult result = this.namesrvController.getRouteInfoManager().registerBroker( RegisterBrokerResult result = this.namesrvController.getRouteInfoManager().registerBroker(
...@@ -227,7 +227,7 @@ public class DefaultRequestProcessor implements NettyRequestProcessor { ...@@ -227,7 +227,7 @@ public class DefaultRequestProcessor implements NettyRequestProcessor {
} else { } else {
topicConfigWrapper = new TopicConfigSerializeWrapper(); topicConfigWrapper = new TopicConfigSerializeWrapper();
topicConfigWrapper.getDataVersion().setCounter(new AtomicLong(0)); topicConfigWrapper.getDataVersion().setCounter(new AtomicLong(0));
topicConfigWrapper.getDataVersion().setTimestatmp(0); topicConfigWrapper.getDataVersion().setTimestamp(0);
} }
RegisterBrokerResult result = this.namesrvController.getRouteInfoManager().registerBroker( RegisterBrokerResult result = this.namesrvController.getRouteInfoManager().registerBroker(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册