com.google.protobuf.DescriptorProtos.FieldOptions Java Examples

/**
 * 插入对象到给定的表中。注意，这里并不自动生成ID。
 * 
 * @param message
 * @param tableName
 * @return
 */
public long insert(M message, String tableName) {
	StringBuilder insertSql = new StringBuilder("insert into ");
	insertSql.append(tableName).append("(");
	StringBuilder fields = new StringBuilder("");
	StringBuilder values = new StringBuilder("");
	List<Object> args = new ArrayList<Object>();
	Map<FieldDescriptor, Object> fieldMap = message.getAllFields();

	for (Entry<FieldDescriptor, Object> entry : fieldMap.entrySet()) {
		FieldDescriptor fieldDescriptor = entry.getKey();
		FieldOptions fieldOptions = fieldDescriptor.getOptions();
		ColumnFieldOption columnFieldOption = fieldOptions
				.getExtension(Taglib.columnOption);
		String fieldName = fieldDescriptor.getName();
		Object value = entry.getValue();

		if (columnFieldOption.getColumnType() == ColumnType.DATETIME
				|| columnFieldOption.getColumnType() == ColumnType.TIMESTAMP) {// datetime类型
			if (value != null && (long) value > 0) {
				fields.append('`').append(fieldName).append("`,");
				values.append("?, ");
				args.add(new Timestamp((long) value));
			}
		} else {
			fields.append('`').append(fieldName).append("`,");
			values.append("?, ");
			args.add(value);
		}
	}
	int tmpIndex = fields.lastIndexOf(",");
	insertSql.append(fields.substring(0, tmpIndex)).append(") values(");
	tmpIndex = values.lastIndexOf(",");
	insertSql.append(values.substring(0, tmpIndex)).append(")");
	String sql = insertSql.toString();
	return update(sql, args);
}
 

private FieldDescriptorProto generateField(Field field) {
  FieldDescriptorProto.Builder builder = FieldDescriptorProto.newBuilder();
  builder.setName(getFieldName(field));
  builder.setNumber(field.getNumber());
  builder.setLabel(toLabel(field.getCardinality()));
  builder.setType(toType(field.getKind()));
  if (field.getKind() == Kind.TYPE_ENUM
      || field.getKind() == Kind.TYPE_MESSAGE
      || field.getKind() == Kind.TYPE_GROUP) {
    builder.setTypeName(getTypeName(field.getTypeUrl()));
  }
  // NOTE: extendee not supported
  // NOTE: default_value not supported
  if (field.getOneofIndex() != 0) {
    // Index in the containing type's oneof_decl is zero-based.
    // Index in google.protobuf.type.Field.oneof_index is one-based.
    builder.setOneofIndex(field.getOneofIndex() - 1);
  }
  if (!Strings.isNullOrEmpty(field.getDefaultValue())) {
    builder.setDefaultValue(field.getDefaultValue());
  }
  FieldOptions options = getFieldOptions(field);
  if (!options.equals(FieldOptions.getDefaultInstance())) {
    builder.setOptions(options);
  }
  return builder.build();
}
 

private FieldOptions getFieldOptions(Field field) {
  FieldOptions.Builder builder = FieldOptions.newBuilder();
  if (field.getPacked()) {
    builder.setPacked(true);
  }
  setOptions(builder, field.getOptionsList(), FIELD_OPTION_NAME_PREFIX);
  return builder.build();
}
 

protected void addColumn(Column column) {
    String fieldName = uniqueIdent(ident(column.getName(), false), fieldNames);
    fieldBuilder = messageBuilder.addFieldBuilder();
    fieldBuilder.setName(fieldName);
    fieldBuilder.setLabel(Label.LABEL_OPTIONAL);
    FieldOptions.Builder fieldBuilderOptions = FieldOptions.newBuilder();
    ColumnOptions.Builder columnOptions = ColumnOptions.newBuilder();
    if (!fieldName.equals(column.getName())) {
        columnOptions.setName(column.getName());
    }
    columnOptions.setSqlType(column.getTypeDescription().toUpperCase());
    columnOptions.setUuid(column.getUuid().toString());
    priorField = null;
    if (priorMessage != null) {
        for (FieldDescriptorProto field : priorMessage.getFieldList()) {
            FieldOptions options = field.getOptions();
            if ((options != null) &&
                (options.hasExtension(ColumnOptions.fdbsql))) {
                ColumnOptions coptions = options.getExtension(ColumnOptions.fdbsql);
                if (coptions.getUuid().equals(columnOptions.getUuid())) {
                    priorField = field;
                    break;
                }
            }
        }
    }
    setColumnType(column, columnOptions);
    setFieldNumber();
    fieldBuilderOptions.setExtension(ColumnOptions.fdbsql, columnOptions.build());
    fieldBuilder.setOptions(fieldBuilderOptions);
    if (column.getNullable() && 
        ((column.getDefaultValue() != null) ||
         (column.getDefaultFunction() != null))) {
        addNullForField(column.getName(), fieldBuilder.getNumber());
    }
}
 

protected void addNullForField(String columnName, int forField) {
    String fieldName = uniqueIdent("_" + ident(columnName, false) + "_is_null", fieldNames);
    fieldBuilder = messageBuilder.addFieldBuilder();
    fieldBuilder.setName(fieldName);
    fieldBuilder.setType(Type.TYPE_BOOL);
    fieldBuilder.setLabel(Label.LABEL_OPTIONAL);
    FieldOptions.Builder fieldBuilderOptions = FieldOptions.newBuilder();
    ColumnOptions.Builder columnOptions = ColumnOptions.newBuilder();
    columnOptions.setNullForField(forField);
    priorField = null;
    if (priorMessage != null) {
        for (FieldDescriptorProto field : priorMessage.getFieldList()) {
            FieldOptions options = field.getOptions();
            if ((options != null) &&
                (options.hasExtension(ColumnOptions.fdbsql))) {
                ColumnOptions coptions = options.getExtension(ColumnOptions.fdbsql);
                if (coptions.hasNullForField() &&
                    (coptions.getNullForField() == forField)) {
                    priorField = field;
                    break;
                }
            }
        }
    }
    setFieldNumber();
    fieldBuilderOptions.setExtension(ColumnOptions.fdbsql, columnOptions.build());
    fieldBuilder.setOptions(fieldBuilderOptions);
}
 

protected void addChildTable(Table table) {
    String fieldName = uniqueIdent(ident(table.getName().getTableName(), false), fieldNames);
    fieldBuilder = messageBuilder.addFieldBuilder();
    fieldBuilder.setName(fieldName);
    fieldBuilder.setLabel(Label.LABEL_REPEATED);
    fieldBuilder.setType(Type.TYPE_MESSAGE);
    fieldBuilder.setTypeName(tableMessageNames.get(table));
    FieldOptions.Builder fieldBuilderOptions = FieldOptions.newBuilder();
    ColumnOptions.Builder columnOptions = ColumnOptions.newBuilder();
    columnOptions.setUuid(table.getUuid().toString());
    priorField = null;
    if (priorMessage != null) {
        for (FieldDescriptorProto field : priorMessage.getFieldList()) {
            FieldOptions options = field.getOptions();
            if ((options != null) &&
                (options.hasExtension(ColumnOptions.fdbsql))) {
                ColumnOptions coptions = options.getExtension(ColumnOptions.fdbsql);
                if (coptions.getUuid().equals(columnOptions.getUuid())) {
                    priorField = field;
                    break;
                }
            }
        }
    }
    setFieldNumber();
    fieldBuilderOptions.setExtension(ColumnOptions.fdbsql, columnOptions.build());
    fieldBuilder.setOptions(fieldBuilderOptions);
}
 

private Builder(EnumSerialization enumSerialization, CaseFormat fromFieldNameFormat,
    CaseFormat toFieldNameFormat) {
  this.serializedNameExtensions = new HashSet<Extension<FieldOptions, String>>();
  this.serializedEnumValueExtensions = new HashSet<Extension<EnumValueOptions, String>>();
  setEnumSerialization(enumSerialization);
  setFieldNameSerializationFormat(fromFieldNameFormat, toFieldNameFormat);
}
 

private ProtoTypeAdapter(EnumSerialization enumSerialization,
    CaseFormat protoFormat,
    CaseFormat jsonFormat,
    Set<Extension<FieldOptions, String>> serializedNameExtensions,
    Set<Extension<EnumValueOptions, String>> serializedEnumValueExtensions) {
  this.enumSerialization = enumSerialization;
  this.protoFormat = protoFormat;
  this.jsonFormat = jsonFormat;
  this.serializedNameExtensions = serializedNameExtensions;
  this.serializedEnumValueExtensions = serializedEnumValueExtensions;
}
 

/**
 * Retrieves the custom field name from the given options, and if not found, returns the specified
 * default name.
 */
private String getCustSerializedName(FieldOptions options, String defaultName) {
  for (Extension<FieldOptions, String> extension : serializedNameExtensions) {
    if (options.hasExtension(extension)) {
      return options.getExtension(extension);
    }
  }
  return protoFormat.to(jsonFormat, defaultName);
}
 

/**
 * 
 * @param message
 * @param conditionFields
 * @param conditionParams
 * @param tableName
 * @return
 */
protected int updateMessageByCondition(M message, String[] conditionFields,
		Object[] conditionParams, String tableName) {
	StringBuilder updateSql = new StringBuilder("update ");
	updateSql.append(tableName).append(" set ");
	StringBuilder options = new StringBuilder("");
	List<Object> args = new ArrayList<Object>();
	Map<FieldDescriptor, Object> fieldMap = message.getAllFields();

	for (Entry<FieldDescriptor, Object> entry : fieldMap.entrySet()) {
		FieldDescriptor fieldDescriptor = entry.getKey();
		if (!Arrays.asList(conditionFields).contains(
				fieldDescriptor.getName())) {
			FieldOptions fieldOptions = fieldDescriptor.getOptions();
			ColumnFieldOption columnFieldOption = fieldOptions
					.getExtension(Taglib.columnOption);
			String fieldName = fieldDescriptor.getName();
			Object value = entry.getValue();
			if (columnFieldOption.getColumnType() == ColumnType.DATETIME
					|| columnFieldOption.getColumnType() == ColumnType.TIMESTAMP) {// datetime类型
				if (value != null && (long) value > 0) {
					options.append(fieldName).append("=?, ");
					args.add(new Timestamp((long) value));
				}
			} else {
				options.append(fieldName).append("=?, ");
				args.add(value);
			}
		}
	}
	int tmpIndex = options.lastIndexOf(",");
	updateSql.append(options.substring(0, tmpIndex)).append(" where 1=1 ");
	StringBuilder condition = new StringBuilder();
	if (conditionFields.length != conditionParams.length) {
		throw new IllegalArgumentException("condition error");
	} else {
		for (int i = 0; i < conditionFields.length; i++) {
			condition.append("AND ").append(conditionFields[i])
					.append("=? ");
			args.add(conditionParams[i]);
		}
		updateSql.append(condition);
		String sql = updateSql.toString();
		logger.debug(sql);
		return update(sql, args);
	}
}
 

/** Get the {@code FieldOptions}, defined in {@code descriptor.proto}. */
public FieldOptions getOptions() { return proto.getOptions(); }
 

/**
 * Adds a field proto annotation that, when set, overrides the default field name
 * serialization/deserialization. For example, if you add the '{@code serialized_name}'
 * annotation and you define a field in your proto like the one below:
 *
 * <pre>
 * string client_app_id = 1 [(serialized_name) = "appId"];
 * </pre>
 *
 * ...the adapter will serialize the field using '{@code appId}' instead of the default '
 * {@code clientAppId}'. This lets you customize the name serialization of any proto field.
 */
public Builder addSerializedNameExtension(
    Extension<FieldOptions, String> serializedNameExtension) {
  serializedNameExtensions.add(checkNotNull(serializedNameExtension));
  return this;
}