HJ_openHarmony_wms/entry/src/main/ets/common/util/UserAuthModel.ets

import relationalStore from '@ohos.data.relationalStore';
import CommonConstants from '../../common/constants/CommonConstants';
import {UserAuthInfo } from '../../viewmodel/wms/UserInfo';

class UserModel {

  private rdbStore: relationalStore.RdbStore |undefined=undefined
  private tableName: string = 'USER'

  /**
   * 初始化表
   */
  async initTaskDB(context:Context){
    // 1.rdb配置
    const config = {
      name: CommonConstants.RDB_NAME,
      securityLevel: relationalStore.SecurityLevel.S1
    }as relationalStore.StoreConfig
    try {
      // 获取数据库存储对象
      this.rdbStore = await relationalStore.getRdbStore(context,config);
    } catch (err) {
      console.info(`getRdbStore err ${JSON.stringify(err)}`);
    }
    // 2.初始化SQL语句
    const sql = `CREATE TABLE IF NOT EXISTS USER (
                  ID INTEGER PRIMARY KEY AUTOINCREMENT,
                  USER_ID INTEGER NOT NULL,
                  USER_NAME VARCHAR NOT NULL,
                  PASSWORD VARCHAR NOT NULL,
                  ORG_ID INTEGER NOT NULL,
                  STATION_ID INTEGER NOT NULL,
                  STATION_NAME VARCHAR NULL,
                  STATION_DICE_VALUE VARCHAR NULL,
                  AVATAR VARCHAR NULL,
                  MAINTAIN_LOGIN_STATUS INTEGER NOT NULL,
                  UPDATE_TIME INTEGER NOT NULL
                 )`
    // 3.获取rdb
    relationalStore.getRdbStore(context, config, (err, rdbStore) => {
      if(err){
        console.log('testTag', '获取rdbStore失败！')
        return
      }
      // 执行Sql
      rdbStore.executeSql(sql).then(()=>{
        console.log('testTag', '创建用户设置表成功！')
      }).catch((err: object)=>{
        console.log('testTag', '创建失败')
      })
      // 保存rdbStore
      this.rdbStore = rdbStore
    })
  }

  // 查询最后一次用户登录的数据
  async getLast(): Promise<UserAuthInfo | undefined> {
    // 1.构建查询条件
    let predicates = new relationalStore.RdbPredicates(this.tableName)
    predicates.orderByDesc('UPDATE_TIME')
    predicates.limitAs(1)
    // 2.查询
    let result = await this.rdbStore?.query(predicates, ['ID', 'USER_ID', 'USER_NAME', 'PASSWORD', 'ORG_ID', 'STATION_ID', 'STATION_NAME', 'STATION_DICE_VALUE', 'AVATAR', 'MAINTAIN_LOGIN_STATUS', 'UPDATE_TIME'])
    // 3.2.遍历封装 只取第一条
    while(!result?.isAtLastRow){
      // 3.3.指针移动到下一行
      result?.goToNextRow()
      // 3.4.获取数据
      let id = result?.getLong(result?.getColumnIndex('ID'))
      let userId = result?.getLong(result?.getColumnIndex('USER_ID'))
      let userName = result?.getString(result?.getColumnIndex('USER_NAME'))
      let password = result?.getString(result?.getColumnIndex('PASSWORD'))
      let orgId = result?.getLong(result?.getColumnIndex('ORG_ID'))
      let stationId = result?.getLong(result?.getColumnIndex('STATION_ID'))
      let stationName = result?.getString(result?.getColumnIndex('STATION_NAME'))
      let stationDictValue = result?.getString(result?.getColumnIndex('STATION_DICE_VALUE'))
      let avatar = result?.getString(result?.getColumnIndex('AVATAR'))
      let maintainLoginStatus = result?.getLong(result?.getColumnIndex('MAINTAIN_LOGIN_STATUS'))
      let updateTime = result?.getLong(result?.getColumnIndex('UPDATE_TIME'))
      return {id: id, userId: userId, userName: userName, password: password, orgId: orgId, stationId: stationId, stationName: stationName, stationDictValue: stationDictValue, avatar: avatar, maintainLoginStatus: maintainLoginStatus, updateTime: updateTime}
    }
    return
  }

  // 查询最后一次用户登录的数据
  async getByUserId(userId: number): Promise<UserAuthInfo | undefined> {
    if (!userId) {
      return
    }
    // 1.构建查询条件
    let predicates = new relationalStore.RdbPredicates(this.tableName)
    predicates.equalTo('USER_ID', userId)
    predicates.orderByDesc('UPDATE_TIME')
    predicates.limitAs(1)
    // 2.查询
    let result: relationalStore.ResultSet | undefined = await this.rdbStore?.query(predicates, ['ID', 'USER_ID', 'USER_NAME', 'PASSWORD', 'ORG_ID', 'STATION_ID', 'STATION_NAME', 'STATION_DICE_VALUE', 'AVATAR', 'MAINTAIN_LOGIN_STATUS', 'UPDATE_TIME'])
    // 3.2.遍历封装 只取第一条
    while(result && !result?.isAtLastRow){
      // 3.3.指针移动到下一行
      result.goToNextRow()
      // 3.4.获取数据
      let id = result.getLong(result.getColumnIndex('ID'))
      let userId = result.getLong(result.getColumnIndex('USER_ID'))
      let userName = result.getString(result.getColumnIndex('USER_NAME'))
      let password = result.getString(result.getColumnIndex('PASSWORD'))
      let orgId = result.getLong(result.getColumnIndex('ORG_ID'))
      let stationId = result.getLong(result.getColumnIndex('STATION_ID'))
      let stationName = result?.getString(result?.getColumnIndex('STATION_NAME'))
      let stationDictValue = result?.getString(result?.getColumnIndex('STATION_DICE_VALUE'))
      let avatar = result.getString(result.getColumnIndex('AVATAR'))
      let maintainLoginStatus = result.getLong(result.getColumnIndex('MAINTAIN_LOGIN_STATUS'))
      let updateTime = result.getLong(result.getColumnIndex('UPDATE_TIME'))
      return {id: id, userId: userId, userName: userName, password: password, orgId: orgId, stationId: stationId, stationName: stationName, stationDictValue: stationDictValue, avatar: avatar, maintainLoginStatus: maintainLoginStatus, updateTime: updateTime}
    }
    return
  }

  /**
   * 添加用户设置
   * @param user 用户信息
   * @returns 用户信息id
   */
  addUser(user: UserAuthInfo): Promise<number> | undefined {
    return this.rdbStore?.insert(this.tableName, {USER_ID: user.userId, USER_NAME: user.userName, PASSWORD: user.password, ORG_ID: user.orgId, STATION_ID: user.stationId, STATION_NAME: user.stationName, STATION_DICE_VALUE: user.stationDictValue, AVATAR: user.avatar, MAINTAIN_LOGIN_STATUS: user.maintainLoginStatus, UPDATE_TIME: user.updateTime} as relationalStore.ValuesBucket)
  }

  /**
   * 根据id更新用户设置状态
   * @param id 任务id
   * @param finished 任务是否完成
   */
  updateUser(user: UserAuthInfo) {

    // 1.要更新的数据
    let data: relationalStore.ValuesBucket = {USER_ID: user.userId as number, USER_NAME: user.userName as string, PASSWORD: user.password as string, ORG_ID: user.orgId as number, STATION_ID: user.stationId as number, STATION_NAME: user.stationName as string, STATION_DICE_VALUE: user.stationDictValue as string, AVATAR: user.avatar as string, MAINTAIN_LOGIN_STATUS: user.maintainLoginStatus as number, UPDATE_TIME: user.updateTime as number}
    // 2.更新的条件
    let predicates = new relationalStore.RdbPredicates(this.tableName)
    predicates.equalTo('ID', user.id)
    // 3.更新操作
    return this.rdbStore?.update(data, predicates)
  }

  /**
   * 根据id删除任务
   * @param id 任务id
   */
  deleteUserById(id: number){
    // 1.删除的条件
    let predicates = new relationalStore.RdbPredicates(this.tableName)
    predicates.equalTo('ID', id)
    // 2.删除操作
    return this.rdbStore?.delete(predicates)
  }
}

let userModel = new UserModel();

export default userModel as UserModel;