lxy/redkale
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Browse Source
This commit is contained in:
Redkale
2016-09-04 08:19:13 +08:00
parent c2fd7be5e2
commit 90884f4621
6 changed files with 145 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

94
src/org/redkale/net/http/HttpRequest.java
View File

@@ -691,6 +691,19 @@ public class HttpRequest extends Request<HttpContext> {
return header.getShortValue(name, defaultValue);
}
/**
* 获取指定的header的short值, 没有返回默认short值
*
* @param radix 进制数
* @param name header名
* @param defaultValue 默认short值
*
* @return header值
*/
public short getShortHeader(int radix, String name, short defaultValue) {
return header.getShortValue(name, defaultValue);
}
/**
* 获取指定的header的short值, 没有返回默认short值
*
@@ -703,6 +716,19 @@ public class HttpRequest extends Request<HttpContext> {
return header.getShortValue(name, (short) defaultValue);
}
/**
* 获取指定的header的short值, 没有返回默认short值
*
* @param radix 进制数
* @param name header名
* @param defaultValue 默认short值
*
* @return header值
*/
public short getShortHeader(int radix, String name, int defaultValue) {
return header.getShortValue(radix, name, (short) defaultValue);
}
/**
* 获取指定的header的int值, 没有返回默认int值
*
@@ -715,6 +741,19 @@ public class HttpRequest extends Request<HttpContext> {
return header.getIntValue(name, defaultValue);
}
/**
* 获取指定的header的int值, 没有返回默认int值
*
* @param radix 进制数
* @param name header名
* @param defaultValue 默认int值
*
* @return header值
*/
public int getIntHeader(int radix, String name, int defaultValue) {
return header.getIntValue(radix, name, defaultValue);
}
/**
* 获取指定的header的long值, 没有返回默认long值
*
@@ -727,6 +766,19 @@ public class HttpRequest extends Request<HttpContext> {
return header.getLongValue(name, defaultValue);
}
/**
* 获取指定的header的long值, 没有返回默认long值
*
* @param radix 进制数
* @param name header名
* @param defaultValue 默认long值
*
* @return header值
*/
public long getLongHeader(int radix, String name, long defaultValue) {
return header.getLongValue(radix, name, defaultValue);
}
/**
* 获取指定的header的float值, 没有返回默认float值
*
@@ -842,6 +894,20 @@ public class HttpRequest extends Request<HttpContext> {
return params.getShortValue(name, defaultValue);
}
/**
* 获取指定的参数short值, 没有返回默认short值
*
* @param radix 进制数
* @param name 参数名
* @param defaultValue 默认short值
*
* @return 参数值
*/
public short getShortParameter(int radix, String name, short defaultValue) {
parseBody();
return params.getShortValue(radix, name, defaultValue);
}
/**
* 获取指定的参数short值, 没有返回默认short值
*
@@ -855,6 +921,20 @@ public class HttpRequest extends Request<HttpContext> {
return params.getShortValue(name, (short) defaultValue);
}
/**
* 获取指定的参数int值, 没有返回默认int值
*
* @param radix 进制数
* @param name 参数名
* @param defaultValue 默认int值
*
* @return 参数值
*/
public int getIntParameter(int radix, String name, int defaultValue) {
parseBody();
return params.getIntValue(radix, name, defaultValue);
}
/**
* 获取指定的参数int值, 没有返回默认int值
*
@@ -881,6 +961,20 @@ public class HttpRequest extends Request<HttpContext> {
return params.getLongValue(name, defaultValue);
}
/**
* 获取指定的参数long值, 没有返回默认long值
*
* @param radix 进制数
* @param name 参数名
* @param defaultValue 默认long值
*
* @return 参数值
*/
public long getLongParameter(int radix, String name, long defaultValue) {
parseBody();
return params.getLongValue(radix, name, defaultValue);
}
/**
* 获取指定的参数float值, 没有返回默认float值
*

39
src/org/redkale/net/http/Rest.java
View File

@@ -223,6 +223,7 @@ public final class Rest {
Map<String, Object> paramMap = new LinkedHashMap<>();
final Class ptype = param.getType();
String n = null;
int radix = 10;
RestHeader annhead = null;
RestCookie anncookie = null;
RestAddress annaddr = null;
@@ -230,6 +231,7 @@ public final class Rest {
annhead = param.getAnnotation(RestHeader.class);
if (annhead != null) {
n = annhead.value();
radix = annhead.radix();
if (n.isEmpty()) throw new RuntimeException("@RestHeader.value is illegal in " + method);
}
anncookie = param.getAnnotation(RestCookie.class);
@@ -237,6 +239,7 @@ public final class Rest {
if (annhead != null) throw new RuntimeException("@RestCookie and @RestHeader cannot on the same Parameter in " + method);
if (ptype != String.class) throw new RuntimeException("@RestCookie must on String Parameter in " + method);
n = anncookie.value();
radix = anncookie.radix();
if (n.isEmpty()) throw new RuntimeException("@RestCookie.value is illegal in " + method);
}
annaddr = param.getAnnotation(RestAddress.class);
@@ -247,6 +250,7 @@ public final class Rest {
}
}
RestParam annpara = param.getAnnotation(RestParam.class);
if (annpara != null) radix = annpara.radix();
if (n == null) n = (annpara == null || annpara.value().isEmpty()) ? null : annpara.value();
if (n == null && ptype == userType) n = "_current_user";
if (n == null) {
@@ -300,13 +304,15 @@ public final class Rest {
} else if (ptype == byte.class) {
mv.visitVarInsn(ALOAD, 1);
mv.visitMethodInsn(INVOKEVIRTUAL, "org/redkale/net/http/HttpRequest", "getRequstURILastPath", "()Ljava/lang/String;", false);
mv.visitMethodInsn(INVOKESTATIC, "java/lang/Byte", "parseByte", "(Ljava/lang/String;)B", false);
mv.visitIntInsn(BIPUSH, radix);
mv.visitMethodInsn(INVOKESTATIC, "java/lang/Byte", "parseByte", "(Ljava/lang/String;I)B", false);
mv.visitVarInsn(ISTORE, maxLocals);
varInsns.add(new int[]{ILOAD, maxLocals});
} else if (ptype == short.class) {
mv.visitVarInsn(ALOAD, 1);
mv.visitMethodInsn(INVOKEVIRTUAL, "org/redkale/net/http/HttpRequest", "getRequstURILastPath", "()Ljava/lang/String;", false);
mv.visitMethodInsn(INVOKESTATIC, "java/lang/Short", "parseShort", "(Ljava/lang/String;)S", false);
mv.visitIntInsn(BIPUSH, radix);
mv.visitMethodInsn(INVOKESTATIC, "java/lang/Short", "parseShort", "(Ljava/lang/String;I)S", false);
mv.visitVarInsn(ISTORE, maxLocals);
varInsns.add(new int[]{ILOAD, maxLocals});
} else if (ptype == char.class) {
@@ -319,7 +325,8 @@ public final class Rest {
} else if (ptype == int.class) {
mv.visitVarInsn(ALOAD, 1);
mv.visitMethodInsn(INVOKEVIRTUAL, "org/redkale/net/http/HttpRequest", "getRequstURILastPath", "()Ljava/lang/String;", false);
mv.visitMethodInsn(INVOKESTATIC, "java/lang/Integer", "parseInt", "(Ljava/lang/String;)I", false);
mv.visitIntInsn(BIPUSH, radix);
mv.visitMethodInsn(INVOKESTATIC, "java/lang/Integer", "parseInt", "(Ljava/lang/String;I)I", false);
mv.visitVarInsn(ISTORE, maxLocals);
varInsns.add(new int[]{ILOAD, maxLocals});
} else if (ptype == float.class) {
@@ -331,7 +338,8 @@ public final class Rest {
} else if (ptype == long.class) {
mv.visitVarInsn(ALOAD, 1);
mv.visitMethodInsn(INVOKEVIRTUAL, "org/redkale/net/http/HttpRequest", "getRequstURILastPath", "()Ljava/lang/String;", false);
mv.visitMethodInsn(INVOKESTATIC, "java/lang/Long", "parseLong", "(Ljava/lang/String;)J", false);
mv.visitIntInsn(BIPUSH, radix);
mv.visitMethodInsn(INVOKESTATIC, "java/lang/Long", "parseLong", "(Ljava/lang/String;I)J", false);
mv.visitVarInsn(LSTORE, maxLocals);
varInsns.add(new int[]{LLOAD, maxLocals});
maxLocals++;
@@ -362,14 +370,16 @@ public final class Rest {
mv.visitLdcInsn(pname);
mv.visitLdcInsn("0");
mv.visitMethodInsn(INVOKEVIRTUAL, "org/redkale/net/http/HttpRequest", ishead ? "getHeader" : "getParameter", "(Ljava/lang/String;Ljava/lang/String;)Ljava/lang/String;", false);
mv.visitMethodInsn(INVOKESTATIC, "java/lang/Byte", "parseByte", "(Ljava/lang/String;)B", false);
mv.visitIntInsn(BIPUSH, radix);
mv.visitMethodInsn(INVOKESTATIC, "java/lang/Byte", "parseByte", "(Ljava/lang/String;I)B", false);
mv.visitVarInsn(ISTORE, maxLocals);
varInsns.add(new int[]{ILOAD, maxLocals});
} else if (ptype == short.class) {
mv.visitVarInsn(ALOAD, 1);
mv.visitIntInsn(BIPUSH, radix);
mv.visitLdcInsn(pname);
mv.visitInsn(ICONST_0);
mv.visitMethodInsn(INVOKEVIRTUAL, "org/redkale/net/http/HttpRequest", ishead ? "getShortHeader" : "getShortParameter", "(Ljava/lang/String;S)S", false);
mv.visitMethodInsn(INVOKEVIRTUAL, "org/redkale/net/http/HttpRequest", ishead ? "getShortHeader" : "getShortParameter", "(ILjava/lang/String;S)S", false);
mv.visitVarInsn(ISTORE, maxLocals);
varInsns.add(new int[]{ILOAD, maxLocals});
} else if (ptype == char.class) {
@@ -383,9 +393,10 @@ public final class Rest {
varInsns.add(new int[]{ILOAD, maxLocals});
} else if (ptype == int.class) {
mv.visitVarInsn(ALOAD, 1);
mv.visitIntInsn(BIPUSH, radix);
mv.visitLdcInsn(pname);
mv.visitInsn(ICONST_0);
mv.visitMethodInsn(INVOKEVIRTUAL, "org/redkale/net/http/HttpRequest", ishead ? "getIntHeader" : "getIntParameter", "(Ljava/lang/String;I)I", false);
mv.visitMethodInsn(INVOKEVIRTUAL, "org/redkale/net/http/HttpRequest", ishead ? "getIntHeader" : "getIntParameter", "(ILjava/lang/String;I)I", false);
mv.visitVarInsn(ISTORE, maxLocals);
varInsns.add(new int[]{ILOAD, maxLocals});
} else if (ptype == float.class) {
@@ -397,9 +408,10 @@ public final class Rest {
varInsns.add(new int[]{FLOAD, maxLocals});
} else if (ptype == long.class) {
mv.visitVarInsn(ALOAD, 1);
mv.visitIntInsn(BIPUSH, radix);
mv.visitLdcInsn(pname);
mv.visitInsn(LCONST_0);
mv.visitMethodInsn(INVOKEVIRTUAL, "org/redkale/net/http/HttpRequest", ishead ? "getLongHeader" : "getLongParameter", "(Ljava/lang/String;J)J", false);
mv.visitMethodInsn(INVOKEVIRTUAL, "org/redkale/net/http/HttpRequest", ishead ? "getLongHeader" : "getLongParameter", "(ILjava/lang/String;J)J", false);
mv.visitVarInsn(LSTORE, maxLocals);
varInsns.add(new int[]{LLOAD, maxLocals});
maxLocals++;
@@ -443,7 +455,7 @@ public final class Rest {
//构建 RestHeader、RestCookie、RestAddress 等赋值操作
Class loop = ptype;
Set<String> fields = new HashSet<>();
Map<String, String> attrParaNames = new LinkedHashMap<>();
Map<String, Object[]> attrParaNames = new LinkedHashMap<>();
do {
if (loop == null || loop.isInterface()) break; //接口时getSuperclass可能会得到null
for (Field field : loop.getDeclaredFields()) {
@@ -468,11 +480,12 @@ public final class Rest {
restname = rc.value();
} else if (ra != null) {
attrFieldName = "_redkale_attr_address_" + restAttributes.size();
//restname = "";
} else {
continue;
}
restAttributes.put(attrFieldName, attr);
attrParaNames.put(attrFieldName, restname);
attrParaNames.put(attrFieldName, new Object[]{restname, field.getType()});
fields.add(field.getName());
}
} while ((loop = loop.getSuperclass()) != Object.class);
@@ -481,17 +494,17 @@ public final class Rest {
mv.visitVarInsn(ALOAD, maxLocals);
Label lif = new Label();
mv.visitJumpInsn(IFNULL, lif); //if(bean != null) {
for (Map.Entry<String, String> en : attrParaNames.entrySet()) {
for (Map.Entry<String, Object[]> en : attrParaNames.entrySet()) {
mv.visitVarInsn(ALOAD, 0);
mv.visitFieldInsn(GETFIELD, newDynName, en.getKey(), attrDesc);
mv.visitVarInsn(ALOAD, maxLocals);
mv.visitVarInsn(ALOAD, 1);
if (en.getKey().contains("_header_")) {
mv.visitLdcInsn(en.getValue());
mv.visitLdcInsn(en.getValue()[0].toString());
mv.visitLdcInsn("");
mv.visitMethodInsn(INVOKEVIRTUAL, "org/redkale/net/http/HttpRequest", "getHeader", "(Ljava/lang/String;Ljava/lang/String;)Ljava/lang/String;", false);
} else if (en.getKey().contains("_cookie_")) {
mv.visitLdcInsn(en.getValue());
mv.visitLdcInsn(en.getValue()[0].toString());
mv.visitLdcInsn("");
mv.visitMethodInsn(INVOKEVIRTUAL, "org/redkale/net/http/HttpRequest", "getCookie", "(Ljava/lang/String;Ljava/lang/String;)Ljava/lang/String;", false);
} else if (en.getKey().contains("_address_")) {

2
src/org/redkale/net/http/RestCookie.java
View File

@@ -25,4 +25,6 @@ public @interface RestCookie {
String value(); //cookie名
String comment() default ""; //备注描述
int radix() default 10; //转换数字byte/short/int/long时所用的进制数， 默认10进制
}

2
src/org/redkale/net/http/RestHeader.java
View File

@@ -25,4 +25,6 @@ public @interface RestHeader {
String value(); //参数名
String comment() default ""; //备注描述
int radix() default 10; //转换数字byte/short/int/long时所用的进制数， 默认10进制
}

1
src/org/redkale/net/http/RestParam.java
View File

@@ -26,4 +26,5 @@ public @interface RestParam {
String comment() default ""; //备注描述
int radix() default 10; //转换数字byte/short/int/long时所用的进制数， 默认10进制
}

20
src/org/redkale/util/AnyValue.java
View File

@@ -386,6 +386,11 @@ public abstract class AnyValue {
return value == null || value.length() == 0 ? defaultValue : Byte.decode(value);
}
public byte getByteValue(int radix, String name, byte defaultValue) {
String value = getValue(name);
return value == null || value.length() == 0 ? defaultValue : (radix == 10 ? Byte.decode(value) : Byte.parseByte(value, radix));
}
public char getCharValue(String name) {
return getValue(name).charAt(0);
}
@@ -404,6 +409,11 @@ public abstract class AnyValue {
return value == null || value.length() == 0 ? defaultValue : Short.decode(value);
}
public short getShortValue(int radix, String name, short defaultValue) {
String value = getValue(name);
return value == null || value.length() == 0 ? defaultValue : (radix == 10 ? Short.decode(value) : Short.parseShort(value, radix));
}
public int getIntValue(String name) {
return Integer.decode(getValue(name));
}
@@ -413,6 +423,11 @@ public abstract class AnyValue {
return value == null || value.length() == 0 ? defaultValue : Integer.decode(value);
}
public int getIntValue(int radix, String name, int defaultValue) {
String value = getValue(name);
return value == null || value.length() == 0 ? defaultValue : (radix == 10 ? Integer.decode(value) : Integer.parseInt(value, radix));
}
public long getLongValue(String name) {
return Long.decode(getValue(name));
}
@@ -422,6 +437,11 @@ public abstract class AnyValue {
return value == null || value.length() == 0 ? defaultValue : Long.decode(value);
}
public long getLongValue(int radix, String name, long defaultValue) {
String value = getValue(name);
return value == null || value.length() == 0 ? defaultValue : (radix == 10 ? Long.decode(value) : Long.parseLong(value, radix));
}
public float getFloatValue(String name) {
return Float.parseFloat(getValue(name));
}