通用接口 - 人脸比对、联动下拉、数据统计等功能
//导入需要使用的包 package com.controller;
import java.io.File; import java.io.FileNotFoundException; import java.io.IOException; import java.text.SimpleDateFormat; import java.util.ArrayList; import java.util.Calendar; import java.util.Date; import java.util.HashMap; import java.util.List; import java.util.Map;
import javax.servlet.http.HttpServletRequest;
import org.apache.commons.lang3.StringUtils; import org.json.JSONObject; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.util.ResourceUtils; import org.springframework.web.bind.annotation.PathVariable; import org.springframework.web.bind.annotation.RequestBody; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RequestParam; import org.springframework.web.bind.annotation.RestController;
import com.annotation.IgnoreAuth; import com.baidu.aip.face.AipFace; import com.baidu.aip.face.MatchRequest; import com.baidu.aip.util.Base64Util; import com.baomidou.mybatisplus.mapper.EntityWrapper; import com.baomidou.mybatisplus.mapper.Wrapper; import com.entity.ConfigEntity; import com.service.CommonService; import com.service.ConfigService; import com.utils.BaiduUtil; import com.utils.FileUtil; import com.utils.R;
/**
-
通用接口 */ @RestController public class CommonController{ @Autowired private CommonService commonService;
@Autowired private ConfigService configService;
private static AipFace client = null;
private static String BAIDU_DITU_AK = null;
//获取经纬度对应的城市名称 @RequestMapping("/location") public R location(String lng,String lat) { if(BAIDU_DITU_AK==null) { //如果BAIDU_DITU_AK为空，则从数据库中获取 BAIDU_DITU_AK = configService.selectOne(new EntityWrapper
().eq('name', 'baidu_ditu_ak')).getValue(); if(BAIDU_DITU_AK==null) { return R.error('请在配置管理中正确配置baidu_ditu_ak'); } } //调用BaiduUtil中的getCityByLonLat方法获取城市信息 Map<String, String> map = BaiduUtil.getCityByLonLat(BAIDU_DITU_AK, lng, lat); return R.ok().put('data', map); } /**
- 人脸比对
- @param face1 人脸1
- @param face2 人脸2
- @return
*/
@RequestMapping("/matchFace")
public R matchFace(String face1, String face2,HttpServletRequest request) {
if(client==null) {
//如果client为空，则从数据库中获取APIKey和SecretKey
String APIKey = configService.selectOne(new EntityWrapper
().eq('name', 'APIKey')).getValue(); String SecretKey = configService.selectOne(new EntityWrapper ().eq('name', 'SecretKey')).getValue(); //调用BaiduUtil中的getAuth方法获取token String token = BaiduUtil.getAuth(APIKey, SecretKey); if(token==null) { return R.error('请在配置管理中正确配置APIKey和SecretKey'); } client = new AipFace(null, APIKey, SecretKey); client.setConnectionTimeoutInMillis(2000); client.setSocketTimeoutInMillis(60000); } JSONObject res = null; try { //获取上传的两张图片 File file1 = new File(request.getSession().getServletContext().getRealPath("/upload")+"/"+face1); File file2 = new File(request.getSession().getServletContext().getRealPath("/upload")+"/"+face2); //将图片转换为Base64编码 String img1 = Base64Util.encode(FileUtil.FileToByte(file1)); String img2 = Base64Util.encode(FileUtil.FileToByte(file2)); //创建MatchRequest对象并添加到ArrayList中 MatchRequest req1 = new MatchRequest(img1, 'BASE64'); MatchRequest req2 = new MatchRequest(img2, 'BASE64'); ArrayList requests = new ArrayList (); requests.add(req1); requests.add(req2); //调用client.match方法进行人脸比对 res = client.match(requests); System.out.println(res.get('result')); } catch (FileNotFoundException e) { e.printStackTrace(); return R.error('文件不存在'); } catch (IOException e) { e.printStackTrace(); } //将比对结果解析为JSON对象并返回 return R.ok().put('data', com.alibaba.fastjson.JSONObject.parse(res.get('result').toString())); }
/**
- 获取table表中的column列表(联动接口)
- @param table
- @param column
- @return
*/
@IgnoreAuth
@RequestMapping("/option/{tableName}/{columnName}")
public R getOption(@PathVariable('tableName') String tableName, @PathVariable('columnName') String columnName,String level,String parent) {
Map<String, Object> params = new HashMap<String, Object>();
params.put('table', tableName);
params.put('column', columnName);
if(StringUtils.isNotBlank(level)) {
params.put('level', level);
}
if(StringUtils.isNotBlank(parent)) {
params.put('parent', parent);
}
//调用commonService中的getOption方法获取下拉选项列表
List
data = commonService.getOption(params); return R.ok().put('data', data); }
/**
- 根据table中的column获取单条记录
- @param table
- @param column
- @return */ @IgnoreAuth @RequestMapping("/follow/{tableName}/{columnName}") public R getFollowByOption(@PathVariable('tableName') String tableName, @PathVariable('columnName') String columnName, @RequestParam String columnValue) { Map<String, Object> params = new HashMap<String, Object>(); params.put('table', tableName); params.put('column', columnName); params.put('columnValue', columnValue); //调用commonService中的getFollowByOption方法获取单条记录 Map<String, Object> result = commonService.getFollowByOption(params); return R.ok().put('data', result); }
/**
- 修改table表的sfsh状态
- @param table
- @param map
- @return */ @RequestMapping("/sh/{tableName}") public R sh(@PathVariable('tableName') String tableName, @RequestBody Map<String, Object> map) { map.put('table', tableName); //调用commonService中的sh方法修改sfsh状态 commonService.sh(map); return R.ok(); }
/**
-
获取需要提醒的记录数
-
@param tableName
-
@param columnName
-
@param type 1:数字 2:日期
-
@param map
-
@return */ @IgnoreAuth @RequestMapping("/remind/{tableName}/{columnName}/{type}") public R remindCount(@PathVariable('tableName') String tableName, @PathVariable('columnName') String columnName, @PathVariable('type') String type,@RequestParam Map<String, Object> map) { map.put('table', tableName); map.put('column', columnName); map.put('type', type);
if(type.equals('2')) { SimpleDateFormat sdf = new SimpleDateFormat('yyyy-MM-dd'); Calendar c = Calendar.getInstance(); Date remindStartDate = null; Date remindEndDate = null; if(map.get('remindstart')!=null) { Integer remindStart = Integer.parseInt(map.get('remindstart').toString()); c.setTime(new Date()); c.add(Calendar.DAY_OF_MONTH,remindStart); remindStartDate = c.getTime(); map.put('remindstart', sdf.format(remindStartDate)); } if(map.get('remindend')!=null) { Integer remindEnd = Integer.parseInt(map.get('remindend').toString()); c.setTime(new Date()); c.add(Calendar.DAY_OF_MONTH,remindEnd); remindEndDate = c.getTime(); map.put('remindend', sdf.format(remindEndDate)); } } //调用commonService中的remindCount方法获取需要提醒的记录数 int count = commonService.remindCount(map); return R.ok().put('count', count); }
/**
- 单列求和 */ @IgnoreAuth @RequestMapping("/cal/{tableName}/{columnName}") public R cal(@PathVariable('tableName') String tableName, @PathVariable('columnName') String columnName) { Map<String, Object> params = new HashMap<String, Object>(); params.put('table', tableName); params.put('column', columnName); //调用commonService中的selectCal方法进行单列求和 Map<String, Object> result = commonService.selectCal(params); return R.ok().put('data', result); }
/**
- 分组统计 */ @IgnoreAuth @RequestMapping("/group/{tableName}/{columnName}") public R group(@PathVariable('tableName') String tableName, @PathVariable('columnName') String columnName) { Map<String, Object> params = new HashMap<String, Object>(); params.put('table', tableName); params.put('column', columnName); List<Map<String, Object>> result = commonService.selectGroup(params); return R.ok().put('data', result); }
/**
- （按值统计） */ @IgnoreAuth @RequestMapping("/value/{tableName}/{xColumnName}/{yColumnName}") public R value(@PathVariable('tableName') String tableName, @PathVariable('yColumnName') String yColumnName, @PathVariable('xColumnName') String xColumnName) { Map<String, Object> params = new HashMap<String, Object>(); params.put('table', tableName); params.put('xColumn', xColumnName); params.put('yColumn', yColumnName); List<Map<String, Object>> result = commonService.selectValue(params); SimpleDateFormat sdf = new SimpleDateFormat('yyyy-MM-dd'); for(Map<String, Object> m : result) { for(String k : m.keySet()) { if(m.get(k) instanceof Date) { m.put(k, sdf.format((Date)m.get(k))); } } } return R.ok().put('data', result); }
原文地址: https://www.cveoy.top/t/topic/nLEz 著作权归作者所有。请勿转载和采集!