Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit f9b68855 authored by Jakub Pawlowski's avatar Jakub Pawlowski Committed by android-build-merger
Browse files

Add IdGenerator 

am: 69d75b47

Change-Id: I974608ed960f88067abfbab07f917de8950f4522
parents ca818226 69d75b47

system/common/Android.bp

+1 −0
Original line number Diff line number Diff line
@@ -40,6 +40,7 @@ cc_test { 
        "repeating_timer_unittest.cc",

        "state_machine_unittest.cc",

        "time_util_unittest.cc",

        "id_generator_unittest.cc",

    ],

    shared_libs: [

        "libprotobuf-cpp-lite",

system/common/id_generator.h

0 → 100644
+48 −0
Original line number Diff line number Diff line 
/*

 * Copyright 2019 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.

 */



#pragma once



#include <array>



/* Helper class generating N unique ids, from 0 to N-1 */

template <std::size_t N>

class IdGenerator {

 public:

  static int ALL_USED;



  IdGenerator() : in_use_{} {}



  /* Returns next free id, or ALL_USED if no ids left */

  int GetNext() {

    for (std::size_t i = 0; i < N; i++) {

      if (!in_use_[i]) {

        in_use_[i] = true;

        return i;

      }

    }

    return ALL_USED;

  }



  /* Release given ID */

  void Release(int id) { in_use_[id] = false; }



 private:

  std::array<bool, N> in_use_;

};



template <std::size_t N>

int IdGenerator<N>::ALL_USED = -1;
 
 No newline at end of file

system/common/id_generator_unittest.cc

0 → 100644
+37 −0
Original line number Diff line number Diff line 
/*

 * Copyright 2019 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.

 */



#include "common/id_generator.h"



#include <gtest/gtest.h>



TEST(IdGeneratorTest, sanity_test) {

  IdGenerator<5> generator;

  ASSERT_EQ(0, generator.GetNext());

  ASSERT_EQ(1, generator.GetNext());

  ASSERT_EQ(2, generator.GetNext());

  ASSERT_EQ(3, generator.GetNext());

  ASSERT_EQ(4, generator.GetNext());

  ASSERT_EQ(generator.ALL_USED, generator.GetNext());



  generator.Release(3);

  ASSERT_EQ(3, generator.GetNext());



  generator.Release(0);

  generator.Release(2);

  ASSERT_EQ(0, generator.GetNext());

  ASSERT_EQ(2, generator.GetNext());

}
 
 No newline at end of file