9 releases (breaking)

0.7.0 May 2, 2024
0.6.0 Jul 10, 2023
0.5.0 Feb 24, 2023
0.4.0 Jan 11, 2023
0.1.0 Dec 6, 2021

#1118 in Database interfaces

MIT license

35KB
527 lines

nene

nene is a command-line tool to generate Rust code for Google Cloud Spanner.
nene uses database schema to generate code by using Information Schema. nene runs SQL queries against tables in INFORMATION_SCHEMA to fetch metadata for a database, and applies the metadata to Go templates to generate code/models to acccess Cloud Spanner.

crates.io CI

Installation

cargo install nene

Usage

export RUST_LOG=info
export SPANNER_DSN=projects/local-project/instances/test-instance/databases/local-database
# if you don't use emulator use GOOGLE_APPLICATION_CREDENTIALS instead of SPANNER_EMULATOR_HOST
export SPANNER_EMULATOR_HOST=localhost:9010 
mkdir ./gen
nene -o ./gen -j -d
  • -i

    • template directory.
    • see template directory structure
    • if not specified default template are used.
  • -o

    • output directory
    • default directory is ./gen
  • -j

    • add serde::Serialize and serde::Deserialize
  • -d

    • implements Default trait

Generated file with default template

Default template generates the files for google-cloud-spanner.

// DON'T EDIT. this code is generated by nene.
use google_cloud_googleapis::spanner::v1::Mutation;
use google_cloud_spanner::client::{Error};
use google_cloud_spanner::key::Key;
use google_cloud_spanner::mutation::{
  delete, insert_or_update_struct, insert_struct, replace_struct, update_struct,
};
use google_cloud_spanner::reader::AsyncIterator;
use google_cloud_spanner::row::{Error as RowError, Row};
use google_cloud_spanner::statement::Statement;
use google_cloud_spanner::transaction::Transaction;
use google_cloud_spanner::transaction::CallOptions;
use google_cloud_spanner_derive::Table;
use std::convert::TryFrom;

pub const TABLE_NAME: &str = "User";
pub const COLUMN_USER_ID: &str = "UserId";
pub const COLUMN_NOT_NULL_INT_64: &str = "NotNullINT64";
pub const COLUMN_NULLABLE_INT_64: &str = "NullableINT64";
pub const COLUMN_NOT_NULL_FLOAT_64: &str = "NotNullFloat64";
pub const COLUMN_NULLABLE_FLOAT_64: &str = "NullableFloat64";
pub const COLUMN_NOT_NULL_BOOL: &str = "NotNullBool";
pub const COLUMN_NULLABLE_BOOL: &str = "NullableBool";
pub const COLUMN_NOT_NULL_BYTE_ARRAY: &str = "NotNullByteArray";
pub const COLUMN_NULLABLE_BYTE_ARRAY: &str = "NullableByteArray";
pub const COLUMN_NOT_NULL_NUMERIC: &str = "NotNullNumeric";
pub const COLUMN_NULLABLE_NUMERIC: &str = "NullableNumeric";
pub const COLUMN_NOT_NULL_TIMESTAMP: &str = "NotNullTimestamp";
pub const COLUMN_NULLABLE_TIMESTAMP: &str = "NullableTimestamp";
pub const COLUMN_NOT_NULL_DATE: &str = "NotNullDate";
pub const COLUMN_NULLABLE_DATE: &str = "NullableDate";
pub const COLUMN_NOT_NULL_ARRAY: &str = "NotNullArray";
pub const COLUMN_NULLABLE_ARRAY: &str = "NullableArray";
pub const COLUMN_NULLABLE_STRING: &str = "NullableString";
pub const COLUMN_UPDATED_AT: &str = "UpdatedAt";

#[derive(Debug,Clone,Table,serde::Serialize,serde::Deserialize)]
pub struct User {
  #[spanner(name = "UserId")]
  pub user_id: String,
  #[spanner(name = "NotNullINT64")]
  pub not_null_int_64: i64,
  #[spanner(name = "NullableINT64")]
  pub nullable_int_64: Option<i64>,
  #[spanner(name = "NotNullFloat64")]
  pub not_null_float_64: f64,
  #[spanner(name = "NullableFloat64")]
  pub nullable_float_64: Option<f64>,
  #[spanner(name = "NotNullBool")]
  pub not_null_bool: bool,
  #[spanner(name = "NullableBool")]
  pub nullable_bool: Option<bool>,
  #[spanner(name = "NotNullByteArray")]
  pub not_null_byte_array: Vec<u8>,
  #[spanner(name = "NullableByteArray")]
  pub nullable_byte_array: Option<Vec<u8>>,
  #[spanner(name = "NotNullNumeric")]
  pub not_null_numeric: google_cloud_spanner::bigdecimal::BigDecimal,
  #[spanner(name = "NullableNumeric")]
  pub nullable_numeric: Option<google_cloud_spanner::bigdecimal::BigDecimal>,
  #[serde(with = "time::serde::rfc3339")]
  #[spanner(name = "NotNullTimestamp")]
  pub not_null_timestamp: time::OffsetDateTime,
  #[serde(default,with = "time::serde::rfc3339::option")]
  #[spanner(name = "NullableTimestamp")]
  pub nullable_timestamp: Option<time::OffsetDateTime>,
  #[spanner(name = "NotNullDate")]
  pub not_null_date: time::Date,
  #[spanner(name = "NullableDate")]
  pub nullable_date: Option<time::Date>,
  #[spanner(name = "NotNullArray")]
  pub not_null_array: Vec<i64>,
  #[spanner(name = "NullableArray")]
  pub nullable_array: Option<Vec<i64>>,
  #[spanner(name = "NullableString")]
  pub nullable_string: Option<String>,
  #[serde(with = "time::serde::rfc3339")]
  #[spanner(name = "UpdatedAt",commitTimestamp)]
  pub updated_at: time::OffsetDateTime,
}

impl Default for User {
  fn default() -> Self {
    Self {
      user_id: Default::default(),
      not_null_int_64: Default::default(),
      nullable_int_64: Default::default(),
      not_null_float_64: Default::default(),
      nullable_float_64: Default::default(),
      not_null_bool: Default::default(),
      nullable_bool: Default::default(),
      not_null_byte_array: Default::default(),
      nullable_byte_array: Default::default(),
      not_null_numeric: Default::default(),
      nullable_numeric: Default::default(),
      not_null_timestamp: time::OffsetDateTime::now_utc(),
      nullable_timestamp: Default::default(),
      not_null_date: time::OffsetDateTime::now_utc().date(),
      nullable_date: Default::default(),
      not_null_array: Default::default(),
      nullable_array: Default::default(),
      nullable_string: Default::default(),
      updated_at: time::OffsetDateTime::now_utc(),
    }
  }
}
impl User {
  pub fn insert(&self) -> Mutation {
    insert_struct(TABLE_NAME, &self)
  }

  pub fn update(&self) -> Mutation {
    update_struct(TABLE_NAME, &self)
  }

  pub fn replace(&self) -> Mutation {
    replace_struct(TABLE_NAME, &self)
  }

  pub fn insert_or_update(&self) -> Mutation {
    insert_or_update_struct(TABLE_NAME, &self)
  }

  pub fn delete(&self) -> Mutation {
    delete(TABLE_NAME, Key::new(&self.user_id))
  }

  pub async fn find_by_pk(
    tx: &mut Transaction, user_id: &str, options: Option<CallOptions>
  ) -> Result<Option<Self>, Error> {
    let mut stmt = Statement::new("SELECT * From User WHERE UserId = @UserId");
    stmt.add_param(COLUMN_USER_ID, &user_id);
    let mut rows = read_by_statement(tx, stmt, options).await?;
    if !rows.is_empty() {
      Ok(rows.pop())
    } else {
      Ok(None)
    }
  }
}

async fn read_by_statement<T: TryFrom<Row, Error = RowError>>(
  tx: &mut Transaction,
  stmt: Statement,
  options: Option<CallOptions>,
) -> Result<Vec<T>, Error> {
  let mut reader = tx.query(stmt).await?;
  if options.is_some() {
    reader.set_call_options(options.unwrap());
  }
  let mut result = vec![];
  while let Some(row) = reader.next().await? {
    result.push(row.try_into()?);
  }
  Ok(result)
}

Dependencies

~28–39MB
~694K SLoC