如何在Firebase安卓应用中为用户分配角色？区分员工与管理者登录

嘿，我来帮你搞定Firebase安卓应用里员工和管理者的角色区分问题！下面是一步步的实现方案，都是实战中常用的套路：

首先得在Firebase里存用户的角色信息，推荐两种方式，选你用的数据库就行：

方式1：Cloud Firestore #

创建一个users集合，每个文档的ID用用户的Firebase Auth UID，文档里加一个role字段，值设为"employee"（员工）或"manager"（管理者）。结构大概是这样：

users >
  {用户UID} >
    role: "manager"
    email: "xxx@example.com"

方式2：实时数据库 #

创建一个users节点，每个子节点用用户UID，下面加role字段：

users >
  {用户UID} >
    role: "employee"
    email: "xxx@example.com"

角色分配一般是管理员操作，可以通过Firebase控制台手动加，也可以写个管理员端的代码（比如只有授权管理者能访问的界面）：

Firestore分配角色代码示例（Kotlin）： #

// 假设你拿到了目标用户的UID
val targetUserUid = "用户的UID"
val db = FirebaseFirestore.getInstance()

db.collection("users").document(targetUserUid)
    .set(mapOf("role" to "manager")) // 这里可以设为employee或manager
    .addOnSuccessListener {
        // 角色设置成功，提示管理员
        Toast.makeText(this, "角色分配完成", Toast.LENGTH_SHORT).show()
    }
    .addOnFailureListener { e ->
        // 处理错误，比如网络问题
        Toast.makeText(this, "分配失败：${e.message}", Toast.LENGTH_SHORT).show()
    }

实时数据库分配角色代码示例： #

val targetUserUid = "用户的UID"
val database = FirebaseDatabase.getInstance()

database.getReference("users").child(targetUserUid).child("role")
    .setValue("employee")
    .addOnSuccessListener {
        Toast.makeText(this, "角色分配完成", Toast.LENGTH_SHORT).show()
    }
    .addOnFailureListener { e ->
        Toast.makeText(this, "分配失败：${e.message}", Toast.LENGTH_SHORT).show()
    }

用户登录成功后，我们需要获取他的角色，然后跳转不同页面或者展示不同功能：

登录核心代码： #

// 邮箱密码登录示例
FirebaseAuth.getInstance().signInWithEmailAndPassword(email, password)
    .addOnCompleteListener { task ->
        if (task.isSuccessful) {
            // 登录成功，获取当前用户UID
            val currentUserUid = FirebaseAuth.getInstance().currentUser?.uid
            currentUserUid?.let { uid ->
                // 去获取用户角色
                fetchUserRole(uid)
            }
        } else {
            // 登录失败，提示用户
            Toast.makeText(this, "登录失败：${task.exception?.message}", Toast.LENGTH_SHORT).show()
        }
    }

Firestore获取角色并跳转页面： #

private fun fetchUserRole(uid: String) {
    val db = FirebaseFirestore.getInstance()
    db.collection("users").document(uid)
        .get()
        .addOnSuccessListener { document ->
            if (document.exists()) {
                val role = document.getString("role")
                when (role) {
                    "manager" -> {
                        // 跳转到管理者主页
                        startActivity(Intent(this, ManagerHomeActivity::class.java))
                        finish()
                    }
                    "employee" -> {
                        // 跳转到员工主页
                        startActivity(Intent(this, EmployeeHomeActivity::class.java))
                        finish()
                    }
                    else -> {
                        // 角色未设置，比如刚注册的用户，默认设为员工或者提示
                        Toast.makeText(this, "角色未设置，将默认设为员工", Toast.LENGTH_SHORT).show()
                        setDefaultRole(uid)
                    }
                }
            } else {
                // 用户文档不存在，初始化用户信息
                initUserDocument(uid)
            }
        }
        .addOnFailureListener { e ->
            Toast.makeText(this, "获取角色失败：${e.message}", Toast.LENGTH_SHORT).show()
        }
}

// 给新用户设置默认角色（员工）
private fun setDefaultRole(uid: String) {
    val db = FirebaseFirestore.getInstance()
    db.collection("users").document(uid)
        .set(mapOf("role" to "employee", "email" to FirebaseAuth.getInstance().currentUser?.email))
        .addOnSuccessListener {
            startActivity(Intent(this, EmployeeHomeActivity::class.java))
            finish()
        }
}

实时数据库获取角色并跳转： #

private fun fetchUserRole(uid: String) {
    val database = FirebaseDatabase.getInstance()
    database.getReference("users").child(uid).child("role")
        .addListenerForSingleValueEvent(object : ValueEventListener {
            override fun onDataChange(snapshot: DataSnapshot) {
                val role = snapshot.getValue(String::class.java)
                when (role) {
                    "manager" -> {
                        startActivity(Intent(this@LoginActivity, ManagerHomeActivity::class.java))
                        finish()
                    }
                    "employee" -> {
                        startActivity(Intent(this@LoginActivity, EmployeeHomeActivity::class.java))
                        finish()
                    }
                    else -> {
                        setDefaultRole(uid)
                    }
                }
            }

            override fun onCancelled(error: DatabaseError) {
                Toast.makeText(this@LoginActivity, "获取角色失败：${error.message}", Toast.LENGTH_SHORT).show()
            }
        })
}

private fun setDefaultRole(uid: String) {
    val database = FirebaseDatabase.getInstance()
    val userData = mapOf(
        "role" to "employee",
        "email" to FirebaseAuth.getInstance().currentUser?.email
    )
    database.getReference("users").child(uid).setValue(userData)
        .addOnSuccessListener {
            startActivity(Intent(this, EmployeeHomeActivity::class.java))
            finish()
        }
}

一定要配置Firebase的安全规则，不然普通用户可能能修改自己或别人的角色！

Firestore安全规则： #

rules_version = '2';
service cloud.firestore {
  match /databases/{database}/documents {
    // 用户只能读取自己的角色信息
    match /users/{userId} {
      allow read: if request.auth != null && request.auth.uid == userId;
      // 只有管理者才能修改用户角色
      allow write: if request.auth != null && get(/databases/$(database)/documents/users/$(request.auth.uid)).data.role == "manager";
    }
  }
}

实时数据库安全规则： #

{
  "rules": {
    "users": {
      "$uid": {
        // 用户只能读自己的信息
        ".read": "auth != null && auth.uid == $uid",
        // 只有管理者才能写用户信息
        ".write": "auth != null && root.child('users').child(auth.uid).child('role').val() == 'manager'"
      }
    }
  }
}

如果你的应用需要频繁验证角色，可以把角色存在Firebase Auth的自定义Claims里，这样不需要每次查数据库，性能更好：

管理员端设置自定义Claims： #

val targetUserUid = "用户UID"
FirebaseAuth.getInstance().getUser(targetUserUid)
    .addOnSuccessListener { user ->
        // 设置管理者角色的Claims
        FirebaseAuth.getInstance().setCustomUserClaims(user.uid, mapOf("role" to "manager"))
            .addOnSuccessListener {
                Toast.makeText(this, "Claims设置成功", Toast.LENGTH_SHORT).show()
            }
    }

登录后获取Claims并验证： #

FirebaseAuth.getInstance().currentUser?.getIdToken(true)
    ?.addOnSuccessListener { result ->
        val claims = result.claims
        val role = claims["role"] as? String
        when(role) {
            "manager" -> startActivity(Intent(this, ManagerHomeActivity::class.java))
            "employee" -> startActivity(Intent(this, EmployeeHomeActivity::class.java))
            else -> setDefaultRole(FirebaseAuth.getInstance().currentUser!!.uid)
        }
    }

这样每次登录后直接从Token里拿角色，速度更快～

内容的提问来源于stack exchange，提问作者Maharshi patil