1
我正在創建一個簡單的程序，允許上傳和下載到文件服務器。在服務器完全在線之前，我需要在上傳和下載之間進行選擇。事情是，我對DataInput/Output流等感到困惑，目前上傳選項不起作用。它會自己工作，但不是我在這裏編碼的方式。它在這個階段破壞我的頭！多個數據輸入/輸出流混淆
Server代碼：
import java.io.*;
import java.net.*;
import javax.swing.JOptionPane;
public class Server_Test {
public static void main(String a[]) throws Exception {
int timeoutsecs = 600;
int port = 4444;
Socket sock;
ServerSocket servsock = new ServerSocket(port, timeoutsecs);
while (true) {
// wait for the next client connection
sock = servsock.accept();
DataInputStream choice = new DataInputStream(sock.getInputStream());
if (choice.read() == 1){
// ****1*****
// Download
// Send I/O streams to the socket
DataOutputStream out = new DataOutputStream(sock.getOutputStream());
new Server_Test().sendFile(out);
}else if (choice.read() == 2){
// ****2****
//Upload
// Receive I/O streams from socket
DataInputStream in = new DataInputStream(sock.getInputStream());
new Server_Test().receiveFile(in);
}
// Close this connection, (not the overall // server socket)
sock.close();
}
}
public void sendFile(DataOutputStream os) throws Exception{
// String fileLoc = JOptionPane.showInputDialog(null, "What is the file location of the file you want to download?");
File file = new File("C:\\TestFile.txt");
FileInputStream fis = new FileInputStream(file);
byte [] mybytearray = new byte [(int)file.length()+1];
BufferedInputStream bis = new BufferedInputStream(fis);
bis.read(mybytearray,0,mybytearray.length);
System.out.println("Sending...");
os.write(mybytearray,0,mybytearray.length);
os.flush();
bis.close();
}
public void receiveFile(DataInputStream in) throws Exception{
int filesize=6022386;
int bytesRead;
int current = 0;
byte [] mybytearray = new byte [filesize];
FileOutputStream fos = new FileOutputStream("C:\\UploaTest.txt");
BufferedOutputStream bos = new BufferedOutputStream(fos);
bytesRead = in.read(mybytearray,0,mybytearray.length);
current = bytesRead;
do {
bytesRead =
in.read(mybytearray, current, (mybytearray.length-current));
if(bytesRead >= 0) current += bytesRead;
} while(bytesRead > -1);
bos.write(mybytearray, 0 , current);
bos.flush();
bos.close();
}
}
客戶端代碼：
import java.io.*;
import java.net.*;
import javax.swing.JOptionPane;
public class Client_Test {
public static void main(String a[]) throws Exception {
Socket sock;
// ServerSocket serverSocket = new ServerSocket(5556);
sock = new Socket("127.0.0.1", 4444);
System.out.println("Connecting...");
Integer choice = Integer.parseInt(JOptionPane.showInputDialog(null, "Please type 1 for download and 2 for upload."));
DataOutputStream outs = new DataOutputStream(sock.getOutputStream());
outs.write(choice);
//outs.flush();
//outs.close();
if(choice == 1){
// *****1*****
//Download
// Get I/O streams from the socket
DataInputStream is = new DataInputStream(sock.getInputStream());
// receive file
new Client_Test().downloadFile(is);
}else if(choice == 2){
//*****2****
//Upload
//Send I/O streams to the socket.
DataOutputStream out = new DataOutputStream(sock.getOutputStream());
new Client_Test().uploadFile(out);
}
sock.close();
}
public void downloadFile(DataInputStream in) throws Exception{
int filesize=6022386;
int bytesRead;
int current = 0;
byte [] mybytearray = new byte [filesize];
FileOutputStream fos = new FileOutputStream("C:\\DowloadTest.txt");
BufferedOutputStream bos = new BufferedOutputStream(fos);
bytesRead = in.read(mybytearray,0,mybytearray.length);
current = bytesRead;
do {
bytesRead =
in.read(mybytearray, current, (mybytearray.length-current));
if(bytesRead >= 0) current += bytesRead;
} while(bytesRead > -1);
bos.write(mybytearray, 0 , current);
bos.flush();
bos.close();
}
public void uploadFile (DataOutputStream out)throws Exception{
File file = new File("C:\\TestFile.txt");
FileInputStream fis = new FileInputStream(file);
byte [] mybytearray = new byte [(int)file.length()+1];
BufferedInputStream bis = new BufferedInputStream(fis);
bis.read(mybytearray,0,mybytearray.length);
System.out.println("Sending...");
out.write(mybytearray,0,mybytearray.length);
out.flush();
bis.close();
}
}