blob: edf921e21b13b9aaeaf57b6f79bea37982820ec1 [file] [log] [blame]
// Copyright (C) 2018 The Android Open Source Project
//
// Licensed 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 com.google.gerrit.entities.converter;
import com.google.errorprone.annotations.Immutable;
import com.google.gerrit.entities.Account;
import com.google.gerrit.entities.ChangeMessage;
import com.google.gerrit.entities.PatchSet;
import com.google.gerrit.proto.Entities;
import com.google.protobuf.Parser;
import java.time.Instant;
import java.util.Objects;
@Immutable
public enum ChangeMessageProtoConverter
implements ProtoConverter<Entities.ChangeMessage, ChangeMessage> {
INSTANCE;
private final ProtoConverter<Entities.ChangeMessage_Key, ChangeMessage.Key>
changeMessageKeyConverter = ChangeMessageKeyProtoConverter.INSTANCE;
private final ProtoConverter<Entities.Account_Id, Account.Id> accountIdConverter =
AccountIdProtoConverter.INSTANCE;
private final ProtoConverter<Entities.PatchSet_Id, PatchSet.Id> patchSetIdConverter =
PatchSetIdProtoConverter.INSTANCE;
@Override
public Entities.ChangeMessage toProto(ChangeMessage changeMessage) {
Entities.ChangeMessage.Builder builder =
Entities.ChangeMessage.newBuilder()
.setKey(changeMessageKeyConverter.toProto(changeMessage.getKey()));
Account.Id author = changeMessage.getAuthor();
if (author != null) {
builder.setAuthorId(accountIdConverter.toProto(author));
}
Instant writtenOn = changeMessage.getWrittenOn();
if (writtenOn != null) {
builder.setWrittenOn(writtenOn.toEpochMilli());
}
// Build proto with template representation of the message. Templates are parsed when message is
// extracted from cache.
String message = changeMessage.getMessage();
if (message != null) {
builder.setMessage(message);
}
PatchSet.Id patchSetId = changeMessage.getPatchSetId();
if (patchSetId != null) {
builder.setPatchset(patchSetIdConverter.toProto(patchSetId));
}
String tag = changeMessage.getTag();
if (tag != null) {
builder.setTag(tag);
}
Account.Id realAuthor = changeMessage.getRealAuthor();
// ChangeMessage#getRealAuthor automatically delegates to ChangeMessage#getAuthor if the real
// author is not set. However, the previous protobuf representation kept 'realAuthor' empty if
// it wasn't set. To ensure binary compatibility, simulate the previous behavior.
if (realAuthor != null && !Objects.equals(realAuthor, author)) {
builder.setRealAuthor(accountIdConverter.toProto(realAuthor));
}
return builder.build();
}
@Override
public ChangeMessage fromProto(Entities.ChangeMessage proto) {
ChangeMessage.Key key =
proto.hasKey() ? changeMessageKeyConverter.fromProto(proto.getKey()) : null;
Account.Id author =
proto.hasAuthorId() ? accountIdConverter.fromProto(proto.getAuthorId()) : null;
Instant writtenOn = proto.hasWrittenOn() ? Instant.ofEpochMilli(proto.getWrittenOn()) : null;
PatchSet.Id patchSetId =
proto.hasPatchset() ? patchSetIdConverter.fromProto(proto.getPatchset()) : null;
// Only template representation of the message is stored in entity. Templates should be replaced
// before being served to the users.
String messageTemplate = proto.hasMessage() ? proto.getMessage() : null;
String tag = proto.hasTag() ? proto.getTag() : null;
Account.Id realAuthor =
proto.hasRealAuthor() ? accountIdConverter.fromProto(proto.getRealAuthor()) : null;
ChangeMessage changeMessage =
ChangeMessage.create(key, author, writtenOn, patchSetId, messageTemplate, realAuthor, tag);
return changeMessage;
}
@Override
public Parser<Entities.ChangeMessage> getParser() {
return Entities.ChangeMessage.parser();
}
}