001 /* 002 * Copyright (C) 2012 eXo Platform SAS. 003 * 004 * This is free software; you can redistribute it and/or modify it 005 * under the terms of the GNU Lesser General Public License as 006 * published by the Free Software Foundation; either version 2.1 of 007 * the License, or (at your option) any later version. 008 * 009 * This software is distributed in the hope that it will be useful, 010 * but WITHOUT ANY WARRANTY; without even the implied warranty of 011 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 012 * Lesser General Public License for more details. 013 * 014 * You should have received a copy of the GNU Lesser General Public 015 * License along with this software; if not, write to the Free 016 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 017 * 02110-1301 USA, or see the FSF site: http://www.fsf.org. 018 */ 019 020 package org.crsh.shell.impl.async; 021 022 import org.crsh.cli.impl.completion.CompletionMatch; 023 import org.crsh.shell.Shell; 024 025 import java.io.Closeable; 026 import java.util.Collections; 027 import java.util.HashSet; 028 import java.util.Set; 029 import java.util.concurrent.ExecutorService; 030 031 public class AsyncShell implements Shell, Closeable { 032 033 /** . */ 034 final Shell shell; 035 036 /** . */ 037 private AsyncProcess current; 038 039 /** . */ 040 final ExecutorService executor; 041 042 /** . */ 043 boolean closed; 044 045 /** . */ 046 final Object lock = new Object(); 047 048 /** . */ 049 final Set<AsyncProcess> processes; 050 051 public AsyncShell(ExecutorService executor, Shell shell) { 052 this.shell = shell; 053 this.current = null; 054 this.executor = executor; 055 this.closed = false; 056 this.processes = Collections.synchronizedSet(new HashSet<AsyncProcess>()); 057 } 058 059 public void close() { 060 061 AsyncProcess[] toCancel = null; 062 synchronized (lock) { 063 if (closed) { 064 toCancel = null; 065 } else { 066 closed = true; 067 toCancel = processes.toArray(new AsyncProcess[processes.size()]); 068 } 069 } 070 071 // Cancel all process 072 if (toCancel != null) { 073 for (AsyncProcess process : toCancel) { 074 process.cancel(); 075 } 076 } 077 } 078 079 // Shell implementation ********************************************************************************************** 080 081 public String getWelcome() { 082 return shell.getWelcome(); 083 } 084 085 public String getPrompt() { 086 return shell.getPrompt(); 087 } 088 089 public CompletionMatch complete(String prefix) { 090 return shell.complete(prefix); 091 } 092 093 public AsyncProcess createProcess(String request) { 094 synchronized (lock) { 095 if (closed) { 096 throw new IllegalStateException(); 097 } 098 } 099 return new AsyncProcess(this, request); 100 } 101 }