Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Generate sequential IDs for each instance of a struct

Tags:

rust

I'm writing a system where I have a collection of Objects, and each Object has a unique integral ID. Here's how I would do it in C++:

class Object {
public:
  Object(): id_(nextId_++) { }

private:
  int id_;
  static int nextId_;
}

int Object::nextId_ = 1;

This is obviously not thread_safe, but if I wanted it to be, I could make nextId_ an std::atomic_int, or wrap a mutex around the nextId_++ expression.

How would I do this in (preferably safe) Rust? There's no static struct members, nor are global mutable variables safe. I could always pass nextId into the new function, but these objects are going to be allocated in a number of places, and I would prefer not to pipe the nextId number hither and yon. Thoughts?

like image 927
anjruu Avatar asked Oct 04 '15 16:10

anjruu


1 Answers

Atomic variables can live in statics, so you can use it relatively straightforwardly (the downside is that you have global state).

Example code: (playground link)

use std::{
    sync::atomic::{AtomicUsize, Ordering},
    thread,
};

static OBJECT_COUNTER: AtomicUsize = AtomicUsize::new(0);

#[derive(Debug)]
struct Object(usize);

impl Object {
    fn new() -> Self {
        Object(OBJECT_COUNTER.fetch_add(1, Ordering::SeqCst))
    }
}

fn main() {
    let threads = (0..10)
        .map(|_| thread::spawn(|| Object::new()))
        .collect::<Vec<_>>();

    for t in threads {
        println!("{:?}", t.join().unwrap());
    }
}
like image 144
bluss Avatar answered Oct 25 '22 23:10

bluss