BytedanceRPC commented on a change in pull request #2307:
URL: https://github.com/apache/thrift/pull/2307#discussion_r569102242
##########
File path: compiler/cpp/src/thrift/generate/t_go_generator.cc
##########
@@ -3463,6 +3543,109 @@ void
t_go_generator::generate_serialize_list_element(ostream& out, t_list* tlist
generate_serialize_field(out, &efield, prefix);
}
+/**
+ * Compares any type
+ */
+void t_go_generator::generate_go_equals(ostream& out, t_type* ori_type, string
tgt, string src) {
+
+ t_type* ttype = get_true_type(ori_type);
+ // Do nothing for void types
+ if (ttype->is_void()) {
+ throw "compiler error: cannot generate equals for void type: " + tgt;
+ }
+
+ if (ttype->is_struct() || ttype->is_xception()) {
+ generate_go_equals_struct(out, ttype, tgt, src);
+ } else if (ttype->is_container()) {
+ generate_go_equals_container(out, ttype, tgt, src);
+ } else if (ttype->is_base_type() || ttype->is_enum()) {
+ out << indent() << "if ";
+ if (ttype->is_base_type()) {
+ t_base_type::t_base tbase = ((t_base_type*)ttype)->get_base();
+ switch (tbase) {
+ case t_base_type::TYPE_VOID:
+ throw "compiler error: cannot equals void: " + tgt;
+ break;
+
+ case t_base_type::TYPE_STRING:
+ if (ttype->is_binary()) {
+ out << "bytes.Compare(" << tgt << ", " << src << ") != 0";
+ } else {
+ out << tgt << " != " << src;
+ }
+ break;
+
+ case t_base_type::TYPE_BOOL:
+ case t_base_type::TYPE_I8:
+ case t_base_type::TYPE_I16:
+ case t_base_type::TYPE_I32:
+ case t_base_type::TYPE_I64:
+ case t_base_type::TYPE_DOUBLE:
+ out << tgt << " != " << src;
+ break;
+
+ default:
+ throw "compiler error: no Go name for base type " +
t_base_type::t_base_name(tbase);
+ }
+ } else if (ttype->is_enum()) {
+ out << tgt << " != " << src;
+ }
+
+ out << " { return false }" << endl;
+ } else {
+ throw "compiler error: Invalid type in generate_go_equals '" +
ttype->get_name() + "' for '"
+ + tgt + "'";
+ }
+}
+
+/**
+ * Compares the members of a struct
+ */
+void t_go_generator::generate_go_equals_struct(ostream& out,
+ t_type* ttype,
+ string tgt,
+ string src) {
+ (void)ttype;
Review comment:
try to keep `generate_go_equals*` with the same arguments, but `ttype`
not really used here.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
