七叶笔记 » java编程 » JPA动态查询语句(代码详解)

JPA动态查询语句(代码详解)

JPA动态查询语句 (代码详解)

我们现在在做一个OA系统,将新增的那些数据都写到数据库的时候是采用jpa规范的,(不太理解jpa的相关知识点,今天看下相关知识,然后再补充jpa的知识点),现在记录jpa中的动态查询语句,其实这些语句都是可以用sql语句写的,但是sql语句写得查询,删除,插入数据等操作不安全,所以采用jpa的语句。我们的项目是分为三层结构,第一层是实体层,在该层中专门定义某一实体的相关字段,它的set(),get()方法。第二层是服务层,将service和dao都放在一个组件中,在dao层中定义和数据库相关的操作方法,在service层中定义相关的业务逻辑层要调用的方法。第三层是restful层,在这层定义的是和前端交互的组件。

首先讲讲第一层:实体层

定义一个实体

/*** 邮件实体**/  @Entity  @Table(name = "mail_tbl")  public class InnerMails implements Serializable {  private static final long serialVersionUID = 4999674279957375152L;  @Id  @GeneratedValue  private long id;  private String subject;// 主题  private String toMails;// 收件人 格式 :姓名<userId>;姓名<userId>  private String urgency;// 紧急程度    @Column(name = "sendDate")  @Temporal(TemporalType.TIMESTAMP)  private Date sendDate;// 发布日期  private String content;// 邮件内容  // 原文附件  @OneToMany(cascade={ CascadeType.MERGE,CascadeType.REMOVE})  @JoinColumn(name = "mail_id")  @OrderBy(value = "id DESC")//注释指明加载Attachment时按id的降序排序private Set<AppendFile> appendFiles=new HashSet<AppendFile>();// 附件  private String mailUser;// 邮件拥有者 格式:userId  private String sendMail;// 邮件发送者 格式:姓名<userId>  private int type;// 状态标示:-1删除;0草稿;1发送;2未读收件,3已读收件  public long getId() {  return id;  }  public void setId(long id) {  this.id = id;  }  public String getSubject() {  return subject;  }  public void setSubject(String subject) {  this.subject = subject;  }  public String getToMails() {  return toMails;  }  public void setToMails(String toMails) {  this.toMails = toMails;  }  public String getUrgency() {  return urgency;  }  public void setUrgency(String urgency) {  this.urgency = urgency;  }  public Date getSendDate() {  return sendDate;  }  public void setSendDate(Date sendDate) {  this.sendDate = sendDate;  }  public String getContent() {  return content;  }  public void setContent(String content) {  this.content = content;  }  public String getMailUser() {  return mailUser;  }    public void setMailUser(String mailUser) {  this.mailUser = mailUser;  }  public Set<AppendFile> getAppendFiles() {  return appendFiles;  }  public void setAppendFiles(Set<AppendFile> appendFiles) {  this.appendFiles = appendFiles;  }  public String getSendMail() {  return sendMail;  }  public void setSendMail(String sendMail) {  this.sendMail = sendMail;  }  public int getType() {  return type;  }  public void setType(int type) {  this.type = type;  }  }

定义查询实体:

package com.gzydt.oa.commons;  import java.util.ArrayList;  import java.util.HashMap;  import java.util.List;  import java.util.Map;  /*** 分页查询参数*  * @author huangzhenwei* @since 2014-11-21*  */  public class QueryParam {  // 排序字段,以“+”、“-”符号连接排序字段名:“+key”表示 按“key”字段升序,“-key”表示按“key”字段降序。  private List<String> sorts = new ArrayList<String>();  // 起始记录下标,从0开始计算  private int first = 0;  // 每页最大记录数  private int max = 10;  // 是否分页标志  private boolean isPage = true;  // 查询参数  private Map<String, String> param = new HashMap<String, String>();  public QueryParam() {  }  public int getFirst() {  return first;  }  public void setFirst(int first) {  this.first = first;  }  public int getMax() {  return max;  }  public void setMax(int max) {  this.max = max;  }  public Map<String, String> getParam() {  return param;  }  public void setParam(Map<String, String> param) {  this.param = param;  }  public boolean isPage() {  return isPage;  }  public void setPage(boolean isPage) {  this.isPage = isPage;  }  public List<String> getSorts() {  return sorts;  }  public void setSorts(List<String> sorts) {  this.sorts = sorts;  }  }

第二层:服务层
dao层:定义和数据库相关操作的方法

package com.gzydt.oa.dao;    import java.util.List;    import com.gzydt.oa.commons.QueryParam;  import com.gzydt.oa.entity.AppendFile;  import com.gzydt.oa.entity.InnerMails;    /*** 邮件发送dao接口**/  public interface InnerMailDao {  /** * 保存邮件 * @param mail * @return */  public InnerMails save(InnerMails mail);  /** * 更新邮件 * @param mail * @return */  public InnerMails update(InnerMails mail);  /** * 删除邮件 * @param id */  public void delete(long id);  /** * 查询邮件 * @param queryParam * @return */  public List<InnerMails> getlist(QueryParam queryParam);  /** * 获取单个邮件 * @param id * @return */  public InnerMails get(long id);  /** * 获取满足条件的邮件的总数 * @param queryParam * @return */  public int getCount(QueryParam queryParam);     /*** 新增附件* @param id* @param appendFile*/  public void addAttach(long id,AppendFile appendFile);  }
package com.gzydt.oa.dao.impl;    import java.text.DateFormat;  import java.text.SimpleDateFormat;  import java.util.ArrayList;  import java.util.Date;  import java.util.HashSet;  import java.util.Iterator;  import java.util.List;  import java.util.Map;  import java.util.Set;    import javax.persistence.EntityManager;  import javax.persistence.TypedQuery;  import javax.persistence.criteria.CriteriaBuilder;  import javax.persistence.criteria.CriteriaQuery;  import javax.persistence.criteria.Predicate;  import javax.persistence.criteria.Root;    import com.gzydt.oa.commons.QueryParam;  import com.gzydt.oa.dao.InnerMailDao;  import com.gzydt.oa.entity.AppendFile;  import com.gzydt.oa.entity.InnerMails;    /*** 邮件实现类*/  public class InnerMailDaoImpl implements InnerMailDao{    private EntityManager entityManager;  public void setEntityManager(EntityManager entityManager) {  this.entityManager = entityManager;  }  /** * 保存邮件 * @param mail * @return */  @Override  public InnerMails save(InnerMails mail) {  try {  entityManager.persist(mail);  entityManager.flush();  return mail;  } catch ( Exception e ) {  e.printStackTrace();  return null;  }  }  /** * 更新邮件 * @param mail * @return */  @Override  public InnerMails update(InnerMails mail) {  try {  entityManager.merge(mail);  return mail;  } catch ( Exception e ) {  e.printStackTrace();  return null;  }  }  /** * 删除邮件 * @param id */  @Override  public void delete(long id) {    entityManager.createQuery("delete from  PhoneRecord e where e.id=:id").setParameter("id", id).executeUpdate();    }  /** * 查询邮件 * @param queryParam * @return */  @Override  public List<InnerMails> getlist(QueryParam queryParam) {  CriteriaBuilder cb = entityManager.getCriteriaBuilder();  CriteriaQuery<InnerMails> criteriaQuery = cb.createQuery(InnerMails.class);  Root<InnerMails> register = criteriaQuery.from(InnerMails.class);  // 过滤条件  Predicate[] predicates = createPredicate(queryParam, cb, register);  criteriaQuery.where(predicates);  int start = queryParam.getFirst();  int end = queryParam.getMax();  TypedQuery<InnerMails> typedQuery = entityManager.createQuery(criteriaQuery);  typedQuery.setFirstResult(start).setMaxResults(end);  return typedQuery.getResultList();  }    //设置查询条件  private Predicate[] createPredicate(QueryParam queryParam, CriteriaBuilder cb, Root<InnerMails> entity) {  List<Predicate> predicates=new ArrayList<Predicate>();  //取出查询条件  Map<String, String> param= queryParam.getParam();  for(Map.Entry entry:param.entrySet()){  String key=entry.getKey().toString();  String value=entry.getValue().toString();  if(key.equals("sendDate")){     Predicate conditionTime = createOperateTime(key,cb,value,entity);     if(null!=conditionTime){     predicates.add(conditionTime);     }}else{  predicates.add(cb.like(entity.<String> get(key),"%"+value+"%"));  }  }  return predicates.toArray(new Predicate[0]);  }    /** * 将时间作为查询条件的方法 * @param cb * @param value * @param entity */  private Predicate createOperateTime(String key,CriteriaBuilder cb, String value, Root<InnerMails> entity) {if(null == value){  return null;  }  String[] operateTime=value.split("~");  if(operateTime.length!=2){  return null;  }  try {  DateFormat df=new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");//格式一定要写正确,  Date t1=df.parse(operateTime[0] + " 00:00:00");  Date t2=df.parse(operateTime[1] + " 23:59:59");  return cb.between(entity.<Date> get(key), t1, t2);  } catch ( Exception e ) {     e.printStackTrace();  }  return null;    }    /** * 获取单个邮件 * @param id * @return */  @Override  public InnerMails get(long id) {  InnerMails innersMails=entityManager.find(InnerMails.class, id);     Iterator<AppendFile> iterator=innersMails.getAppendFiles().iterator();  Set<AppendFile> attachs=new HashSet<AppendFile>();  while(iterator.hasNext()){  AppendFile appendFile=new AppendFile();  appendFile=iterator.next();  attachs.add(appendFile);}  innersMails.setAppendFiles(attachs);     return innersMails;  }  /** * 获取满足条件的邮件的总数 * @param queryParam * @return */  @Override  public int getCount(QueryParam queryParam) {  CriteriaBuilder cb = entityManager.getCriteriaBuilder();  CriteriaQuery<Long> criteriaQuery = cb.createQuery(Long.class);  Root<InnerMails> mails = criteriaQuery.from(InnerMails.class);  criteriaQuery.select(cb.countDistinct(mails));  // 过滤条件  Predicate[] predeicates = createPredicate(queryParam, cb, mails);  criteriaQuery.where(predeicates);  TypedQuery<Long> typedQuery = entityManager.createQuery(criteriaQuery);  int count = 0;  try {  count = typedQuery.getSingleResult().intValue();  } catch ( Exception e ) {  e.printStackTrace();  }  return count;  }  /** * 新增附件 * @param id * @param appendFile */  @Override  public void addAttach(long id, AppendFile appendFile) {  InnerMails entity=this.get(id);  entity.getAppendFiles().add(appendFile);  entityManager.merge(entity);    }    }

动态查询语句的相关知识

1:查询User表中的字段adminlever的小于给定值的数据

第一种写法:(安全,推荐使用这种)

/*** 查询某一级别以上的用户*/     @Override     public List<User> getOnLeverUper(int lever) {   CriteriaBuilder cb =entityManager.getCriteriaBuilder();   CriteriaQuery<User> criterQuery=cb.createQuery(User.class);   Root<User> entity=criterQuery.from(User.class);   Path<Integer> adminLever=entity.<Integer> get("adminlever") ;   criterQuery.where(cb.lessThan(adminLever, lever));   TypedQuery<User> typedQuery=entityManager.createQuery(criterQuery);   return typedQuery.getResultList();     }

第二种写法:(不太安全,)

/*** 查询某一级别以上的用户*/     @Override     public List<User> getOnLeverUper(int lever) {     List<User> users=entityManager.createQuery("from User u where u.adminlever<:adminlever")     .setParameter("adminlever",lever).getResultList();  return users;   }

第二种删除数据(有时候会由于某实体和另一实体设置了一对一或者多对一,或者多对多的关系而导致不能正常删除数据),解决方法:

/** * 删除登记信息 *   * @param id *登记编号 */  @Override  public void handleDelete(long id) throws Exception {  ReceiptEntity entity = entityManager.find(ReceiptEntity.class, id);  entityManager.remove(entity);  //下面的的方法删除应为存在外键关联会删除失败     /*entityManager.createQuery("delete from  ReceiptEntity e where e.id=:id").setParameter("id", id).executeUpdate();*/  }

service层:接口

package com.gzydt.oa.service;    import java.util.List;    import com.gzydt.oa.commons.QueryParam;  import com.gzydt.oa.entity.AppendFile;  import com.gzydt.oa.entity.InnerMails;    /*** 内部邮件接口**/  public interface InnerMailService {  /** * 保存邮件 * @param mail * @return */  public InnerMails save(InnerMails mail);  /** * 更新邮件 * @param mail * @return */  public InnerMails update(InnerMails mail);  /** * 删除邮件 * @param id */  public void delete(long id);  /** * 查询邮件 * @param queryParam * @return */  public List<InnerMails> getlist(QueryParam queryParam);  /** * 获取单个邮件 * @param id * @return */  public InnerMails get(long id);  /** * 获取满足条件的邮件的总数 * @param queryParam * @return */  public int getCount(QueryParam queryParam);  /** * 发邮件 * @param content *//*public void sendMail(String content);*/  /** * 新增附件 * @param id * @param appendFile */  public void addAttach(long id,AppendFile appendFile);    }

service层:实现类

package com.gzydt.oa.service.impl;    import java.util.List;    import com.gzydt.oa.commons.QueryParam;  import com.gzydt.oa.dao.InnerMailDao;  import com.gzydt.oa.entity.AppendFile;  import com.gzydt.oa.entity.InnerMails;  import com.gzydt.oa.service.InnerMailService;    /*** 内部邮件服务类**/  public class InnerMailServiceImpl implements InnerMailService{    private InnerMailDao mailDao;    public void setMailDao(InnerMailDao mailDao) {  this.mailDao = mailDao;  }  /** * 保存邮件 * @param mail * @return */  @Override  public InnerMails save(InnerMails mail) {  return mailDao.save(mail);  }    @Override  public InnerMails update(InnerMails mail) {  return mailDao.update(mail);  }  /** * 删除邮件 * @param id */  @Override  public void delete(long id) {     mailDao.delete(id);    }  /** * 查询邮件 * @param queryParam * @return */  @Override  public List<InnerMails> getlist(QueryParam queryParam) {     return mailDao.getlist(queryParam);  }  /** * 获取单个邮件 * @param id * @return */  @Override  public InnerMails get(long id) {     return mailDao.get(id);  }  /** * 获取满足条件的邮件的总数 * @param queryParam * @return */  @Override  public int getCount(QueryParam queryParam) {  return mailDao.getCount(queryParam);  }       /* @Overridepublic void sendMail(String content) {}*/  /** * 新增附件 * @param id * @param appendFile */  @Override  public void addAttach(long id, AppendFile appendFile) {     mailDao.addAttach(id, appendFile);    }    }

在服务层中定义相关的服务配置

<?xml version="1.0" encoding="UTF-8"?>  <blueprint default-activation="eager"  xmlns="http://www.osgi.org/xmlns/blueprint/v1.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"  xmlns:jpa="http://aries.apache.org/xmlns/jpa/v1.0.0" xmlns:tx="http://aries.apache.org/xmlns/transactions/v1.0.0"  xsi:schemaLocation="    http://www.osgi.org/xmlns/blueprint/v1.0.0 http://www.osgi.org/xmlns/blueprint/v1.0.0/blueprint.xsd    http://cxf.apache.org/blueprint/jaxrs http://cxf.apache.org/schemas/blueprint/jaxrs.xsd    http://cxf.apache.org/blueprint/core http://cxf.apache.org/schemas/blueprint/core.xsd">    <!-- This gets the container-managed EntityManager and injects it into the  ServiceImpl bean. -->    <!-- dao -->  <bean id="mailDao" class="com.gzydt.oa.dao.impl.InnerMailDaoImpl">  <jpa:context unitname="com.gzydt.jpa.persistence"  property="entityManager" />  <tx:transaction method="*" value="Required" />  </bean>    <!--新增结束 -->      <!-- bean -->    <bean id="mailService" class="com.gzydt.oa.service.impl.InnerMailServiceImpl">  <property name="mailDao" ref="mailDao" />  </bean>      <!--新增结束 -->    <!-- service -->  <service ref="mailService" interface="com.gzydt.oa.service.InnerMailService" />     <!-- This bundle makes use of Karaf commands to demonstrate core persistence  operations. Feel free to remove it. -->  <command-bundle xmlns="http://karaf.apache.org/xmlns/shell/v1.1.0">  <command name="msg/add">  <action class="com.gzydt.oa.command.AddMessage">  <property name="messageService" ref="messageService" />  </action>  </command>  <command name="msg/list">  <action class="com.gzydt.oa.command.GetMessage">  <property name="messageService" ref="messageService" />  </action>  </command>  <command name="msg/delete">  <action class="com.gzydt.oa.command.DeleteMessage">  <property name="messageService" ref="messageService" />  </action>  </command>    <command name="dept/add">  <action class="com.gzydt.oa.command.DeptAddCommand">  <property name="deptService" ref="deptService" />  </action>  </command>  <command name="dept/list">  <action class="com.gzydt.oa.command.DeptGetCommand">  <property name="deptService" ref="deptService" />  </action>  </command>  <command name="dept/delete">  <action class="com.gzydt.oa.command.DeptDeleteCommand">  <property name="deptService" ref="deptService" />  </action>  </command>  </command-bundle>    </blueprint>

第三层:restful层

package com.gzydt.oa.resource;    import java.text.ParseException;  import java.util.List;    import javax.ws.rs.Consumes;  import javax.ws.rs.DELETE;  import javax.ws.rs.GET;  import javax.ws.rs.HeaderParam;  import javax.ws.rs.POST;  import javax.ws.rs.PUT;  import javax.ws.rs.Path;  import javax.ws.rs.PathParam;  import javax.ws.rs.Produces;  import javax.ws.rs.QueryParam;  import javax.ws.rs.core.MediaType;  import javax.ws.rs.core.Response;    import org.apache.cxf.jaxrs.ext.multipart.Attachment;  import org.apache.cxf.jaxrs.ext.multipart.Multipart;    /*** 内部邮件的restful*/  @Path("/mails")  public interface InnerMailsResource {  /** * 新增邮件 * @param content * @return */  @POST  @Path("/")  @Consumes("multipart/form-data")  public Response save(@Multipart("content") String content,  List<Attachment> attachments) throws ParseException ;  /** * 更新邮件 * @param id * @param content * @return */  @PUT  @Path("/{id}")  @Consumes("multipart/form-data")  public Response update(@PathParam("id") long id,@Multipart("content") String content,  List<Attachment> attachments) throws ParseException;  /** * 查询邮件 * @param id * @return */  @GET  @Path("/{id}")  public Response get(@PathParam("id") long id);  /** * 查询邮件列表 * @param range * @param query * @return */  @GET  @Path("/list")  public Response getList(@HeaderParam("range") String range,@QueryParam("query") String query);  /** * 删除邮件 * @param id * @return */  @DELETE  @Path("/{id}")  public Response delete(@PathParam("id") long id);  /** * 发送邮件 * @param content * @return  数据格式:{"data":{},"toMail":""} */  @POST  @Path("/sendMails/{id}")  public Response sendMail(@PathParam("id")long id) ;  /** * 下载附件 * @param id(附件的id) * @return */  @GET  @Path("/getAttach/{id}")  @Produces(MediaType.APPLICATION_OCTET_STREAM)  public Response downLoadAttach(@PathParam("id") long id);        }

实现类:

package com.gzydt.oa.resource.impl;    import java.io.File;  import java.io.FileInputStream;  import java.io.FileOutputStream;  import java.io.IOException;  import java.io.InputStream;  import java.io.OutputStream;  import java.text.ParseException;  import java.text.SimpleDateFormat;  import java.util.Date;  import java.util.HashSet;  import java.util.List;  import java.util.Set;  import java.util.regex.Matcher;  import java.util.regex.Pattern;    import javax.activation.DataHandler;  import javax.ws.rs.WebApplicationException;  import javax.ws.rs.core.MediaType;  import javax.ws.rs.core.Response;  import javax.ws.rs.core.Response.Status;  import javax.ws.rs.core.StreamingOutput;    import net.sf.json.JSONArray;  import net.sf.json.JSONObject;  import net.sf.json.JsonConfig;    import org.apache.cxf.jaxrs.ext.multipart.Attachment;  /*import org.json.JSONArray;*/    import com.gzydt.oa.commons.QueryParam;  import com.gzydt.oa.entity.AppendFile;  import com.gzydt.oa.entity.InnerMails;  import com.gzydt.oa.resource.InnerMailsResource;  import com.gzydt.oa.service.AppendFileService;  import com.gzydt.oa.service.InnerMailService;  import com.gzydt.oa.util.Constant;  import com.gzydt.oa.util.QueryUtil;    public class InnerMailsResourceImpl implements InnerMailsResource {  private InnerMailService emailService;    public void setEmailService(InnerMailService emailService) {  this.emailService = emailService;  }    private AppendFileService appendFileService;    public void setAppendFileService(AppendFileService appendFileService) {  this.appendFileService = appendFileService;  }    private static final String PATH = "data/oa/upload/mails";    @Override  public Response save(String content, List<Attachment> attachments) throws ParseException {  //去掉懒加载字段  JsonConfig jsonConfig = Constant.jsonDateConfig;  jsonConfig.setExcludes(new String[] { "appendFiles"});  JSONObject preceInfo = JSONObject.fromObject(content);  JSONObject backInfo = new JSONObject();  InnerMails entity = new InnerMails();  Date sendDate = null;  if ( preceInfo.optString("sendDate") != null && preceInfo.optString("sendDate") != "" ) {  //这里的MM必须是要大写,若是写为mm,则是分钟,,格式一定要按照正规的来写<span class="con">yyyy-MM-dd HH:mm:ss</span>,  //该大写就大写,小写就小写,并且中间有空格等,都不能错误。不然时间会出错  SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd");  sendDate = df.parse(preceInfo.optString("sendDate"));  }  preceInfo.put("sendDate", sendDate);  entity = (InnerMails) JSONObject.toBean(preceInfo, InnerMails.class);  if ( !preceInfo.has("type") ) {  entity.setType(0);  }  entity = emailService.save(entity);   // 新增附件到附件表中  Set<AppendFile> appfiles=addAttach(attachments, entity);  entity.setAppendFiles(appfiles);  entity=emailService.update(entity);    if ( null != entity ) {  backInfo = JSONObject.fromObject(entity);  return Response.ok(backInfo.toString(), MediaType.APPLICATION_JSON).build();  }  backInfo.put("message", "保存失败");  return Response.ok(backInfo.toString(), MediaType.APPLICATION_JSON).build();    }    // 保存并关联附件  private  Set<AppendFile>  addAttach(List<Attachment> attachments,InnerMails entity){  Set<AppendFile> appenFiles=new HashSet<AppendFile>();  for (Attachment attachment : attachments) {  if (attachment.getContentType().toString().startsWith("application/octet-stream")) {  DataHandler dh = attachment.getDataHandler();  long time = new Date().getTime();  String fileName = null;  try {  fileName = new String(dh.getName().getBytes("ISO-8859-1"), "UTF-8");  writeFile(dh, fileName);  } catch (Exception e) {  e.printStackTrace();  }  AppendFile file = new AppendFile();  file.setSerialNumber(time);// 唯一标识  file.setFileName(fileName);// 文件名  file.setExtension(fileName.substring(fileName.lastIndexOf(".") + 1));// 文件后缀file.setType("email");// 文件类型  emailService.addAttach(entity.getId(), file);  AppendFile result = null;  result = appendFileService.getByNumber(time);  appenFiles.add(result);    }  }  return appenFiles;  }    // 写文件  private void writeFile(DataHandler dh, String fileName) throws IOException {  InputStream is = dh.getInputStream();  File file = new File(PATH);  if ( !file.exists() ) {  file.mkdirs();  }  // LOG.info("附件目录:" + file.getAbsolutePath());  writeToFile(is, PATH + fileName);  }    private void writeToFile(InputStream is, String path) throws IOException {    File file = new File(path);  OutputStream out = new FileOutputStream(file);  int len = 0;  byte[] bytes = new byte[1024];  while ( (len = is.read(bytes)) != -1 ) {  out.write(bytes, 0, len);  }  out.flush();  out.close();    }    @Override  public Response update(long id, String content, List<Attachment> attachments) throws ParseException {  InnerMails entity = emailService.get(id);  JSONObject preceInfo = JSONObject.fromObject(content);  JSONObject backInfo = new JSONObject();  if ( null != entity ) {  entity.setSubject(preceInfo.optString("subject"));  entity.setToMails(preceInfo.optString("toMails"));  entity.setUrgency(preceInfo.optString("urgency"));  Date sendDate = null;  if ( preceInfo.optString("sendDate") != null && preceInfo.optString("sendDate") != "" ) {  SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd");  sendDate = df.parse(preceInfo.optString("sendDate"));  }    //保存附件  Set<AppendFile> appfiles=addAttach(attachments, entity);  entity.setAppendFiles(appfiles);  entity.setSendDate(sendDate);  entity.setContent(preceInfo.optString("content"));  entity.setMailUser(preceInfo.optString("mailUser"));  entity.setSendMail(preceInfo.optString("sendMail"));  entity.setType(preceInfo.optInt("type"));  addAttach(attachments, entity);  entity = emailService.update(entity);  if ( entity != null ) {  backInfo = JSONObject.fromObject(entity);  return Response.ok(backInfo.toString(), MediaType.APPLICATION_JSON).build();  } else {  backInfo.put("message", "修改失败");  return Response.ok(backInfo.toString(), MediaType.APPLICATION_JSON).build();  }    }  backInfo.put("message", "没有找到指定的邮件");  return Response.ok(backInfo.toString(), MediaType.APPLICATION_JSON).build();  }    @Override  public Response get(long id) {  JSONObject backInfo = new JSONObject();  InnerMails entity = emailService.get(id);  JSONObject jo;  /*JsonConfig JSONConfig = Constant.jsonDateConfigWithHour;JSONConfig.setExcludes(new String[] {"appendFiles"});*/  // 去掉延迟加载的字段  jo = JSONObject.fromObject(entity);  //修改状态为已读  entity.setType(3);  emailService.update(entity);  if ( null != entity ) {  backInfo = JSONObject.fromObject(jo);  return Response.ok(backInfo.toString(), MediaType.APPLICATION_JSON).build();  }  backInfo.put("message", "没有找到指定的内部邮件");  return Response.ok(backInfo.toString(), MediaType.APPLICATION_JSON).build();  }    @Override  public Response getList(String range, String query) {  QueryParam queryParam = new QueryParam();  int from = 0;  int to = 9;  try {  String[] ranges = range.replace("items=",  "").split("-");  from = Integer.parseInt(ranges[0]);  to = Integer.parseInt(ranges[1]);  } catch ( Exception e ) {  e.printStackTrace();  }    queryParam.setFirst(from);  int max = to - from + 1;  if ( max > 0 ) {  queryParam.setMax(max);  }  if(null!=query){  QueryUtil.prepareQuery(query, queryParam);  }  int count=emailService.getCount(queryParam);  List<InnerMails> list=emailService.getlist(queryParam);     JsonConfig jsonconfig=Constant.jsonDateConfig;  jsonconfig.setExcludes(new String[] {"appendFiles"});  String contentRange=String.format("items %d-%d/%d", from,to,count);  JSONArray ja = JSONArray.fromObject(list,jsonconfig);  String entity = ja.toString();  return Response.ok(entity, MediaType.APPLICATION_JSON).header("Content-Range", contentRange).build();    }    @Override  public Response delete(long id) {     JSONObject backInfo=new JSONObject();     try {  emailService.delete(id);  backInfo.put("message", "删除成功");  return Response.ok(backInfo.toString(), MediaType.APPLICATION_JSON).build();  } catch ( Exception e ) {  backInfo.put("message","删除失败");  return Response.ok(backInfo.toString(),MediaType.APPLICATION_JSON).build();  }  }    @Override  public Response sendMail(/*String content,List<Attachment> attachments*/long id){     JSONObject backInfo=new JSONObject();  //通过id找到对应的邮件  InnerMails entity=emailService.get(id);  //将A的邮件mail状态改为发送  entity.setType(1);  entity=emailService.update(entity);  //找到收件人,根据收件人的个数来新增多条邮件  String toMail=entity.getToMails();  String[] toMails=toMail.split(",");    for(String tomail:toMails){  //新增邮件,修改mail1的拥有者,修改状态为未读  InnerMails newMails=new InnerMails();  newMails.setSubject(entity.getSubject());  newMails.setToMails(entity.getToMails());  newMails.setUrgency(entity.getUrgency());  newMails.setAppendFiles(entity.getAppendFiles());  newMails.setSendDate(entity.getSendDate());  newMails.setContent(entity.getContent());  newMails.setSendMail(entity.getSendMail());  newMails.setType(2);  newMails.setMailUser(getNoFromChar(tomail));  emailService.save(newMails);  }    backInfo.put("发送邮件的人数", toMails.length);  return Response.ok(backInfo.toString(), MediaType.APPLICATION_JSON).build();      }  //截取字符串中的数字   private String  getNoFromChar(String params) {  String regex="[^0-9]";  Pattern p=Pattern.compile(regex);  Matcher m=p.matcher(params);  return m.replaceAll("").trim();  }    @Override  public Response downLoadAttach(long id) {     //根据附件名称去指定路径中找附件  AppendFile appendFile=appendFileService.get(id);  if ( null == appendFile ) {  return Response.status(Status.NOT_FOUND).entity("找不到文件").build();  }  final File file=new File(PATH, appendFile.getFileName());  JSONObject preceInfo=new JSONObject();  if(!file.exists()||!file.isFile()){  preceInfo.put("message","没有找到指定的文件");  return Response.status(Status.NOT_FOUND).entity("找不到文件:"+file.getName()).build();  }  //下载附件  StreamingOutput entity=downLoad(file);  String fileName=file.getName().toLowerCase();  String type=MediaType.APPLICATION_OCTET_STREAM;  if(fileName.endsWith(".jpg")||fileName.endsWith(".png")){  type="image/jpeg";  }else if(fileName.endsWith(".doc")){  type="application/msword;charset=utf-8";  }else if(fileName.endsWith(".pdf")){  type="application/pdf;charset=utf-8";  }  try {  //结局中文名字乱码的问题  fileName=new String(file.getName().getBytes("UTF-8"),"ISO-8859-1");  } catch ( Exception e ) {  // TODO: handle exception  }  return Response.ok(entity, type).header("Content-disposition", "inline;filename="+fileName).build();  }    //下载附件方法  private StreamingOutput downLoad(final File file) {  StreamingOutput entity=new StreamingOutput() {    @Override  public void write(OutputStream output) throws IOException, WebApplicationException {  int len=0;  byte[] buffer=new byte[1024];  InputStream intpStream=new FileInputStream(file);  while((len = intpStream.read(buffer))>0){  output.write(buffer, 0,len);  }  intpStream.close();  output.flush();  output.close();    }  };  return entity;  }  }

restful层的配置文件:

<?xml version="1.0" encoding="UTF-8"?><blueprint xmlns="http://www.osgi.org/xmlns/blueprint/v1.0.0"xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:jaxrs="http://cxf.apache.org/blueprint/jaxrs"xmlns:cxf="http://cxf.apache.org/blueprint/core"xsi:schemaLocation="  http://www.osgi.org/xmlns/blueprint/v1.0.0 http://www.osgi.org/xmlns/blueprint/v1.0.0/blueprint.xsd  http://cxf.apache.org/blueprint/jaxrs http://cxf.apache.org/schemas/blueprint/jaxrs.xsd  http://cxf.apache.org/blueprint/core http://cxf.apache.org/schemas/blueprint/core.xsd"><jaxrs:server id="OARestService" address="/oa">  <jaxrs:serviceBeans>   <ref component-id="mailRestService" />  </jaxrs:serviceBeans>  <jaxrs:providers>   <ref component-id="authFilter" />  </jaxrs:providers></jaxrs:server><!-- implements OAuthDataProvider --><bean id="oauthProvider" class="com.gzydt.oa.auth.OAuthManager" /><bean id="authorizationService"  class="org.apache.cxf.rs.security.oauth2.services.AuthorizationCodeGrantService">  <property name="dataProvider" ref="oauthProvider" /></bean><jaxrs:server id="appServer" address="/myapp">  <jaxrs:serviceBeans>   <ref component-id="authorizationService" />  </jaxrs:serviceBeans></jaxrs:server><!-- <cxf:bus> <cxf:features> <cxf:logging /> </cxf:features> </cxf:bus> --><!-- We are using the OSGi Blueprint XML syntax to define a bean that we  referred to in our JAX-RS server setup. This bean carries a set of JAX-RS  annotations that allow its methods to be mapped to incoming requests. --><bean id="authRestService" class="com.gzydt.oa.resource.impl.AuthResourceImpl">  <property name="userService" ref="userService" /></bean><bean id="authFilter" class="com.gzydt.oa.auth.AuthenticationFilter"></bean><bean id="backlogRestService" class="com.gzydt.oa.resource.impl.BacklogResourceImpl">  <property name="registerService" ref="registerService" /></bean><bean id="securityResource" class="com.gzydt.oa.resource.impl.SecurityResourceImpl"  scope="singleton" init-method="init" destroy-method="destroy">  <property name="userService" ref="userService" />  <property name="deptService" ref="deptService" />  <property name="dutyUsersService" ref="dutyUsersService" />  <property name="unitService" ref="unitService" /></bean><!--添加bean --><bean id="mailRestService" class="com.gzydt.oa.resource.impl.InnerMailsResourceImpl">  <property name="emailService" ref="emailService" />  <property name="appendFileService" ref="appendFileService" /></bean><!--添加bean结束 --><reference id="emailService" interface="com.gzydt.oa.service.InnerMailService" />  <!--添加reference结束 --></blueprint>

解析前端传来的参数:

package com.gzydt.oa.util;import java.util.ArrayList;import java.util.HashMap;import java.util.Iterator;import java.util.List;import java.util.Map;import net.sf.json.JSONObject;import com.gzydt.oa.commons.QueryParam;public class QueryUtil {/** * 解析url中的查询条件的参数 * * @param query *:查询标示 * @param queryParam *:url中的查询参数 * @return 为空 */public static void prepareQuery(String query, QueryParam queryParam) {try {JSONObject jo = JSONObject.fromObject(query);Map<String, String> param = new HashMap<String, String>();List<String> sorts = new ArrayList<String>();for ( @SuppressWarnings("unchecked")Iterator<String> iterator = jo.keySet().iterator(); iterator.hasNext(); ) {String key = iterator.next();String value = jo.optString(key);if ( !value.isEmpty() ) {if ( "sort".equals(key) ) {for ( String s : value.split(",") ) {if ( null != s ) {if ( s.startsWith("8") ) {// 前端无法传“+”s = "+" + s.substring(1, s.length());} else {s = "-" + s.substring(1, s.length());}sorts.add(s);}}} else {param.put(key, value);}}}queryParam.setParam(param);queryParam.setSorts(sorts);} catch ( Exception e ) {e.printStackTrace();}}}

内部邮件的测试类:

package com.gzydt.oa.resource;import java.io.File;import java.io.FileNotFoundException;import java.io.IOException;import java.io.UnsupportedEncodingException;import java.util.ArrayList;import java.util.List;import org.apache.commons.httpclient.HttpClient;import org.apache.commons.httpclient.HttpException;import org.apache.commons.httpclient.NameValuePair;import org.apache.commons.httpclient.methods.DeleteMethod;import org.apache.commons.httpclient.methods.GetMethod;import org.apache.commons.httpclient.methods.PostMethod;import org.apache.commons.httpclient.methods.PutMethod;import org.apache.commons.httpclient.methods.RequestEntity;import org.apache.commons.httpclient.methods.StringRequestEntity;import org.apache.commons.httpclient.methods.multipart.FilePart;import org.apache.commons.httpclient.methods.multipart.MultipartRequestEntity;import org.apache.commons.httpclient.methods.multipart.Part;import org.apache.commons.httpclient.methods.multipart.StringPart;import org.apache.commons.httpclient.params.HttpMethodParams;import org.json.JSONObject;import org.junit.Assert;import org.junit.Test;import org.slf4j.Logger;import org.slf4j.LoggerFactory;public class MailTest extends Tester {private static final String TEST_URL = "http://localhost:8181/cxf/oa/mails";private static final Logger LOG = LoggerFactory.getLogger(MailTest.class);/** * 登记信息写邮件 * * @throws FileNotFoundException */@Testpublic void uploadOrigText() throws FileNotFoundException {/* * JSONObject jo = new JSONObject(); jo.put("subject", "周末计划");// 主题 * jo.put("toMails", "aa<13>,bb<2>");// 收件人 格式 :姓名<userId>;姓名<userId> * jo.put("urgency", "加急");// 紧急程度 jo.put("sendDate", "2015-4-11");// * 发布日期 jo.put("content", "周末购物");// 邮件内容 jo.put("mailUser", "14");// * 邮件拥有者 格式:userId jo.put("sendMail", "cc<14>");// 邮件发送者 * 格式:姓名<userId>,若只是新建,则不需要改字段 jo.put("type", "0");// * 状态标示:-1删除;0草稿;1发送;2未读收件,3已读收件 //新增不需要增加type */// 要上传的文件String path = "F:\\1.doc";String path1 = "F:\\3.doc";long start = System.currentTimeMillis();File file = new File(path);File fileText = new File(path1);// 'type': '0',String content = "{ 'content': '周末野炊','sendDate': '2015-04-11','toMails': 'aa<13>,bb<2>','mailUser': '14','subject': '周末计划','sendMail': '','urgency': '加急'}";Part[] parts = { new FilePart("file", file, "application/octet-stream", "UTF-8"),new FilePart("file", fileText, "application/octet-stream", "UTF-8"),new StringPart("content", content, "UTF-8") };PostMethod post = new PostMethod(TEST_URL);post.setRequestEntity(new MultipartRequestEntity(parts, post.getParams()));HttpClient httpclient = new HttpClient();String res = "";try {int result = httpclient.executeMethod(post);LOG.info("Response status code: " + result);LOG.info("Response body: ");res = getStringFromInputStream(post.getResponseBodyAsStream());LOG.info(res);} catch ( Exception e ) {LOG.error("Error connecting to {}", TEST_URL);Assert.fail("Connection error");} finally {// Release current connection to the connection pool once you// are// donepost.releaseConnection();}LOG.info("断言:验证成功返回【ok】响应!");Assert.assertTrue("ok".equals(res));long end = System.currentTimeMillis();LOG.info("验证用时(毫秒):" + (end - start));}/** * 发邮件 * @throws Exception * @throws FileNotFoundException */@Testpublic void sendEmail() throws Exception {long id = 2l;LOG.info("开始测试发送邮件");PostMethod post = new PostMethod(TEST_URL +"/sendMails/"+ id);post.addRequestHeader("Accept", "application/json");post.getParams().setParameter(HttpMethodParams.HTTP_CONTENT_CHARSET, "UTF-8");     /* JSONObject jo = new JSONObject();jo.put("subject", "周末计划");// 主题jo.put("toMails", "aa<13>,bb<2>");// 收件人 格式 :姓名<userId>;姓名<userId>jo.put("urgency", "加急");// 紧急程度jo.put("sendDate", "2015-4-11");// 发布日期jo.put("content", "周末购物");// 邮件内容jo.put("mailUser", "14");// 邮件拥有者 格式:userIdjo.put("sendMail", "cc<14>");// 邮件发送者 格式:姓名<userId>,若只是新建,则不需要改字段*/  //  LOG.debug("设置请求参数:" + jo.toString());JSONObject jo = new JSONObject();RequestEntity entity = new StringRequestEntity(jo.toString(), "application/json", "UTF-8");post.setRequestEntity(entity);HttpClient httpclient = new HttpClient();String res = "";LOG.info("发送post请求");int result = httpclient.executeMethod(post);LOG.info(" 响应状态:" + result);res = this.getStringFromInputStream(post.getResponseBodyAsStream());LOG.info("响应结果::" + res);LOG.info("断言:");    }// 将邮件放进回收站,是将状态改为-1@Testpublic void updateTest() throws FileNotFoundException {LOG.info("开始测试更新");long id = 1;PutMethod put = new PutMethod(TEST_URL + "/" + id);// 要上传的文件String path = "F:\\1.doc";String path1 = "F:\\3.doc";long start = System.currentTimeMillis();File file = new File(path);File fileText = new File(path1);String content = "{ 'content': '周末加班','sendDate': '2015-4-11','toMails': 'aa<13>,bb<2>','mailUser': '14','subject': '周末计划','type': '0','sendMail': '','urgency': '加急'}";Part[] parts = { new FilePart("file", file, "application/octet-stream", "UTF-8"),new FilePart("file", fileText, "application/octet-stream", "UTF-8"),new StringPart("content", content, "UTF-8") };put.addRequestHeader("Accept", "application/json");put.getParams().setParameter(HttpMethodParams.HTTP_CONTENT_CHARSET, "UTF-8");put.setRequestEntity(new MultipartRequestEntity(parts, put.getParams()));HttpClient httpclient = new HttpClient();String res = "";try {int result = httpclient.executeMethod(put);LOG.info("Response status code: " + result);LOG.info("Response body: ");res = getStringFromInputStream(put.getResponseBodyAsStream());LOG.info(res);} catch ( Exception e ) {LOG.error("Error connecting to {}", TEST_URL);Assert.fail("Connection error");} finally {put.releaseConnection();}LOG.info("断言:验证成功返回【ok】响应!");Assert.assertTrue("ok".equals(res));long end = System.currentTimeMillis();LOG.info("验证用时(毫秒):" + (end - start));}/** * 根据特定的id来找到值班人员的用户信息 */@Testpublic void findTest() {long id = 15L;GetMethod get = new GetMethod(TEST_URL + "/" + id);HttpClient client = new HttpClient();String res = "";try {int retCode = client.executeMethod(get);LOG.info("响应状态 " + retCode);res = this.getStringFromInputStream(get.getResponseBodyAsStream());LOG.info("响应结果" + res);} catch ( Exception e ) {LOG.error("该url路径出错,服务未开启,请检查", TEST_URL + "/" + id);Assert.fail("连接失败.");} finally {get.releaseConnection();}}@Testpublic void queryTest() {LOG.info("开始测试分页查询");GetMethod get = new GetMethod(TEST_URL + "/list");get.getParams().setParameter(HttpMethodParams.HTTP_CONTENT_CHARSET, "UTF-8");List<NameValuePair> params = new ArrayList<NameValuePair>();// 设置分页有信息get.setRequestHeader("Range", "items=0-9");JSONObject jo = new JSONObject();LOG.debug("请求参数::" + jo.toString());// jo.put("mailUser", "14");jo.put("sendDate", "2015-01-10~2015-01-13");/* * jo.put("type", "0"); jo.put("content","周末"); */// jo.put("issueDate", "2015-01-10~2015-02-24");// jo.put("sendFileDate", "2015-01-14~2015-02-04");// jo.put("getFileDate", "2015-01-11~2015-02-04");// jo.put("fromUnit", "Scgovernment");/* jo.put("issueDate", "2015-3") */// jo.put("number","Yfp");// jo.put("refNumber", "a11111");// jo.put("sendUnit", "Shengbangongting");// jo.put("title","22222");JSONObject jb = new JSONObject();params.add(new NameValuePair("query", jo.toString()));// 从0开始的get.setQueryString(params.toArray(new NameValuePair[0]));HttpClient httpClient = new HttpClient();String res = "";try {int result = httpClient.executeMethod(get);LOG.info("响应状态 " + result);res = this.getStringFromInputStream(get.getResponseBodyAsStream());LOG.info("响应结果 " + res);} catch ( Exception e ) {LOG.error("该url路径出错,服务未开启,请检查", TEST_URL);Assert.fail("连接失败.");} finally {get.releaseConnection();}}/** * 测试删除周知事项 */@Testpublic void TestDelete() {LOG.info("开始测试删除通知");long id = 1L;DeleteMethod delete = new DeleteMethod(TEST_URL + "/" + id);HttpClient client = new HttpClient();String res = "";try {LOG.info("发送delete请求删除通知");int retCode = client.executeMethod(delete);LOG.info("响应状态:" + retCode);res = this.getStringFromInputStream(delete.getResponseBodyAsStream());LOG.info("响应结果: " + res);} catch ( Exception e ) {LOG.error("测试错误", e);Assert.fail("连接出错");} finally {/* 释放url的资源 */delete.releaseConnection();}LOG.info(res);}}

在添加一个正常的测试新增的方法:

@Testpublic void testAdd(){  LOG.info("开始测试增加");  PostMethod post = new PostMethod(TEST_URL);  post.addRequestHeader("Accept", "application/json");  post.getParams().setParameter(HttpMethodParams.HTTP_CONTENT_CHARSET,"UTF-8");  try {   JSONObject jo = new JSONObject();     jo.put("day", "2015-4-10");   jo.put("type", "0");   jo.put("content", "gfdz参加了2014年广东省某某某某活动");   jo.put("mainLeaderIds", "2,3");   jo.put("relevantLeaderIds", "5,6");   // date   jo.put("goverEvent", true);   jo.put("ownEvent", false);   jo.put("ownerId", "2");   LOG.debug("设置请求参数:" + jo.toString());   RequestEntity entity = new StringRequestEntity( jo.toString(), "application/json", "UTF-8");   post.setRequestEntity(entity);   HttpClient httpclient = new HttpClient();   String res = "";   LOG.info("发送post请求");   int result = httpclient.executeMethod(post);   LOG.info(" 响应状态:" + result);   res = this.getStringFromInputStream(post.getResponseBodyAsStream());   LOG.info("响应结果::" + res);   Assert.assertTrue(res.contains("增加值班表"));  } catch (Exception e) {   LOG.error("测试错误", e);   Assert.fail("连接出错");  } finally {   post.releaseConnection();  }}

感谢大家的阅读,希望大家收益多多。

本文转自: http://community.itbbs.cn/thread/758207/

推荐教程:《java视频教程》

相关文章