您的位置:首页 > 理论基础 > 计算机网络

Android:网络:文件断点上传

2016-01-04 10:40 363 查看
HTTP不支持文件断点上传,只能自己实现断点服务器。客户端和服务器之间需要自定义协议,当然基于TCP/IP协议。



public class MainActivity extends Activity {

    private EditText filenameText;

    private TextView resultView;

    private ProgressBar uploadbar;

    private UploadLogService service;

    private Handler handler = new Handler(){

        @Override

        public void handleMessage(Message msg) {

            uploadbar.setProgress(msg.getData().getInt("length"));

            float num = (float)uploadbar.getProgress() / (float)uploadbar.getMax();

            int result = (int)(num * 100);

            resultView.setText(result + "%");

            if(uploadbar.getProgress() == uploadbar.getMax()){

                Toast.makeText(MainActivity.this, R.string.success, 1).show();

            }

        }

    };

    

    @Override

    public void onCreate(Bundle savedInstanceState) {

        super.onCreate(savedInstanceState);

        setContentView(R.layout.main);

        

        service =  new UploadLogService(this);

        filenameText = (EditText)findViewById(R.id.filename);

        resultView = (TextView)findViewById(R.id.result);

        uploadbar = (ProgressBar)findViewById(R.id.uploadbar);

        Button button = (Button)findViewById(R.id.button);

        button.setOnClickListener(new View.OnClickListener() {    

            public void onClick(View v) {

                String filename = filenameText.getText().toString();

                if(Environment.getExternalStorageState().equals(Environment.MEDIA_MOUNTED)){

                    File file = new File(Environment.getExternalStorageDirectory(), filename);

                    if(file.exists()){

                        uploadbar.setMax((int)file.length());

                        uploadFile(file);

                    }else{

                        Toast.makeText(MainActivity.this, R.string.notexsit, 1).show();

                    }

                }else{

                    Toast.makeText(MainActivity.this, R.string.sdcarderror, 1).show();

                }

            }

        });

    }

    private void uploadFile(final File file) {

        new Thread(new Runnable() {    

            public void run() {

                try {
                    String sourceid = service.getBindId(file);

                    Socket socket = new Socket("192.168.1.100", 7878);

                    OutputStream outStream = socket.getOutputStream();

                    String head = "Content-Length="+ file.length() + ";filename="+ file.getName()

                        + ";sourceid="+(sourceid!=null ? sourceid : "")+"\r\n";

                    outStream.write(head.getBytes());

                    

                    PushbackInputStream inStream = new PushbackInputStream(socket.getInputStream());    

                    String response = StreamTool.readLine(inStream);

                    String[] items = response.split(";");

                    String responseSourceid = items[0].substring(items[0].indexOf("=")+1);

                    String position = items[1].substring(items[1].indexOf("=")+1);

                    if(sourceid==null){//如果是第一次上传文件,在数据库中不存在该文件所绑定的资源id
                        service.save(responseSourceid, file);

                    }

                    RandomAccessFile fileOutStream = new RandomAccessFile(file, "r");

                    fileOutStream.seek(Integer.valueOf(position));

                    byte[] buffer = new byte[1024];

                    int len = -1;

                    int length = Integer.valueOf(position);

                    while( (len = fileOutStream.read(buffer)) != -1){

                        outStream.write(buffer, 0, len);

                        length += len;//累加已经上传的数据长度

                        Message msg = new Message();

                        msg.getData().putInt("length", length);

                        handler.sendMessage(msg);

                    }

                    if(length == file.length()) service.delete(file);

                    fileOutStream.close();

                    outStream.close();

                    inStream.close();

                    socket.close();

                } catch (Exception e) {                    

                    Toast.makeText(MainActivity.this, R.string.error, 1).show();

                }

            }

        }).start();

    }

}

布局文件

<?xml version="1.0" encoding="utf-8"?>

<LinearLayout xmlns:android="http://schemas.android.com/apk/res/android"

    android:orientation="vertical"

    android:layout_width="fill_parent"

    android:layout_height="fill_parent"

    >

<TextView  

    android:layout_width="fill_parent"

    android:layout_height="wrap_content"

    android:text="@string/filename"

    />

    

    <EditText  

    android:layout_width="fill_parent"

    android:layout_height="wrap_content"

    android:text="QQWubiSetup.exe"

    android:id="@+id/filename"

    />

    

   <Button  

        android:layout_width="wrap_content"

        android:layout_height="wrap_content"

        android:text="@string/button"

        android:id="@+id/button"

        />

        

    <ProgressBar

        android:layout_width="fill_parent"

        android:layout_height="20px"

        style="?android:attr/progressBarStyleHorizontal"

        android:id="@+id/uploadbar"

        />

    <TextView  

        android:layout_width="fill_parent"

        android:layout_height="wrap_content"

        android:gravity="center"

        android:id="@+id/result"

        />    

</LinearLayout>

StreamTool

public class StreamTool {

    

     public static void save(File file, byte[] data) throws Exception {

         FileOutputStream outStream = new FileOutputStream(file);

         outStream.write(data);

         outStream.close();

     }

    

     public static String readLine(PushbackInputStream in) throws IOException {

            char buf[] = new char[128];

            int room = buf.length;

            int offset = 0;

            int c;

loop:        while (true) {

                switch (c = in.read()) {

                    case -1:

                    case '\n':

                        break loop;

                    case '\r':

                        int c2 = in.read();

                        if ((c2 != '\n') && (c2 != -1)) in.unread(c2);

                        break loop;

                    default:

                        if (--room < 0) {

                            char[] lineBuffer = buf;

                            buf = new char[offset + 128];

                            room = buf.length - offset - 1;

                            System.arraycopy(lineBuffer, 0, buf, 0, offset);

                           

                        }

                        buf[offset++] = (char) c;

                        break;

                }

            }

            if ((c == -1) && (offset == 0)) return null;

            return String.copyValueOf(buf, 0, offset);

    }

    

    /**

    * 读取流

    * @param inStream

    * @return 字节数组

    * @throws Exception

    */

    public static byte[] readStream(InputStream inStream) throws Exception{

            ByteArrayOutputStream outSteam = new ByteArrayOutputStream();

            byte[] buffer = new byte[1024];

            int len = -1;

            while( (len=inStream.read(buffer)) != -1){

                outSteam.write(buffer, 0, len);

            }

            outSteam.close();

            inStream.close();

            return outSteam.toByteArray();

    }

}

UploadLogService

public class UploadLogService {

    private DBOpenHelper dbOpenHelper;

    

    public UploadLogService(Context context){

        dbOpenHelper = new DBOpenHelper(context);

    }

    

    public String getBindId(File file){

        SQLiteDatabase db = dbOpenHelper.getReadableDatabase();

        Cursor cursor = db.rawQuery("select sourceid from uploadlog where path=?", new String[]{file.getAbsolutePath()});

        if(cursor.moveToFirst()){

            return cursor.getString(0);

        }

        return null;

    }

    

    public void save(String sourceid, File file){

        SQLiteDatabase db = dbOpenHelper.getWritableDatabase();

        db.execSQL("insert into uploadlog(path,sourceid) values(?,?)",

                new Object[]{file.getAbsolutePath(), sourceid});

    }

    

    public void delete(File file){

        SQLiteDatabase db = dbOpenHelper.getWritableDatabase();

        db.execSQL("delete from uploadlog where path=?", new Object[]{file.getAbsolutePath()});

    }

 }

DBOpenHelper

public class DBOpenHelper extends SQLiteOpenHelper {

    public DBOpenHelper(Context context) {

        super(context, "itcast.db", null, 1);

    }

    @Override

    public void onCreate(SQLiteDatabase db) {

        db.execSQL("CREATE TABLE IF NOT EXISTS uploadlog (_id integer primary key autoincrement, path varchar(20), sourceid varchar(20))");

    }

    @Override

    public void onUpgrade(SQLiteDatabase db, int oldVersion, int newVersion) {

        // TODO Auto-generated method stub

    }

}

服务端

public class FileServer {

    

     private ExecutorService executorService;//线程池

     private int port;//监听端口

     private boolean quit = false;//退出

     private ServerSocket server;

     private Map<Long, FileLog> datas = new HashMap<Long, FileLog>();//存放断点数据

    

     public FileServer(int port){

         this.port = port;

         //创建线程池,池中具有(cpu个数*50)条线程

         executorService = Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors() * 50);

     }

     /**

      * 退出

      */

     public void quit(){

        this.quit = true;

        try {

            server.close();

        } catch (IOException e) {

        }

     }

     /**

      * 启动服务

      * @throws Exception

      */

     public void start() throws Exception{

         server = new ServerSocket(port);

         while(!quit){

             try {

               Socket socket = server.accept();

               //为支持多用户并发访问,采用线程池管理每一个用户的连接请求

               executorService.execute(new SocketTask(socket));

             } catch (Exception e) {

               //  e.printStackTrace();

             }

         }

     }

    

     private final class SocketTask implements Runnable{

        private Socket socket = null;

        public SocketTask(Socket socket) {

            this.socket = socket;

        }

        

        public void run() {

            try {

                System.out.println("accepted connection "+ socket.getInetAddress()+ ":"+ socket.getPort());

                PushbackInputStream inStream = new PushbackInputStream(socket.getInputStream());

                //得到客户端发来的第一行协议数据:Content-Length=143253434;filename=xxx.3gp;sourceid=

                //如果用户初次上传文件,sourceid的值为空。

                String head = StreamTool.readLine(inStream);

                System.out.println(head);

                if(head!=null){

                    //下面从协议数据中提取各项参数值

                    String[] items = head.split(";");

                    String filelength = items[0].substring(items[0].indexOf("=")+1);

                    String filename = items[1].substring(items[1].indexOf("=")+1);

                    String sourceid = items[2].substring(items[2].indexOf("=")+1);        

                    long id = System.currentTimeMillis();//生产资源id,如果需要唯一性,可以采用UUID

                    FileLog log = null;

                    if(sourceid!=null && !"".equals(sourceid)){

                        id = Long.valueOf(sourceid);

                        log = find(id);//查找上传的文件是否存在上传记录

                    }

                    File file = null;

                    int position = 0;

                    if(log==null){//如果不存在上传记录,为文件添加跟踪记录

                        String path = new SimpleDateFormat("yyyy/MM/dd/HH/mm").format(new Date());

                        File dir = new File("file/"+ path);

                        if(!dir.exists()) dir.mkdirs();

                        file = new File(dir, filename);

                        if(file.exists()){//如果上传的文件发生重名,然后进行改名

                            filename = filename.substring(0, filename.indexOf(".")-1)+ dir.listFiles().length+ filename.substring(filename.indexOf("."));

                            file = new File(dir, filename);

                        }

                        save(id, file);

                    }else{// 如果存在上传记录,读取已经上传的数据长度

                        file = new File(log.getPath());//从上传记录中得到文件的路径

                        if(file.exists()){

                            File logFile = new File(file.getParentFile(), file.getName()+".log");

                            if(logFile.exists()){

                                Properties properties = new Properties();

                                properties.load(new FileInputStream(logFile));

                                position = Integer.valueOf(properties.getProperty("length"));//读取已经上传的数据长度

                            }

                        }

                    }

                    

                    OutputStream outStream = socket.getOutputStream();

                    String response = "sourceid="+ id+ ";position="+ position+ "\r\n";

                    //服务器收到客户端的请求信息后,给客户端返回响应信息:sourceid=1274773833264;position=0

                    //sourceid由服务器端生成,唯一标识上传的文件,position指示客户端从文件的什么位置开始上传

                    outStream.write(response.getBytes());

                    

                    RandomAccessFile fileOutStream = new RandomAccessFile(file, "rwd");

                    if(position==0) fileOutStream.setLength(Integer.valueOf(filelength));//设置文件长度

                    fileOutStream.seek(position);//指定从文件的特定位置开始写入数据

                    byte[] buffer = new byte[1024];

                    int len = -1;

                    int length = position;

                    while( (len=inStream.read(buffer)) != -1){//从输入流中读取数据写入到文件中

                        fileOutStream.write(buffer, 0, len);

                        length += len;

                        Properties properties = new Properties();

                        properties.put("length", String.valueOf(length));

                        FileOutputStream logFile = new FileOutputStream(new File(file.getParentFile(), file.getName()+".log"));

                        properties.store(logFile, null);//实时记录已经接收的文件长度

                        logFile.close();

                    }

                    if(length==fileOutStream.length()) delete(id);

                    fileOutStream.close();                    

                    inStream.close();

                    outStream.close();

                    file = null;

                    

                }

            } catch (Exception e) {

                e.printStackTrace();

            }finally{

                try {

                    if(socket!=null && !socket.isClosed()) socket.close();

                } catch (IOException e) {}

            }

        }

     }

    

     public FileLog find(Long sourceid){

         return datas.get(sourceid);

     }

     //保存上传记录

     public void save(Long id, File saveFile){

         //日后可以改成通过数据库存放

         datas.put(id, new FileLog(id, saveFile.getAbsolutePath()));

     }

     //当文件上传完毕,删除记录

     public void delete(long sourceid){

         if(datas.containsKey(sourceid)) datas.remove(sourceid);

     }

    

     private class FileLog{

        private Long id;

        private String path;

        public Long getId() {

            return id;

        }

        public void setId(Long id) {

            this.id = id;

        }

        public String getPath() {

            return path;

        }

        public void setPath(String path) {

            this.path = path;

        }

        public FileLog(Long id, String path) {

            this.id = id;

            this.path = path;

        }    

     }

}

JavaSE的客户端

public class SocketClient {

    /**

     * @param args

     */

    public static void main(String[] args) {

        try {

            Socket socket = new Socket("127.0.0.1", 7878);

            OutputStream outStream = socket.getOutputStream();            

            String filename = "QQWubiSetup.exe";

            File file = new File(filename);

            String head = "Content-Length="+ file.length() + ";filename="+ filename + ";sourceid=\r\n";

            outStream.write(head.getBytes());

            

            PushbackInputStream inStream = new PushbackInputStream(socket.getInputStream());    

            String response = StreamTool.readLine(inStream);

            System.out.println(response);

            String[] items = response.split(";");

            String position = items[1].substring(items[1].indexOf("=")+1);

            

            RandomAccessFile fileOutStream = new RandomAccessFile(file, "r");

            fileOutStream.seek(Integer.valueOf(position));

            byte[] buffer = new byte[1024];

            int len = -1;

            while( (len = fileOutStream.read(buffer)) != -1){

                outStream.write(buffer, 0, len);

            }

            fileOutStream.close();

            outStream.close();

            inStream.close();

            socket.close();

        } catch (Exception e) {                    

            e.printStackTrace();

        }

    }

    /**

    * 读取流

    * @param inStream

    * @return 字节数组

    * @throws Exception

    */

    public static byte[] readStream(InputStream inStream) throws Exception{

            ByteArrayOutputStream outSteam = new ByteArrayOutputStream();

            byte[] buffer = new byte[1024];

            int len = -1;

            while( (len=inStream.read(buffer)) != -1){

                outSteam.write(buffer, 0, len);

            }

            outSteam.close();

            inStream.close();

            return outSteam.toByteArray();

    }

}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: