北大青鸟第一单元项目 （小说管理系统）

注意：

   此项目使用DOM4j 解析XML文件 请自备DOM4j 并导入

小说类

package books;

import java.io.Serializable;

public class Book implements Serializable {
    private String booktype;            //小说类型
    private String bookname;           //小说名称
    private String bookauthor;         //小说作者
    private String bookintroduction;   //小说简介
    private String bookContent;      //小说内容
    
    
    
    public String getBooktype() {
        return booktype;
    }
    public void setBooktype(String booktype) {
        this.booktype = booktype;
    }
    public String getBookname() {
        return bookname;
    }
    public void setBookname(String bookname) {
        this.bookname = bookname;
    }
    public String getBookauthor() {
        return bookauthor;
    }
    public void setBookauthor(String bookauthor) {
        this.bookauthor = bookauthor;
    }
    public String getBookintroduction() {
        return bookintroduction;
    }
    public void setBookintroduction(String bookintroduction) {
        this.bookintroduction = bookintroduction;
    }
    public String getBookContent() {
        return bookContent;
    }
    public void setBookContent(String bookContent) {
        this.bookContent = bookContent;
    }
    
    

}

序列化传送对象

package books;

import java.io.File;
import java.io.Serializable;
import java.util.List;

/**
 * 序列化传送对象
 * @author Empty-heart
 *
 */
public class Datas implements Serializable {
    private String flag;               //功能块标记
    private String type;               //小说类型标记
    private Use use;                   //用户类  
    private Book book;                 //小说类
    private List list;                 //集合类
    private File file;                 //文件
    private String path;               //小说路径

    public String getPath() {
        return path;
    }

    public void setPath(String path) {
        this.path = path;
    }

    public File getFile() {
        return file;
    }

    public void setFile(File file) {
        this.file = file;
    }

    public List getList() {
        return list;
    }

    public void setList(List list) {
        this.list = list;
    }

    public String getType() {
        return type;
    }
    
    public void setType(String type) {
        this.type = type;
    }
    public Book getBook() {
        return book;
    }

    public void setBook(Book book) {
        this.book = book;
    }

    public Use getUse() {
        return use;
    }

    public void setUse(Use use) {
        this.use = use;
    }

    public String getFlag() {
        return flag;
    }

    public void setFlag(String flag) {
        this.flag = flag;
    }
 
}

客户端功能类

package books;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.net.Socket;
import java.util.List;

import javax.swing.Box.Filler;


public class scoket_data {
     Socket s = null;
     ObjectOutputStream objout = null;
     ObjectInputStream objin = null;
     Datas datas = null;
     
     /**
      * 创建Socket端点
      * 创建输入、输出流
      */
    public void initSocket(){
        try {
            s = new Socket("192.168.10.222", 1314);
            objout = new ObjectOutputStream(s.getOutputStream());
            objin = new ObjectInputStream(s.getInputStream());
            datas = new Datas();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
    
    /**
     * 关闭各种资源
     */
    public void closeSocket(){
        try {
            objin.close();
            objout.close();
            s.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    
    /**
     * 登陆功能
     */
    public Datas login(String usename, String password) {
        try {
            initSocket();
            datas.setFlag(Use.SYS_LOGIN);
            Use use = new Use();
            use.setUsename(usename);
            use.setPassword(password);
            datas.setUse(use);
            objout.writeObject(datas);
            datas = (Datas) objin.readObject();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }finally {
            closeSocket();
        }
        return datas;
    }

    /**
     * 注册功能
     * @param name
     * @param pass1
     * @return
     */
    public Datas registration(String name, String pass1) {
          try {
              initSocket();
              datas.setFlag(Use.SYS_REGISTRATION);
              Use use = new Use();
              use.setUsename(name);
              use.setPassword(pass1);
              datas.setUse(use);
            objout.writeObject(datas);
            datas = (Datas) objin.readObject();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }finally {
            closeSocket();
        }
        return datas;
    }

    /**
     * 查看小说
     * @return
     */
    public Datas book(String type) {
        try {
            initSocket();
            datas.setFlag(Use.SYS_BOOK);
            if(type.equals("wuxia")) {
            datas.setType(Use.SYS_WUXIABOOK);
            }else if(type.equals("yanqing")) {
                datas.setType(Use.SYS_YANQINGBOOK);
            }
            objout.writeObject(datas);
            datas = (Datas) objin.readObject();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }finally {
            closeSocket();
        }
        return datas;
    }

    /**
     * 下载小说
     * @param bookname
     */
    public Datas downbook(String type, String bookname) {
        try {
            initSocket();
            datas.setFlag(Use.SYS_DOWNLOAD);
            Book book = new Book();
            book.setBookname(bookname);
            book.setBooktype(type);
            datas.setBook(book);
            objout.writeObject(datas);
            datas = (Datas) objin.readObject();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }finally {
            closeSocket();
        }
        return datas;
    }

    /**
     * 上传小说
     * @param bookname
     * @param bookauthor
     * @param bookintroduction
     * @param path
     * @return
     */
    public Datas upbook(String bookname, String bookauthor,
            String bookintroduction, String path, String type) {
         
        try {
            FileInputStream fis = new FileInputStream(path);
            System.setIn(fis);
            BufferedReader buf = new BufferedReader(new InputStreamReader(System.in));
            String str1 = "";
            String str2 = null;
            while((str2 = buf.readLine()) != null){
                str1 = str1.concat(str2);
            }
            fis.close();
            initSocket();
            datas.setFlag(Use.SYS_UPLOAD);
            datas.setPath(path);
            Book book = new Book();
            book.setBookname(bookname);
            book.setBookauthor(bookauthor);
            book.setBookintroduction(bookintroduction);
            book.setBooktype(type);
            book.setBookContent(str1);
            datas.setBook(book);
            objout.writeObject(datas);
            datas = (Datas) objin.readObject();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }finally {
            closeSocket();
            
        }
        
        return datas;
    }

}

服务端功能类

package books;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.io.UnsupportedEncodingException;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.dom4j.io.OutputFormat;
import org.dom4j.io.SAXReader;
import org.dom4j.io.XMLWriter;



public class server_data {
    /**
     * 登陆功能
     * 建立document对象 读取配置文件
     */
    public int Login(String usename, String password) {
        int val = 0;     //返回值 
        SAXReader sax = new SAXReader();
        try {
            Document doc = sax.read(new File("user.xml"));
            List<?> list = doc.selectNodes("/uses/use");
            Iterator<?> iterator = list.iterator();
            while(iterator.hasNext()) {
                Element use = (Element) iterator.next();
                String name = use.attribute("usename").getValue();
                String pass = use.attribute("password").getValue();
                if(name.equals(usename) && pass.equals(password)) {        
                    val = 1;
                }
            }
            
        } catch (DocumentException e) {
            e.printStackTrace();
        }

        return val;
    }
/**
 * 注册功能
 * @param usename
 * @param password
 * @return
 */
    public int registration(String usename, String password) {
        int val = 0;
        try {
            SAXReader sax = new SAXReader();
            Document doc = sax.read(new File("user.xml"));
            List<?> listuse = doc.selectNodes("/uses/use");
            Iterator<?> iterator = listuse.iterator();
            while(iterator.hasNext()) {
                Element elementuse = (Element) iterator.next();
                String nameuse = elementuse.attribute("usename").getValue();
                if(!(nameuse.equals(usename))) {
                    List<?> listuses = doc.selectNodes("uses");
                    Element uses = (Element) listuses.get(0);
                    Element use = uses.addElement("use");
                    use.addAttribute("usename", usename);
                    use.addAttribute("password", password);

                    XMLWriter writer = new XMLWriter(new FileOutputStream(new File("user.xml")));
                    writer.write(doc);
                    writer.close();
                    val = 1;
                }
            }
            
        } catch (DocumentException e) {
            e.printStackTrace();
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return val;
    }
    
    /**
     * 武侠小说集合
     */
    public  List<Book> wuxiabooklist() {
        try {
            SAXReader sax = new SAXReader();
            Document doc = sax.read(new File("wuxiabooks.xml"));
            List<?> list = doc.selectNodes("/books/book");
            Iterator<?> iterator = list.iterator();
            List<Book> booklist = new ArrayList<Book>();
            while(iterator.hasNext()) {
                Book book = new Book();
                Element bookx = (Element) iterator.next();
                String bookname = bookx.attribute("bookname").getValue();
                String bookauthor = bookx.attribute("bookauthor").getValue();
                String bookintroduction = bookx.attribute("bookintroduction").getValue();
                String bookContent = bookx.attribute("bookContent").getValue();
                
                book.setBookname(bookname);
                book.setBookauthor(bookauthor);
                book.setBookintroduction(bookintroduction);
                book.setBookContent(bookContent);
                booklist.add(book);
            }
            return booklist;        
        } catch (DocumentException e) {
            e.printStackTrace();
        }
        return null;
        
    }
    
    /**
     * 言情小说集合
     * @return
     */
    public  List<Book> yanqingbooklist() {
        try {
            SAXReader sax = new SAXReader();
            Document doc = sax.read(new File("yanqingbooks.xml"));
            List<?> list = doc.selectNodes("/books/book");
            Iterator<?> iterator = list.iterator();
            List<Book> booklist = new ArrayList<Book>();
            while(iterator.hasNext()) {
                Book book = new Book();
                Element bookx = (Element) iterator.next();
                String bookname = bookx.attribute("bookname").getValue();
                String bookauthor = bookx.attribute("bookauthor").getValue();
                String bookintroduction = bookx.attribute("bookintroduction").getValue();
                String bookContent = bookx.attribute("bookContent").getValue();
                
                book.setBookname(bookname);
                book.setBookauthor(bookauthor);
                book.setBookintroduction(bookintroduction);
                book.setBookContent(bookContent);
                
                booklist.add(book);
            }
            
            return booklist;        
        } catch (DocumentException e) {
            e.printStackTrace();
        }
        return null;
        
    }
    public File downbook(String type ,String bookname) {
        try {
            File file = new File("F://downbook");
            file.mkdir();
            File filebook = new File(file, bookname + ".txt");
            filebook.createNewFile();
            List list = new ArrayList();
            if(type.equals("wuxia")) {
                list = wuxiabooklist();
            }else if(type.equals("yanqing")) {
                list = yanqingbooklist();
            }
            for(int i = 0; i<list.size(); i++) {
                Book book = (Book) list.get(i);
                if(bookname.equals(book.getBookname()));
                String bookContent = book.getBookContent();
                PrintWriter pw = new PrintWriter(filebook);
                pw.write(bookContent);
                pw.flush();
            }
            return file;
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return null;
    }
    
    /**
     * 长传小说
     * @param book
     * @param file
     * @return
     */
    public int upbook(Book book, String path) {
        int val = 0;
        SAXReader sax = new SAXReader();
        Document doc = null;
        File files = null;
            try {
/*                FileReader fr = new FileReader(new File(path));
                char[] buf = new char[1024];
                fr.read(buf);
                String by = buf.toString();
                buf.*/
                if(book.getBooktype().equals("wuxia")) {
                 doc = sax.read(new File("wuxiabooks.xml"));
                 files = new File("wuxiabooks.xml");
                }else if(book.getBooktype().equals("yanqing")) {
                 doc = sax.read(new File("yanqingbooks.xml"));
                 files = new File("yanqingbooks.xml");
                }
                List list = doc.selectNodes("/books");
                Element books = (Element) list.get(0);
                Element upbook = books.addElement("book");
                upbook.addAttribute("bookname", book.getBookname());
                upbook.addAttribute("bookauthor", book.getBookauthor());
                upbook.addAttribute("bookintroduction", book.getBookintroduction());
                upbook.addAttribute("bookContent", book.getBookContent());
                val = 1;
//                fr.close();
                
                XMLWriter writer =  null;//new XMLWriter(new FileWriter(files));
                OutputFormat format = OutputFormat.createPrettyPrint();
                format.setEncoding("GBK");
                writer = new XMLWriter(new OutputStreamWriter(new 
                        FileOutputStream(files), format.getEncoding()), format);
                         writer.write(doc);
//                writer.write(doc);
                writer.close();
            } catch (DocumentException e) {
                e.printStackTrace();
            } catch (FileNotFoundException e) {
                e.printStackTrace();
            } catch (IOException e) {
                e.printStackTrace();
            }
            
        
        return val;
    }
}

服务端

package books;

import java.io.File;
import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;
import java.util.List;

public class serversocket {

    /**
     * @param args
     * @throws IOException
     * @throws ClassNotFoundException
     */
    public static server_data server = new server_data();
    public static int val = 0; // 标记
    public static Datas datas = new Datas();
    public static void main(String[] args) {
        ServerSocket ss = null;
        ObjectInputStream objin = null;
        ObjectOutputStream objout= null;
        Socket s = null;
        try {
            ss = new ServerSocket(1314);
            while (true) {
                 s = ss.accept();
                System.out.println(s.getInetAddress().getHostAddress());
                
                 objin = new ObjectInputStream(s.getInputStream());
                 objout = new ObjectOutputStream(
                        s.getOutputStream());
                
                datas = (Datas) objin.readObject();
                if (datas.getFlag().equals(Use.SYS_LOGIN)) {
                    val = server.Login(datas.getUse().getUsename(), datas.getUse()
                            .getPassword());
                    if (val == 1) {
                        datas.setFlag(Use.SYS_OK);
                    } 
                    objout.writeObject(datas);
                }else if(datas.getFlag().equals(Use.SYS_REGISTRATION)) {
                    val = server.registration(datas.getUse().getUsename(), datas.getUse().getPassword());
                    if(val == 1) {
                        datas.setFlag(Use.SYS_OK);
                    }
                    objout.writeObject(datas);
                }else if(datas.getFlag().equals(Use.SYS_BOOK)) {
                    List list = new ArrayList();
                    if(datas.getType().equals(Use.SYS_WUXIABOOK)) {
                        list = server.wuxiabooklist();
                    }else if(datas.getType().equals(Use.SYS_YANQINGBOOK)){
                        list = server.yanqingbooklist();

                    }
                    datas.setList(list);
                    objout.writeObject(datas);
                }else if(datas.getFlag().equals(Use.SYS_DOWNLOAD)) {
                    File file = server.downbook(datas.getBook().getBooktype(), datas.getBook().getBookname());
                    datas.setFlag(Use.SYS_OK);
                    datas.setFile(file);
                    objout.writeObject(datas);
                }else if(datas.getFlag().equals(Use.SYS_UPLOAD)) {
                    int val = server.upbook(datas.getBook(), datas.getPath());
                    if(val == 1) {
                        datas.setFlag(Use.SYS_OK);
                    }
                    objout.writeObject(datas);
                }
            }     
        } catch (IOException e) {
            e.printStackTrace();
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }finally {
            try {
                s.close();
                ss.close();
                objin.close();
                objout.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}

客户端

package books;

import java.util.ArrayList;
import java.util.Scanner;



public class socket {

    /**
     * @param args
     * @throws IOException 
     * @throws UnknownHostException 
     */
    public static Scanner scanner = new Scanner(System.in);
    public static scoket_data socket = new scoket_data();
    public static Datas datas = new Datas();
    public static void main(String[] args) {
             

 
            UI();
            System.out.println("程序结束 ！");
        
    }

    public static void UI() {
            System.out.println("欢迎使用在线迷你TXT小说管理器");
            System.out.println("------------------------------------");
            System.out.println("1、登陆\n2、注册\n3、退出");
            System.out.println("------------------------------------");
            System.out.print("请选择：");
            int i =  scanner.nextInt();
            if(i == 1) {
                  log();
            }else if(i==2) {
                while(true) {
                    System.out.println("```````````````````````````````");
                    System.out.println("当前操作：注册用户");
                    System.out.println("```````````````````````````````");
                    System.out.print("请输入用户名：");
                    String name = scanner.next();
                    System.out.print("请输入密码：");
                    String pass1 = scanner.next();
                    System.out.print("确认密码：");
                    String pass2 = scanner.next();
                    if(pass1.equals(pass2)) {
                        datas = socket.registration(name, pass1);
                        if(datas.getFlag().equals(Use.SYS_OK)) {
                            System.out.println("用户注册成功，请登陆！");
                            log();
                            break;
                        }else {
                            System.out.println("用户已存在！");
                        }
                    }else {
                        System.out.println("两次密码不同！");
                    }
                }

                
            }
    }
    
    /**
     * 用户登陆界面
     */
    public static void log() {
        while(true) {
            System.out.println("```````````````````````````````");
            System.out.println("当前操作：用户登陆");
            System.out.println("```````````````````````````````");
            System.out.print("请输入用户名：");
            String name = scanner.next();
            System.out.print("请输入密码：");
            String password = scanner.next();
            datas = socket.login(name, password);
            if(datas.getFlag().equals(Use.SYS_OK)) {
                lookbook();
                break;
            }else {
                System.out.println("*************************************");
                System.out.println("登陆失败！");
                System.out.println("*************************************");
            }
       }
    }           
    /**
     * 登陆成功
     * 小说列表界面
     */
    public static void lookbook() {
        System.out.println("*************************************");
        System.out.println("登陆成功！");
        System.out.println("*************************************");
        System.out.println("-------------------------------------");
        System.out.println("0、返回上级菜单\n1、武侠\n2、言情");
        System.out.println("-------------------------------------");
        System.out.print("请选择：");
        int val = scanner.nextInt();
        if(val == 0) {
            UI();
        }else if(val == 1) {
            booklist("wuxia");
        }else if(val == 2) {
            booklist("yanqing");
        }
    }
    
    /**
     * 小说选择及查看
     */
    public static void booklist(String type) {
    System.out.println("--------------------武俠小说列表--------------------");
    System.out.println("序号\t名称\t\t作者\t\t简介");
    if(type.equals("wuxia")){
       datas = socket.book(type);
    }else if(type.equals("yanqing")) {
        datas = socket.book(type);
    }
    ArrayList list = (ArrayList) datas.getList();
    for(int i=0; i<list.size(); i++) {
        Book book =  (Book) list.get(i);
        System.out.println((i+1) + "\t" + book.getBookname() + "\t\t" +
        book.getBookauthor() + "\t\t" + book.getBookintroduction());
    }
    System.out.println("--------------------小说列表结束--------------------");
    System.out.print("阅读和下载请选择文件序号，上传TXT请输入-1，返回请输入0：");
    int v = scanner.nextInt();
    if(v == 0) {
        lookbook();
    }else if(v == -1) {
        upbook(type);
        }else if(v < (list.size()+1)){
            Book book = (Book) list.get(v-1);
            System.out.println(book.getBookContent() + "\n");
            System.out.println("......,省略内容，请下载阅读");
            System.out.print("继续显示列表请输入1，下载TXT请输入2：");
            int down = scanner.nextInt();
            if(down == 1) {
                booklist(type);
            }else if (down == 2) {
                datas = socket.downbook(type,book.getBookname());
                if(datas.getFlag().equals(Use.SYS_OK)) {
                    System.out.println("*************************************");
                    System.out.println("下载后的文件名是：" + book.getBookname() + ".txt");
                    System.out.println("下载后的路径是：" + datas.getFile().getPath());
                    System.out.println("文件下载结果：true");
                    System.out.println("*************************************");
                    System.out.print("返回请输入0：");
                    int y = scanner.nextInt();
                    if(y == 0) {
                        lookbook();
                    }
                }else {
                    System.out.println("服务器上的小说不存在，请联系管理员！");
                }
            }
       }   
       }
  
    
    /**
     * 上传小说
     */
    public static void upbook(String type) {
        System.out.print("请输入小说名：");
        String bookname = scanner.next();
        System.out.print("请输入作者：");
        String bookauthor = scanner.next();
        System.out.print("请输入简介：");
        String bookintroduction = scanner.next();
        System.out.print("请输入上传文件的txt(请注意路径用//):");
        String path = scanner.next();
        datas = socket.upbook(bookname, bookauthor, bookintroduction, path, type); 
        if(datas.getFlag().equals(Use.SYS_OK)) {
               System.out.println("*************************************");
               System.out.println("小说保存成功");
               System.out.println("*************************************");
               System.out.print("继续上传请输入1，返回输入0：");
               int n = scanner.nextInt();
              if(n == 1) {
                  upbook(type);
              }else if(n == 0) {
                  lookbook();
               }
       }
    } 
} 

用户类和标记常量

package books;

import java.io.Serializable;

public class Use implements Serializable {
    private String usename;
    private String password;
    
    public static final String SYS_LOGIN = "LOGIN";      //登陸
    public static final String SYS_REGISTRATION = "REGISTRATION";      //注册
    public static final String SYS_BOOK = "BOOK";      //看书
    public static final String SYS_WUXIABOOK = "WUXIABOOK";      //武侠小说    
    public static final String SYS_YANQINGBOOK = "YANQINGWUXIABOOK";      //武侠小说    
    public static final String SYS_UPLOAD = "UPLOAD";      //上传
    public static final String SYS_DOWNLOAD = "DOWNLOAD";      //下载
    public static final String SYS_OK = "OK";      //成功

    public String getUsename() {
        return usename;
    }
    public void setUsename(String usename) {
        this.usename = usename;
    }
    public String getPassword() {
        return password;
    }
    public void setPassword(String password) {
        this.password = password;
    }
    

}