org.beangle.security.session.DefaultSession.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of beangle-security Show documentation
The Beangle Data Library
The newest version!
/*
 * Copyright (C) 2005, The Beangle Software.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published
 * by the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see .
 */

package org.beangle.security.session

import java.io.{ObjectInput, ObjectOutput}
import java.security.Principal
import java.time.Instant

import org.beangle.security.authc.DefaultAccount

object DefaultSessionBuilder extends SessionBuilder {
  def build(id: String, principal: Principal, loginAt: Instant, agent: Session.Agent, ttiSeconds: Int): Session = {
    new DefaultSession(id, principal.asInstanceOf[DefaultAccount], loginAt, agent, ttiSeconds)
  }
}

class DefaultSession extends Session {
  var id: String = _
  var principal: DefaultAccount = _
  var loginAt: Instant = _
  var lastAccessAt: Instant = _
  var agent: Session.Agent = _
  var ttiSeconds: Int = _

  def this(id: String, principal: DefaultAccount, loginAt: Instant, agent: Session.Agent, ttiSeconds: Int) = {
    this()
    this.id = id
    this.principal = principal
    this.loginAt = loginAt
    this.lastAccessAt = loginAt
    this.agent = agent
    this.ttiSeconds = ttiSeconds
  }

  def writeExternal(out: ObjectOutput): Unit = {
    out.writeObject(id)
    principal.writeExternal(out)
    out.writeLong(loginAt.getEpochSecond)
    out.writeLong(lastAccessAt.getEpochSecond)
    out.writeObject(agent.name)
    out.writeObject(agent.ip)
    out.writeObject(agent.os)
    out.writeInt(ttiSeconds)
  }

  def readExternal(in: ObjectInput): Unit = {
    id = readString(in)
    principal = new DefaultAccount()
    principal.readExternal(in)
    loginAt = Instant.ofEpochSecond(in.readLong)
    lastAccessAt = Instant.ofEpochSecond(in.readLong)
    agent = new Session.Agent(readString(in), readString(in), readString(in))
    ttiSeconds = in.readInt
  }

  @inline
  private def readString(in: ObjectInput): String = {
    in.readObject.asInstanceOf[String]
  }

  override def expired: Boolean = {
    lastAccessAt.plusSeconds(ttiSeconds).isBefore(Instant.now)
  }

  override def access(accessAt: Instant): Long = {
    val expired = lastAccessAt.plusSeconds(ttiSeconds).isBefore(accessAt)
    if (expired) {
      -1
    } else {
      var elapse = accessAt.getEpochSecond - this.lastAccessAt.getEpochSecond
      if (elapse < 0) {
        elapse = 0
      }else{
        lastAccessAt = accessAt
      }
      elapse
    }
  }
}