mirror of
https://gitee.com/dolphinscheduler/DolphinScheduler.git
synced 2024-12-05 05:38:30 +08:00
Merge remote-tracking branch 'upstream/dev' into dev
This commit is contained in:
commit
4660152f38
19
.github/workflows/ci_ut.yml
vendored
19
.github/workflows/ci_ut.yml
vendored
@ -51,17 +51,16 @@ jobs:
|
|||||||
CODECOV_TOKEN="09c2663f-b091-4258-8a47-c981827eb29a" bash <(curl -s https://codecov.io/bash)
|
CODECOV_TOKEN="09c2663f-b091-4258-8a47-c981827eb29a" bash <(curl -s https://codecov.io/bash)
|
||||||
- name: Run SonarCloud analysis
|
- name: Run SonarCloud analysis
|
||||||
run: >
|
run: >
|
||||||
mvn clean --batch-mode
|
mvn clean --batch-mode
|
||||||
org.jacoco:jacoco-maven-plugin:prepare-agent
|
verify
|
||||||
verify
|
org.sonarsource.scanner.maven:sonar-maven-plugin:3.6.1.1688:sonar
|
||||||
org.sonarsource.scanner.maven:sonar-maven-plugin:sonar
|
-Dsonar.host.url=https://sonarcloud.io
|
||||||
-Dsonar.host.url=https://sonarcloud.io
|
-Dsonar.organization=apache
|
||||||
-Dsonar.organization=apache
|
-Dsonar.projectKey=apache-dolphinscheduler
|
||||||
-Dsonar.projectKey=apache-dolphinscheduler
|
-Dsonar.login=e4058004bc6be89decf558ac819aa1ecbee57682
|
||||||
-Dsonar.login=e4058004bc6be89decf558ac819aa1ecbee57682
|
|
||||||
env:
|
env:
|
||||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||||
SONAR_TOKEN: ${{ secrets.SONAR_TOKEN }}
|
SONAR_TOKEN: ${{ secrets.SONAR_TOKEN }}
|
||||||
- name: Collect logs
|
- name: Collect logs
|
||||||
run: |
|
run: |
|
||||||
mkdir -p ${LOG_DIR}
|
mkdir -p ${LOG_DIR}
|
||||||
|
@ -250,5 +250,23 @@
|
|||||||
<version>${servlet-api.version}</version>
|
<version>${servlet-api.version}</version>
|
||||||
</dependency>
|
</dependency>
|
||||||
|
|
||||||
|
<dependency>
|
||||||
|
<groupId>org.powermock</groupId>
|
||||||
|
<artifactId>powermock-module-junit4</artifactId>
|
||||||
|
<scope>test</scope>
|
||||||
|
</dependency>
|
||||||
|
|
||||||
|
<dependency>
|
||||||
|
<groupId>org.powermock</groupId>
|
||||||
|
<artifactId>powermock-api-mockito2</artifactId>
|
||||||
|
<scope>test</scope>
|
||||||
|
<exclusions>
|
||||||
|
<exclusion>
|
||||||
|
<groupId>org.mockito</groupId>
|
||||||
|
<artifactId>mockito-core</artifactId>
|
||||||
|
</exclusion>
|
||||||
|
</exclusions>
|
||||||
|
</dependency>
|
||||||
|
|
||||||
</dependencies>
|
</dependencies>
|
||||||
</project>
|
</project>
|
@ -16,43 +16,211 @@
|
|||||||
*/
|
*/
|
||||||
package org.apache.dolphinscheduler.api.service;
|
package org.apache.dolphinscheduler.api.service;
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.api.ApiApplicationServer;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
||||||
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
||||||
import org.apache.dolphinscheduler.api.enums.Status;
|
import org.apache.dolphinscheduler.api.enums.Status;
|
||||||
import org.apache.dolphinscheduler.api.utils.PageInfo;
|
import org.apache.dolphinscheduler.api.utils.PageInfo;
|
||||||
|
import org.apache.dolphinscheduler.api.utils.Result;
|
||||||
import org.apache.dolphinscheduler.common.Constants;
|
import org.apache.dolphinscheduler.common.Constants;
|
||||||
|
import org.apache.dolphinscheduler.common.enums.UdfType;
|
||||||
import org.apache.dolphinscheduler.common.enums.UserType;
|
import org.apache.dolphinscheduler.common.enums.UserType;
|
||||||
|
import org.apache.dolphinscheduler.common.utils.CollectionUtils;
|
||||||
|
import org.apache.dolphinscheduler.common.utils.PropertyUtils;
|
||||||
|
import org.apache.dolphinscheduler.dao.entity.Resource;
|
||||||
|
import org.apache.dolphinscheduler.dao.entity.UdfFunc;
|
||||||
import org.apache.dolphinscheduler.dao.entity.User;
|
import org.apache.dolphinscheduler.dao.entity.User;
|
||||||
|
import org.apache.dolphinscheduler.dao.mapper.ResourceMapper;
|
||||||
|
import org.apache.dolphinscheduler.dao.mapper.UDFUserMapper;
|
||||||
|
import org.apache.dolphinscheduler.dao.mapper.UdfFuncMapper;
|
||||||
import org.junit.Assert;
|
import org.junit.Assert;
|
||||||
|
import org.junit.Before;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
import org.junit.runner.RunWith;
|
import org.junit.runner.RunWith;
|
||||||
|
import org.mockito.InjectMocks;
|
||||||
|
import org.mockito.Mock;
|
||||||
|
import org.mockito.Mockito;
|
||||||
|
import org.powermock.api.mockito.PowerMockito;
|
||||||
|
import org.powermock.core.classloader.annotations.PrepareForTest;
|
||||||
|
import org.powermock.modules.junit4.PowerMockRunner;
|
||||||
import org.slf4j.Logger;
|
import org.slf4j.Logger;
|
||||||
import org.slf4j.LoggerFactory;
|
import org.slf4j.LoggerFactory;
|
||||||
import org.springframework.beans.factory.annotation.Autowired;
|
|
||||||
import org.springframework.boot.test.context.SpringBootTest;
|
|
||||||
import org.springframework.test.context.junit4.SpringRunner;
|
|
||||||
|
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.Date;
|
||||||
|
import java.util.List;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
|
|
||||||
@RunWith(SpringRunner.class)
|
@RunWith(PowerMockRunner.class)
|
||||||
@SpringBootTest(classes = ApiApplicationServer.class)
|
@PrepareForTest(PropertyUtils.class)
|
||||||
public class UdfFuncServiceTest {
|
public class UdfFuncServiceTest {
|
||||||
private static final Logger logger = LoggerFactory.getLogger(UdfFuncServiceTest.class);
|
private static final Logger logger = LoggerFactory.getLogger(UdfFuncServiceTest.class);
|
||||||
|
|
||||||
@Autowired
|
@InjectMocks
|
||||||
private UdfFuncService udfFuncService;
|
private UdfFuncService udfFuncService;
|
||||||
|
@Mock
|
||||||
|
private ResourceMapper resourceMapper;
|
||||||
|
@Mock
|
||||||
|
private UdfFuncMapper udfFuncMapper;
|
||||||
|
@Mock
|
||||||
|
private UDFUserMapper udfUserMapper;
|
||||||
|
|
||||||
|
|
||||||
|
@Before
|
||||||
|
public void setUp() {
|
||||||
|
PowerMockito.mockStatic(PropertyUtils.class);
|
||||||
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void queryUdfFuncListPaging(){
|
public void testCreateUdfFunction(){
|
||||||
|
|
||||||
User loginUser = new User();
|
PowerMockito.when(PropertyUtils.getResUploadStartupState()).thenReturn(false);
|
||||||
loginUser.setId(-1);
|
//hdfs not start
|
||||||
loginUser.setUserType(UserType.GENERAL_USER);
|
Result result = udfFuncService.createUdfFunction(getLoginUser(), "UdfFuncServiceTest", "org.apache.dolphinscheduler.api.service.UdfFuncServiceTest", "String", "UdfFuncServiceTest", "UdfFuncServiceTest", UdfType.HIVE, Integer.MAX_VALUE);
|
||||||
|
logger.info(result.toString());
|
||||||
|
Assert.assertEquals(Status.HDFS_NOT_STARTUP.getMsg(),result.getMsg());
|
||||||
|
//resource not exist
|
||||||
|
PowerMockito.when(PropertyUtils.getResUploadStartupState()).thenReturn(true);
|
||||||
|
result = udfFuncService.createUdfFunction(getLoginUser(), "UdfFuncServiceTest", "org.apache.dolphinscheduler.api.service.UdfFuncServiceTest", "String", "UdfFuncServiceTest", "UdfFuncServiceTest", UdfType.HIVE, Integer.MAX_VALUE);
|
||||||
|
logger.info(result.toString());
|
||||||
|
Assert.assertEquals(Status.RESOURCE_NOT_EXIST.getMsg(),result.getMsg());
|
||||||
|
// success
|
||||||
|
PowerMockito.when(resourceMapper.selectById(1)).thenReturn(getResource());
|
||||||
|
result = udfFuncService.createUdfFunction(getLoginUser(), "UdfFuncServiceTest", "org.apache.dolphinscheduler.api.service.UdfFuncServiceTest", "String", "UdfFuncServiceTest", "UdfFuncServiceTest", UdfType.HIVE, 1);
|
||||||
|
logger.info(result.toString());
|
||||||
|
Assert.assertEquals(Status.SUCCESS.getMsg(),result.getMsg());
|
||||||
|
}
|
||||||
|
|
||||||
Map<String, Object> map = udfFuncService.queryUdfFuncListPaging(loginUser, "", 1, 10);
|
@Test
|
||||||
Assert.assertEquals(Status.SUCCESS, map.get(Constants.STATUS));
|
public void testQueryUdfFuncDetail(){
|
||||||
|
|
||||||
PageInfo pageInfo = (PageInfo) map.get("data");
|
PowerMockito.when(udfFuncMapper.selectById(1)).thenReturn(getUdfFunc());
|
||||||
logger.info(pageInfo.getLists().toString());
|
//resource not exist
|
||||||
|
Map<String, Object> result = udfFuncService.queryUdfFuncDetail(2);
|
||||||
|
logger.info(result.toString());
|
||||||
|
Assert.assertEquals(Status.RESOURCE_NOT_EXIST,result.get(Constants.STATUS));
|
||||||
|
// success
|
||||||
|
result = udfFuncService.queryUdfFuncDetail(1);
|
||||||
|
logger.info(result.toString());
|
||||||
|
Assert.assertEquals(Status.SUCCESS,result.get(Constants.STATUS));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testUpdateUdfFunc(){
|
||||||
|
|
||||||
|
PowerMockito.when(PropertyUtils.getResUploadStartupState()).thenReturn(false);
|
||||||
|
PowerMockito.when(udfFuncMapper.selectUdfById(1)).thenReturn(getUdfFunc());
|
||||||
|
PowerMockito.when(resourceMapper.selectById(1)).thenReturn(getResource());
|
||||||
|
|
||||||
|
//UDF_FUNCTION_NOT_EXIST
|
||||||
|
Map<String, Object> result = udfFuncService.updateUdfFunc(12, "UdfFuncServiceTest", "org.apache.dolphinscheduler.api.service.UdfFuncServiceTest", "String", "UdfFuncServiceTest", "UdfFuncServiceTest", UdfType.HIVE, 1);
|
||||||
|
logger.info(result.toString());
|
||||||
|
Assert.assertEquals(Status.UDF_FUNCTION_NOT_EXIST,result.get(Constants.STATUS));
|
||||||
|
|
||||||
|
//HDFS_NOT_STARTUP
|
||||||
|
result = udfFuncService.updateUdfFunc(1, "UdfFuncServiceTest", "org.apache.dolphinscheduler.api.service.UdfFuncServiceTest", "String", "UdfFuncServiceTest", "UdfFuncServiceTest", UdfType.HIVE, 1);
|
||||||
|
logger.info(result.toString());
|
||||||
|
Assert.assertEquals(Status.HDFS_NOT_STARTUP,result.get(Constants.STATUS));
|
||||||
|
|
||||||
|
//RESOURCE_NOT_EXIST
|
||||||
|
PowerMockito.when(udfFuncMapper.selectUdfById(11)).thenReturn(getUdfFunc());
|
||||||
|
PowerMockito.when(PropertyUtils.getResUploadStartupState()).thenReturn(true);
|
||||||
|
result = udfFuncService.updateUdfFunc(11, "UdfFuncServiceTest", "org.apache.dolphinscheduler.api.service.UdfFuncServiceTest", "String", "UdfFuncServiceTest", "UdfFuncServiceTest", UdfType.HIVE, 12);
|
||||||
|
logger.info(result.toString());
|
||||||
|
Assert.assertEquals(Status.RESOURCE_NOT_EXIST,result.get(Constants.STATUS));
|
||||||
|
|
||||||
|
//success
|
||||||
|
result = udfFuncService.updateUdfFunc(11, "UdfFuncServiceTest", "org.apache.dolphinscheduler.api.service.UdfFuncServiceTest", "String", "UdfFuncServiceTest", "UdfFuncServiceTest", UdfType.HIVE, 1);
|
||||||
|
logger.info(result.toString());
|
||||||
|
Assert.assertEquals(Status.SUCCESS,result.get(Constants.STATUS));
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testQueryUdfFuncListPaging(){
|
||||||
|
|
||||||
|
IPage<UdfFunc> page = new Page<>(1,10);
|
||||||
|
page.setTotal(1L);
|
||||||
|
page.setRecords(getList());
|
||||||
|
Mockito.when(udfFuncMapper.queryUdfFuncPaging(Mockito.any(Page.class), Mockito.eq(0),Mockito.eq("test"))).thenReturn(page);
|
||||||
|
Map<String, Object> result = udfFuncService.queryUdfFuncListPaging(getLoginUser(),"test",1,10);
|
||||||
|
logger.info(result.toString());
|
||||||
|
PageInfo pageInfo = (PageInfo) result.get(Constants.DATA_LIST);
|
||||||
|
Assert.assertTrue(CollectionUtils.isNotEmpty(pageInfo.getLists()));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testQueryResourceList(){
|
||||||
|
Mockito.when(udfFuncMapper.getUdfFuncByType(1, 1)).thenReturn(getList());
|
||||||
|
Map<String, Object> result = udfFuncService.queryResourceList(getLoginUser(),1);
|
||||||
|
logger.info(result.toString());
|
||||||
|
Assert.assertEquals(Status.SUCCESS,result.get(Constants.STATUS));
|
||||||
|
List<UdfFunc> udfFuncList = (List<UdfFunc>) result.get(Constants.DATA_LIST);
|
||||||
|
Assert.assertTrue(CollectionUtils.isNotEmpty(udfFuncList));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testDelete(){
|
||||||
|
Result result= udfFuncService.delete(122);
|
||||||
|
logger.info(result.toString());
|
||||||
|
Assert.assertEquals(Status.SUCCESS.getMsg(),result.getMsg());
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testVerifyUdfFuncByName(){
|
||||||
|
|
||||||
|
//success
|
||||||
|
Mockito.when(udfFuncMapper.queryUdfByIdStr(null, "UdfFuncServiceTest")).thenReturn(getList());
|
||||||
|
Result result = udfFuncService.verifyUdfFuncByName("test");
|
||||||
|
logger.info(result.toString());
|
||||||
|
Assert.assertEquals(Status.SUCCESS.getMsg(),result.getMsg());
|
||||||
|
//exist
|
||||||
|
result = udfFuncService.verifyUdfFuncByName("UdfFuncServiceTest");
|
||||||
|
logger.info(result.toString());
|
||||||
|
Assert.assertEquals(Status.UDF_FUNCTION_EXISTS.getMsg(),result.getMsg());
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* create admin user
|
||||||
|
* @return
|
||||||
|
*/
|
||||||
|
private User getLoginUser(){
|
||||||
|
|
||||||
|
User loginUser = new User();
|
||||||
|
loginUser.setUserType(UserType.ADMIN_USER);
|
||||||
|
loginUser.setId(1);
|
||||||
|
return loginUser;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* get resourceId
|
||||||
|
*/
|
||||||
|
private Resource getResource(){
|
||||||
|
|
||||||
|
Resource resource = new Resource();
|
||||||
|
resource.setId(1);
|
||||||
|
resource.setAlias("test");
|
||||||
|
return resource;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
private List<UdfFunc> getList(){
|
||||||
|
List<UdfFunc> udfFuncList = new ArrayList<>();
|
||||||
|
udfFuncList.add(getUdfFunc());
|
||||||
|
return udfFuncList;
|
||||||
|
}
|
||||||
|
/**
|
||||||
|
* get UdfFunc id
|
||||||
|
*/
|
||||||
|
private UdfFunc getUdfFunc(){
|
||||||
|
UdfFunc udfFunc = new UdfFunc();
|
||||||
|
udfFunc.setFuncName("UdfFuncServiceTest");
|
||||||
|
udfFunc.setClassName("org.apache.dolphinscheduler.api.service.UdfFuncServiceTest");
|
||||||
|
udfFunc.setResourceId(0);
|
||||||
|
udfFunc.setResourceName("UdfFuncServiceTest");
|
||||||
|
udfFunc.setCreateTime(new Date());
|
||||||
|
udfFunc.setDatabase("database");
|
||||||
|
udfFunc.setUpdateTime(new Date());
|
||||||
|
udfFunc.setType(UdfType.HIVE);
|
||||||
|
return udfFunc;
|
||||||
|
}
|
||||||
}
|
}
|
@ -1,699 +0,0 @@
|
|||||||
/*
|
|
||||||
* Licensed to the Apache Software Foundation (ASF) under one or more
|
|
||||||
* contributor license agreements. See the NOTICE file distributed with
|
|
||||||
* this work for additional information regarding copyright ownership.
|
|
||||||
* The ASF licenses this file to You under the Apache License, Version 2.0
|
|
||||||
* (the "License"); you may not use this file except in compliance with
|
|
||||||
* the License. You may obtain a copy of the License at
|
|
||||||
*
|
|
||||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
*
|
|
||||||
* Unless required by applicable law or agreed to in writing, software
|
|
||||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
* See the License for the specific language governing permissions and
|
|
||||||
* limitations under the License.
|
|
||||||
*/
|
|
||||||
package org.apache.dolphinscheduler.common.utils;
|
|
||||||
|
|
||||||
import org.slf4j.Logger;
|
|
||||||
import org.slf4j.LoggerFactory;
|
|
||||||
|
|
||||||
import java.math.BigDecimal;
|
|
||||||
import java.math.BigInteger;
|
|
||||||
import java.nio.ByteBuffer;
|
|
||||||
import java.nio.charset.StandardCharsets;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Utility class that handles Bytes
|
|
||||||
*/
|
|
||||||
public class Bytes {
|
|
||||||
|
|
||||||
private static final Logger logger = LoggerFactory.getLogger(Bytes.class);
|
|
||||||
public static final String UTF8_ENCODING = "UTF-8";
|
|
||||||
//An empty instance.
|
|
||||||
public static final byte [] EMPTY_BYTE_ARRAY = new byte [0];
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Size of int in bytes
|
|
||||||
*/
|
|
||||||
public static final int SIZEOF_INT = Integer.SIZE / Byte.SIZE;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Size of long in bytes
|
|
||||||
*/
|
|
||||||
public static final int SIZEOF_LONG = Long.SIZE / Byte.SIZE;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Size of short in bytes
|
|
||||||
*/
|
|
||||||
public static final int SIZEOF_SHORT = Short.SIZE / Byte.SIZE;
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Put bytes at the specified byte array position.
|
|
||||||
* @param tgtBytes the byte array
|
|
||||||
* @param tgtOffset position in the array
|
|
||||||
* @param srcBytes array to write out
|
|
||||||
* @param srcOffset source offset
|
|
||||||
* @param srcLength source length
|
|
||||||
* @return incremented offset
|
|
||||||
*/
|
|
||||||
public static int putBytes(byte[] tgtBytes, int tgtOffset, byte[] srcBytes,
|
|
||||||
int srcOffset, int srcLength) {
|
|
||||||
System.arraycopy(srcBytes, srcOffset, tgtBytes, tgtOffset, srcLength);
|
|
||||||
return tgtOffset + srcLength;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Write a single byte out to the specified byte array position.
|
|
||||||
* @param bytes the byte array
|
|
||||||
* @param offset position in the array
|
|
||||||
* @param b byte to write out
|
|
||||||
* @return incremented offset
|
|
||||||
*/
|
|
||||||
public static int putByte(byte[] bytes, int offset, byte b) {
|
|
||||||
bytes[offset] = b;
|
|
||||||
return offset + 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns a new byte array, copied from the passed ByteBuffer.
|
|
||||||
* @param bb A ByteBuffer
|
|
||||||
* @return the byte array
|
|
||||||
*/
|
|
||||||
public static byte[] toBytes(ByteBuffer bb) {
|
|
||||||
int length = bb.limit();
|
|
||||||
byte [] result = new byte[length];
|
|
||||||
System.arraycopy(bb.array(), bb.arrayOffset(), result, 0, length);
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @param b Presumed UTF-8 encoded byte array.
|
|
||||||
* @return String made from <code>b</code>
|
|
||||||
*/
|
|
||||||
public static String toString(final byte [] b) {
|
|
||||||
if (b == null) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
return toString(b, 0, b.length);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Joins two byte arrays together using a separator.
|
|
||||||
* @param b1 The first byte array.
|
|
||||||
* @param sep The separator to use.
|
|
||||||
* @param b2 The second byte array.
|
|
||||||
* @return two byte arrays together using a separator.
|
|
||||||
*/
|
|
||||||
public static String toString(final byte [] b1,
|
|
||||||
String sep,
|
|
||||||
final byte [] b2) {
|
|
||||||
return toString(b1, 0, b1.length) + sep + toString(b2, 0, b2.length);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* This method will convert utf8 encoded bytes into a string. If
|
|
||||||
* an UnsupportedEncodingException occurs, this method will eat it
|
|
||||||
* and return null instead.
|
|
||||||
*
|
|
||||||
* @param b Presumed UTF-8 encoded byte array.
|
|
||||||
* @param off offset into array
|
|
||||||
* @param len length of utf-8 sequence
|
|
||||||
* @return String made from <code>b</code> or null
|
|
||||||
*/
|
|
||||||
public static String toString(final byte [] b, int off, int len) {
|
|
||||||
if (b == null) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
if (len == 0) {
|
|
||||||
return "";
|
|
||||||
}
|
|
||||||
return new String(b, off, len, StandardCharsets.UTF_8);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Converts a string to a UTF-8 byte array.
|
|
||||||
* @param s string
|
|
||||||
* @return the byte array
|
|
||||||
*/
|
|
||||||
public static byte[] toBytes(String s) {
|
|
||||||
return s.getBytes(StandardCharsets.UTF_8);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Convert a boolean to a byte array. True becomes -1
|
|
||||||
* and false becomes 0.
|
|
||||||
*
|
|
||||||
* @param b value
|
|
||||||
* @return <code>b</code> encoded in a byte array.
|
|
||||||
*/
|
|
||||||
public static byte [] toBytes(final boolean b) {
|
|
||||||
return new byte[] { b ? (byte) -1 : (byte) 0 };
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Reverses {@link #toBytes(boolean)}
|
|
||||||
* @param b array
|
|
||||||
* @return True or false.
|
|
||||||
*/
|
|
||||||
public static boolean toBoolean(final byte [] b) {
|
|
||||||
if (b.length != 1) {
|
|
||||||
throw new IllegalArgumentException("Array has wrong size: " + b.length);
|
|
||||||
}
|
|
||||||
return b[0] != (byte) 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Convert a long value to a byte array using big-endian.
|
|
||||||
*
|
|
||||||
* @param val value to convert
|
|
||||||
* @return the byte array
|
|
||||||
*/
|
|
||||||
public static byte[] toBytes(long val) {
|
|
||||||
byte [] b = new byte[8];
|
|
||||||
for (int i = 7; i > 0; i--) {
|
|
||||||
b[i] = (byte) val;
|
|
||||||
val >>>= 8;
|
|
||||||
}
|
|
||||||
b[0] = (byte) val;
|
|
||||||
return b;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Converts a byte array to a long value. Reverses
|
|
||||||
* {@link #toBytes(long)}
|
|
||||||
* @param bytes array
|
|
||||||
* @return the long value
|
|
||||||
*/
|
|
||||||
public static long toLong(byte[] bytes) {
|
|
||||||
return toLong(bytes, 0, SIZEOF_LONG);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Converts a byte array to a long value. Assumes there will be
|
|
||||||
* {@link #SIZEOF_LONG} bytes available.
|
|
||||||
*
|
|
||||||
* @param bytes bytes
|
|
||||||
* @param offset offset
|
|
||||||
* @return the long value
|
|
||||||
*/
|
|
||||||
public static long toLong(byte[] bytes, int offset) {
|
|
||||||
return toLong(bytes, offset, SIZEOF_LONG);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Converts a byte array to a long value.
|
|
||||||
*
|
|
||||||
* @param bytes array of bytes
|
|
||||||
* @param offset offset into array
|
|
||||||
* @param length length of data (must be {@link #SIZEOF_LONG})
|
|
||||||
* @return the long value
|
|
||||||
* @throws IllegalArgumentException if length is not {@link #SIZEOF_LONG} or
|
|
||||||
* if there's not enough room in the array at the offset indicated.
|
|
||||||
*/
|
|
||||||
public static long toLong(byte[] bytes, int offset, final int length) {
|
|
||||||
if (length != SIZEOF_LONG || offset + length > bytes.length) {
|
|
||||||
throw explainWrongLengthOrOffset(bytes, offset, length, SIZEOF_LONG);
|
|
||||||
}
|
|
||||||
long l = 0;
|
|
||||||
for(int i = offset; i < offset + length; i++) {
|
|
||||||
l <<= 8;
|
|
||||||
l ^= bytes[i] & 0xFF;
|
|
||||||
}
|
|
||||||
return l;
|
|
||||||
}
|
|
||||||
|
|
||||||
private static IllegalArgumentException
|
|
||||||
explainWrongLengthOrOffset(final byte[] bytes,
|
|
||||||
final int offset,
|
|
||||||
final int length,
|
|
||||||
final int expectedLength) {
|
|
||||||
String reason;
|
|
||||||
if (length != expectedLength) {
|
|
||||||
reason = "Wrong length: " + length + ", expected " + expectedLength;
|
|
||||||
} else {
|
|
||||||
reason = "offset (" + offset + ") + length (" + length + ") exceed the"
|
|
||||||
+ " capacity of the array: " + bytes.length;
|
|
||||||
}
|
|
||||||
return new IllegalArgumentException(reason);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Put a long value out to the specified byte array position.
|
|
||||||
* @param bytes the byte array
|
|
||||||
* @param offset position in the array
|
|
||||||
* @param val long to write out
|
|
||||||
* @return incremented offset
|
|
||||||
* @throws IllegalArgumentException if the byte array given doesn't have
|
|
||||||
* enough room at the offset specified.
|
|
||||||
*/
|
|
||||||
public static int putLong(byte[] bytes, int offset, long val) {
|
|
||||||
if (bytes.length - offset < SIZEOF_LONG) {
|
|
||||||
throw new IllegalArgumentException("Not enough room to put a long at"
|
|
||||||
+ " offset " + offset + " in a " + bytes.length + " byte array");
|
|
||||||
}
|
|
||||||
for(int i = offset + 7; i > offset; i--) {
|
|
||||||
bytes[i] = (byte) val;
|
|
||||||
val >>>= 8;
|
|
||||||
}
|
|
||||||
bytes[offset] = (byte) val;
|
|
||||||
return offset + SIZEOF_LONG;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Presumes float encoded as IEEE 754 floating-point "single format"
|
|
||||||
* @param bytes byte array
|
|
||||||
* @return Float made from passed byte array.
|
|
||||||
*/
|
|
||||||
public static float toFloat(byte [] bytes) {
|
|
||||||
return toFloat(bytes, 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Presumes float encoded as IEEE 754 floating-point "single format"
|
|
||||||
* @param bytes array to convert
|
|
||||||
* @param offset offset into array
|
|
||||||
* @return Float made from passed byte array.
|
|
||||||
*/
|
|
||||||
public static float toFloat(byte [] bytes, int offset) {
|
|
||||||
return Float.intBitsToFloat(toInt(bytes, offset, SIZEOF_INT));
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @param bytes byte array
|
|
||||||
* @param offset offset to write to
|
|
||||||
* @param f float value
|
|
||||||
* @return New offset in <code>bytes</code>
|
|
||||||
*/
|
|
||||||
public static int putFloat(byte [] bytes, int offset, float f) {
|
|
||||||
return putInt(bytes, offset, Float.floatToRawIntBits(f));
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @param f float value
|
|
||||||
* @return the float represented as byte []
|
|
||||||
*/
|
|
||||||
public static byte [] toBytes(final float f) {
|
|
||||||
// Encode it as int
|
|
||||||
return Bytes.toBytes(Float.floatToRawIntBits(f));
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @param bytes byte array
|
|
||||||
* @return Return double made from passed bytes.
|
|
||||||
*/
|
|
||||||
public static double toDouble(final byte [] bytes) {
|
|
||||||
return toDouble(bytes, 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @param bytes byte array
|
|
||||||
* @param offset offset where double is
|
|
||||||
* @return Return double made from passed bytes.
|
|
||||||
*/
|
|
||||||
public static double toDouble(final byte [] bytes, final int offset) {
|
|
||||||
return Double.longBitsToDouble(toLong(bytes, offset, SIZEOF_LONG));
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @param bytes byte array
|
|
||||||
* @param offset offset to write to
|
|
||||||
* @param d value
|
|
||||||
* @return New offset into array <code>bytes</code>
|
|
||||||
*/
|
|
||||||
public static int putDouble(byte [] bytes, int offset, double d) {
|
|
||||||
return putLong(bytes, offset, Double.doubleToLongBits(d));
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Serialize a double as the IEEE 754 double format output. The resultant
|
|
||||||
* array will be 8 bytes long.
|
|
||||||
*
|
|
||||||
* @param d value
|
|
||||||
* @return the double represented as byte []
|
|
||||||
*/
|
|
||||||
public static byte [] toBytes(final double d) {
|
|
||||||
// Encode it as a long
|
|
||||||
return Bytes.toBytes(Double.doubleToRawLongBits(d));
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Convert an int value to a byte array
|
|
||||||
* @param val value
|
|
||||||
* @return the byte array
|
|
||||||
*/
|
|
||||||
public static byte[] toBytes(int val) {
|
|
||||||
byte [] b = new byte[4];
|
|
||||||
for(int i = 3; i > 0; i--) {
|
|
||||||
b[i] = (byte) val;
|
|
||||||
val >>>= 8;
|
|
||||||
}
|
|
||||||
b[0] = (byte) val;
|
|
||||||
return b;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Converts a byte array to an int value
|
|
||||||
* @param bytes byte array
|
|
||||||
* @return the int value
|
|
||||||
*/
|
|
||||||
public static int toInt(byte[] bytes) {
|
|
||||||
return toInt(bytes, 0, SIZEOF_INT);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Converts a byte array to an int value
|
|
||||||
* @param bytes byte array
|
|
||||||
* @param offset offset into array
|
|
||||||
* @return the int value
|
|
||||||
*/
|
|
||||||
public static int toInt(byte[] bytes, int offset) {
|
|
||||||
return toInt(bytes, offset, SIZEOF_INT);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Converts a byte array to an int value
|
|
||||||
* @param bytes byte array
|
|
||||||
* @param offset offset into array
|
|
||||||
* @param length length of int (has to be {@link #SIZEOF_INT})
|
|
||||||
* @return the int value
|
|
||||||
* @throws IllegalArgumentException if length is not {@link #SIZEOF_INT} or
|
|
||||||
* if there's not enough room in the array at the offset indicated.
|
|
||||||
*/
|
|
||||||
public static int toInt(byte[] bytes, int offset, final int length) {
|
|
||||||
if (length != SIZEOF_INT || offset + length > bytes.length) {
|
|
||||||
throw explainWrongLengthOrOffset(bytes, offset, length, SIZEOF_INT);
|
|
||||||
}
|
|
||||||
int n = 0;
|
|
||||||
for(int i = offset; i < (offset + length); i++) {
|
|
||||||
n <<= 8;
|
|
||||||
n ^= bytes[i] & 0xFF;
|
|
||||||
}
|
|
||||||
return n;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Put an int value out to the specified byte array position.
|
|
||||||
* @param bytes the byte array
|
|
||||||
* @param offset position in the array
|
|
||||||
* @param val int to write out
|
|
||||||
* @return incremented offset
|
|
||||||
* @throws IllegalArgumentException if the byte array given doesn't have
|
|
||||||
* enough room at the offset specified.
|
|
||||||
*/
|
|
||||||
public static int putInt(byte[] bytes, int offset, int val) {
|
|
||||||
if (bytes.length - offset < SIZEOF_INT) {
|
|
||||||
throw new IllegalArgumentException("Not enough room to put an int at"
|
|
||||||
+ " offset " + offset + " in a " + bytes.length + " byte array");
|
|
||||||
}
|
|
||||||
for(int i= offset + 3; i > offset; i--) {
|
|
||||||
bytes[i] = (byte) val;
|
|
||||||
val >>>= 8;
|
|
||||||
}
|
|
||||||
bytes[offset] = (byte) val;
|
|
||||||
return offset + SIZEOF_INT;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Convert a short value to a byte array of {@link #SIZEOF_SHORT} bytes long.
|
|
||||||
* @param val value
|
|
||||||
* @return the byte array
|
|
||||||
*/
|
|
||||||
public static byte[] toBytes(short val) {
|
|
||||||
byte[] b = new byte[SIZEOF_SHORT];
|
|
||||||
b[1] = (byte) val;
|
|
||||||
val >>= 8;
|
|
||||||
b[0] = (byte) val;
|
|
||||||
return b;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Converts a byte array to a short value
|
|
||||||
* @param bytes byte array
|
|
||||||
* @return the short value
|
|
||||||
*/
|
|
||||||
public static short toShort(byte[] bytes) {
|
|
||||||
return toShort(bytes, 0, SIZEOF_SHORT);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Converts a byte array to a short value
|
|
||||||
* @param bytes byte array
|
|
||||||
* @param offset offset into array
|
|
||||||
* @return the short value
|
|
||||||
*/
|
|
||||||
public static short toShort(byte[] bytes, int offset) {
|
|
||||||
return toShort(bytes, offset, SIZEOF_SHORT);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Converts a byte array to a short value
|
|
||||||
* @param bytes byte array
|
|
||||||
* @param offset offset into array
|
|
||||||
* @param length length, has to be {@link #SIZEOF_SHORT}
|
|
||||||
* @return the short value
|
|
||||||
* @throws IllegalArgumentException if length is not {@link #SIZEOF_SHORT}
|
|
||||||
* or if there's not enough room in the array at the offset indicated.
|
|
||||||
*/
|
|
||||||
public static short toShort(byte[] bytes, int offset, final int length) {
|
|
||||||
if (length != SIZEOF_SHORT || offset + length > bytes.length) {
|
|
||||||
throw explainWrongLengthOrOffset(bytes, offset, length, SIZEOF_SHORT);
|
|
||||||
}
|
|
||||||
short n = 0;
|
|
||||||
n ^= bytes[offset] & 0xFF;
|
|
||||||
n <<= 8;
|
|
||||||
n ^= bytes[offset+1] & 0xFF;
|
|
||||||
return n;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* This method will get a sequence of bytes from pos to limit,
|
|
||||||
* but will restore pos after.
|
|
||||||
* @param buf buffer
|
|
||||||
* @return byte array
|
|
||||||
*/
|
|
||||||
public static byte[] getBytes(ByteBuffer buf) {
|
|
||||||
int savedPos = buf.position();
|
|
||||||
byte [] newBytes = new byte[buf.remaining()];
|
|
||||||
buf.get(newBytes);
|
|
||||||
buf.position(savedPos);
|
|
||||||
return newBytes;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Put a short value out to the specified byte array position.
|
|
||||||
* @param bytes the byte array
|
|
||||||
* @param offset position in the array
|
|
||||||
* @param val short to write out
|
|
||||||
* @return incremented offset
|
|
||||||
* @throws IllegalArgumentException if the byte array given doesn't have
|
|
||||||
* enough room at the offset specified.
|
|
||||||
*/
|
|
||||||
public static int putShort(byte[] bytes, int offset, short val) {
|
|
||||||
if (bytes.length - offset < SIZEOF_SHORT) {
|
|
||||||
throw new IllegalArgumentException("Not enough room to put a short at"
|
|
||||||
+ " offset " + offset + " in a " + bytes.length + " byte array");
|
|
||||||
}
|
|
||||||
bytes[offset+1] = (byte) val;
|
|
||||||
val >>= 8;
|
|
||||||
bytes[offset] = (byte) val;
|
|
||||||
return offset + SIZEOF_SHORT;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Convert a BigDecimal value to a byte array
|
|
||||||
*
|
|
||||||
* @param val value
|
|
||||||
* @return the byte array
|
|
||||||
*/
|
|
||||||
public static byte[] toBytes(BigDecimal val) {
|
|
||||||
byte[] valueBytes = val.unscaledValue().toByteArray();
|
|
||||||
byte[] result = new byte[valueBytes.length + SIZEOF_INT];
|
|
||||||
int offset = putInt(result, 0, val.scale());
|
|
||||||
putBytes(result, offset, valueBytes, 0, valueBytes.length);
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Converts a byte array to a BigDecimal
|
|
||||||
*
|
|
||||||
* @param bytes bytes array
|
|
||||||
* @return the char value
|
|
||||||
*/
|
|
||||||
public static BigDecimal toBigDecimal(byte[] bytes) {
|
|
||||||
return toBigDecimal(bytes, 0, bytes.length);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Converts a byte array to a BigDecimal value
|
|
||||||
*
|
|
||||||
* @param bytes bytes array
|
|
||||||
* @param offset offset
|
|
||||||
* @param length length
|
|
||||||
* @return the char value
|
|
||||||
*/
|
|
||||||
public static BigDecimal toBigDecimal(byte[] bytes, int offset, final int length) {
|
|
||||||
if (bytes == null || length < SIZEOF_INT + 1 ||
|
|
||||||
(offset + length > bytes.length)) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
int scale = toInt(bytes, offset);
|
|
||||||
byte[] tcBytes = new byte[length - SIZEOF_INT];
|
|
||||||
System.arraycopy(bytes, offset + SIZEOF_INT, tcBytes, 0, length - SIZEOF_INT);
|
|
||||||
return new BigDecimal(new BigInteger(tcBytes), scale);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Put a BigDecimal value out to the specified byte array position.
|
|
||||||
*
|
|
||||||
* @param bytes the byte array
|
|
||||||
* @param offset position in the array
|
|
||||||
* @param val BigDecimal to write out
|
|
||||||
* @return incremented offset
|
|
||||||
*/
|
|
||||||
public static int putBigDecimal(byte[] bytes, int offset, BigDecimal val) {
|
|
||||||
if (bytes == null) {
|
|
||||||
return offset;
|
|
||||||
}
|
|
||||||
|
|
||||||
byte[] valueBytes = val.unscaledValue().toByteArray();
|
|
||||||
byte[] result = new byte[valueBytes.length + SIZEOF_INT];
|
|
||||||
offset = putInt(result, offset, val.scale());
|
|
||||||
return putBytes(result, offset, valueBytes, 0, valueBytes.length);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @param a lower half
|
|
||||||
* @param b upper half
|
|
||||||
* @return New array that has a in lower half and b in upper half.
|
|
||||||
*/
|
|
||||||
public static byte [] add(final byte [] a, final byte [] b) {
|
|
||||||
return add(a, b, EMPTY_BYTE_ARRAY);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @param a first third
|
|
||||||
* @param b second third
|
|
||||||
* @param c third third
|
|
||||||
* @return New array made from a, b and c
|
|
||||||
*/
|
|
||||||
public static byte [] add(final byte [] a, final byte [] b, final byte [] c) {
|
|
||||||
byte [] result = new byte[a.length + b.length + c.length];
|
|
||||||
System.arraycopy(a, 0, result, 0, a.length);
|
|
||||||
System.arraycopy(b, 0, result, a.length, b.length);
|
|
||||||
System.arraycopy(c, 0, result, a.length + b.length, c.length);
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @param a array
|
|
||||||
* @param length amount of bytes to grab
|
|
||||||
* @return First <code>length</code> bytes from <code>a</code>
|
|
||||||
*/
|
|
||||||
public static byte [] head(final byte [] a, final int length) {
|
|
||||||
if (a.length < length) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
byte [] result = new byte[length];
|
|
||||||
System.arraycopy(a, 0, result, 0, length);
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @param a array
|
|
||||||
* @param length amount of bytes to snarf
|
|
||||||
* @return Last <code>length</code> bytes from <code>a</code>
|
|
||||||
*/
|
|
||||||
public static byte [] tail(final byte [] a, final int length) {
|
|
||||||
if (a.length < length) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
byte [] result = new byte[length];
|
|
||||||
System.arraycopy(a, a.length - length, result, 0, length);
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @param a array
|
|
||||||
* @param length new array size
|
|
||||||
* @return Value in <code>a</code> plus <code>length</code> prepended 0 bytes
|
|
||||||
*/
|
|
||||||
public static byte [] padHead(final byte [] a, final int length) {
|
|
||||||
byte[] padding = getPadding(length);
|
|
||||||
return add(padding,a);
|
|
||||||
}
|
|
||||||
|
|
||||||
private static byte[] getPadding(int length) {
|
|
||||||
byte[] padding = new byte[length];
|
|
||||||
for (int i = 0; i < length; i++) {
|
|
||||||
padding[i] = 0;
|
|
||||||
}
|
|
||||||
return padding;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @param a array
|
|
||||||
* @param length new array size
|
|
||||||
* @return Value in <code>a</code> plus <code>length</code> appended 0 bytes
|
|
||||||
*/
|
|
||||||
public static byte [] padTail(final byte [] a, final int length) {
|
|
||||||
byte[] padding = getPadding(length);
|
|
||||||
return add(a,padding);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @param bytes array to hash
|
|
||||||
* @param offset offset to start from
|
|
||||||
* @param length length to hash
|
|
||||||
* @return hash code
|
|
||||||
* */
|
|
||||||
public static int hashCode(byte[] bytes, int offset, int length) {
|
|
||||||
int hash = 1;
|
|
||||||
for (int i = offset; i < offset + length; i++) {
|
|
||||||
hash = (31 * hash) + (int) bytes[i];
|
|
||||||
}
|
|
||||||
return hash;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @param t operands
|
|
||||||
* @return Array of byte arrays made from passed array of Text
|
|
||||||
*/
|
|
||||||
public static byte [][] toByteArrays(final String [] t) {
|
|
||||||
byte [][] result = new byte[t.length][];
|
|
||||||
for (int i = 0; i < t.length; i++) {
|
|
||||||
result[i] = Bytes.toBytes(t[i]);
|
|
||||||
}
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @param column operand
|
|
||||||
* @return A byte array of a byte array where first and only entry is
|
|
||||||
* <code>column</code>
|
|
||||||
*/
|
|
||||||
public static byte [][] toByteArrays(final String column) {
|
|
||||||
return toByteArrays(toBytes(column));
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @param column operand
|
|
||||||
* @return A byte array of a byte array where first and only entry is
|
|
||||||
* <code>column</code>
|
|
||||||
*/
|
|
||||||
public static byte [][] toByteArrays(final byte [] column) {
|
|
||||||
byte [][] result = new byte[1][];
|
|
||||||
result[0] = column;
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
}
|
|
@ -0,0 +1,119 @@
|
|||||||
|
/*
|
||||||
|
* Licensed to the Apache Software Foundation (ASF) under one or more
|
||||||
|
* contributor license agreements. See the NOTICE file distributed with
|
||||||
|
* this work for additional information regarding copyright ownership.
|
||||||
|
* The ASF licenses this file to You under the Apache License, Version 2.0
|
||||||
|
* (the "License"); you may not use this file except in compliance with
|
||||||
|
* the License. You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
package org.apache.dolphinscheduler.common.utils;
|
||||||
|
|
||||||
|
import org.junit.Assert;
|
||||||
|
import org.junit.Test;
|
||||||
|
import org.junit.runner.RunWith;
|
||||||
|
import org.mockito.Mockito;
|
||||||
|
import org.powermock.api.mockito.PowerMockito;
|
||||||
|
import org.powermock.core.classloader.annotations.PrepareForTest;
|
||||||
|
import org.powermock.modules.junit4.PowerMockRunner;
|
||||||
|
import org.slf4j.LoggerFactory;
|
||||||
|
|
||||||
|
import java.io.File;
|
||||||
|
import java.io.FileNotFoundException;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.Arrays;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
@RunWith(PowerMockRunner.class)
|
||||||
|
@PrepareForTest({ LoggerFactory.class, FileUtils.class })
|
||||||
|
public class SchemaUtilsTest {
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testReplaceBlank() {
|
||||||
|
Assert.assertEquals("abc", SchemaUtils.replaceBlank(" abc"));
|
||||||
|
Assert.assertEquals("abc", SchemaUtils.replaceBlank("abc "));
|
||||||
|
Assert.assertEquals("abc", SchemaUtils.replaceBlank("a b c"));
|
||||||
|
Assert.assertEquals("abc", SchemaUtils.replaceBlank("a b c"));
|
||||||
|
Assert.assertEquals("", SchemaUtils.replaceBlank(" "));
|
||||||
|
Assert.assertEquals("", SchemaUtils.replaceBlank(null));
|
||||||
|
Assert.assertEquals("我怕的你", SchemaUtils.replaceBlank("我怕的 你"));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGetSoftVersion() {
|
||||||
|
// file not found
|
||||||
|
try {
|
||||||
|
SchemaUtils.getSoftVersion();
|
||||||
|
} catch (RuntimeException e) {
|
||||||
|
Assert.assertEquals("Failed to get the product version description file. The file could not be found",
|
||||||
|
e.getMessage());
|
||||||
|
}
|
||||||
|
|
||||||
|
// file exists, fmt is invalid
|
||||||
|
FileUtils.writeContent2File("32432423", "sql/soft_version");
|
||||||
|
Assert.assertEquals("32432423", SchemaUtils.getSoftVersion());
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testIsAGreatVersion() {
|
||||||
|
// param is null
|
||||||
|
try {
|
||||||
|
SchemaUtils.isAGreatVersion(null, null);
|
||||||
|
} catch (RuntimeException e) {
|
||||||
|
Assert.assertEquals("schemaVersion or version is empty", e.getMessage());
|
||||||
|
}
|
||||||
|
|
||||||
|
// param is ""
|
||||||
|
try {
|
||||||
|
SchemaUtils.isAGreatVersion("", "");
|
||||||
|
} catch (RuntimeException e) {
|
||||||
|
Assert.assertEquals("schemaVersion or version is empty", e.getMessage());
|
||||||
|
}
|
||||||
|
Assert.assertFalse(SchemaUtils.isAGreatVersion("1", "1"));
|
||||||
|
Assert.assertTrue(SchemaUtils.isAGreatVersion("2", "1"));
|
||||||
|
Assert.assertTrue(SchemaUtils.isAGreatVersion("1.1", "1"));
|
||||||
|
Assert.assertTrue(SchemaUtils.isAGreatVersion("1.1", "1.0.1"));
|
||||||
|
Assert.assertFalse(SchemaUtils.isAGreatVersion("1.1", "1.2"));
|
||||||
|
Assert.assertTrue(SchemaUtils.isAGreatVersion("1.1.1", "1.1"));
|
||||||
|
Assert.assertTrue(SchemaUtils.isAGreatVersion("10.1.1", "1.01.100"));
|
||||||
|
try {
|
||||||
|
SchemaUtils.isAGreatVersion("10.1.1", ".1");
|
||||||
|
} catch (Exception e) {
|
||||||
|
Assert.assertNotNull(e);
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
SchemaUtils.isAGreatVersion("a.1.1", "b.1");
|
||||||
|
} catch (Exception e) {
|
||||||
|
Assert.assertNotNull(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGetAllSchemaList() {
|
||||||
|
//normal
|
||||||
|
PowerMockito.mockStatic(FileUtils.class);
|
||||||
|
File[] files = new File[4];
|
||||||
|
files[0] = new File("sql/upgrade/1.2.0_schema");
|
||||||
|
files[1] = new File("sql/upgrade/1.0.1_schema");
|
||||||
|
files[2] = new File("sql/upgrade/1.0.2_schema");
|
||||||
|
files[3] = new File("sql/upgrade/1.1.0_schema");
|
||||||
|
PowerMockito.when(FileUtils.getAllDir("sql/upgrade")).thenReturn(files);
|
||||||
|
List<String> real = SchemaUtils.getAllSchemaList();
|
||||||
|
List<String> expect = Arrays.asList("1.0.1_schema", "1.0.2_schema",
|
||||||
|
"1.1.0_schema", "1.2.0_schema");
|
||||||
|
Assert.assertTrue(CollectionUtils.isEqualCollection(real, expect));
|
||||||
|
|
||||||
|
//normal
|
||||||
|
files = new File[0];
|
||||||
|
PowerMockito.when(FileUtils.getAllDir("sql/upgrade")).thenReturn(files);
|
||||||
|
real = SchemaUtils.getAllSchemaList();
|
||||||
|
Assert.assertNull(real);
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,48 @@
|
|||||||
|
/*
|
||||||
|
* Licensed to the Apache Software Foundation (ASF) under one or more
|
||||||
|
* contributor license agreements. See the NOTICE file distributed with
|
||||||
|
* this work for additional information regarding copyright ownership.
|
||||||
|
* The ASF licenses this file to You under the Apache License, Version 2.0
|
||||||
|
* (the "License"); you may not use this file except in compliance with
|
||||||
|
* the License. You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
package org.apache.dolphinscheduler.common.utils;
|
||||||
|
|
||||||
|
import org.junit.Assert;
|
||||||
|
import org.junit.Test;
|
||||||
|
import org.junit.runner.RunWith;
|
||||||
|
import org.mockito.Mock;
|
||||||
|
import org.mockito.Mockito;
|
||||||
|
import org.powermock.api.mockito.PowerMockito;
|
||||||
|
import org.powermock.core.classloader.annotations.PrepareForTest;
|
||||||
|
import org.powermock.modules.junit4.PowerMockRunner;
|
||||||
|
import org.slf4j.LoggerFactory;
|
||||||
|
|
||||||
|
@RunWith(PowerMockRunner.class)
|
||||||
|
@PrepareForTest(LoggerFactory.class)
|
||||||
|
public class TaskParametersUtilsTest {
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGetParameters() {
|
||||||
|
Assert.assertNull(TaskParametersUtils.getParameters("xx", "ttt"));
|
||||||
|
Assert.assertNull(TaskParametersUtils.getParameters("SHELL", "ttt"));
|
||||||
|
Assert.assertNotNull(TaskParametersUtils.getParameters("SHELL", "{}"));
|
||||||
|
Assert.assertNotNull(TaskParametersUtils.getParameters("SQL", "{}"));
|
||||||
|
Assert.assertNotNull(TaskParametersUtils.getParameters("SUB_PROCESS", "{}"));
|
||||||
|
Assert.assertNotNull(TaskParametersUtils.getParameters("PROCEDURE", "{}"));
|
||||||
|
Assert.assertNotNull(TaskParametersUtils.getParameters("MR", "{}"));
|
||||||
|
Assert.assertNotNull(TaskParametersUtils.getParameters("SPARK", "{}"));
|
||||||
|
Assert.assertNotNull(TaskParametersUtils.getParameters("PYTHON", "{}"));
|
||||||
|
Assert.assertNotNull(TaskParametersUtils.getParameters("DEPENDENT", "{}"));
|
||||||
|
Assert.assertNotNull(TaskParametersUtils.getParameters("FLINK", "{}"));
|
||||||
|
Assert.assertNotNull(TaskParametersUtils.getParameters("HTTP", "{}"));
|
||||||
|
}
|
||||||
|
}
|
@ -93,6 +93,10 @@ public class ParamUtils {
|
|||||||
* @return Map of converted
|
* @return Map of converted
|
||||||
*/
|
*/
|
||||||
public static Map<String,String> convert(Map<String,Property> paramsMap){
|
public static Map<String,String> convert(Map<String,Property> paramsMap){
|
||||||
|
if(paramsMap == null){
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
Map<String,String> map = new HashMap<>();
|
Map<String,String> map = new HashMap<>();
|
||||||
Iterator<Map.Entry<String, Property>> iter = paramsMap.entrySet().iterator();
|
Iterator<Map.Entry<String, Property>> iter = paramsMap.entrySet().iterator();
|
||||||
while (iter.hasNext()){
|
while (iter.hasNext()){
|
||||||
|
@ -27,7 +27,6 @@ import org.apache.dolphinscheduler.common.process.HttpProperty;
|
|||||||
import org.apache.dolphinscheduler.common.process.Property;
|
import org.apache.dolphinscheduler.common.process.Property;
|
||||||
import org.apache.dolphinscheduler.common.task.AbstractParameters;
|
import org.apache.dolphinscheduler.common.task.AbstractParameters;
|
||||||
import org.apache.dolphinscheduler.common.task.http.HttpParameters;
|
import org.apache.dolphinscheduler.common.task.http.HttpParameters;
|
||||||
import org.apache.dolphinscheduler.common.utils.Bytes;
|
|
||||||
import org.apache.dolphinscheduler.common.utils.DateUtils;
|
import org.apache.dolphinscheduler.common.utils.DateUtils;
|
||||||
import org.apache.dolphinscheduler.common.utils.ParameterUtils;
|
import org.apache.dolphinscheduler.common.utils.ParameterUtils;
|
||||||
import org.apache.dolphinscheduler.common.utils.SpringApplicationContext;
|
import org.apache.dolphinscheduler.common.utils.SpringApplicationContext;
|
||||||
@ -50,6 +49,7 @@ import org.apache.http.util.EntityUtils;
|
|||||||
import org.slf4j.Logger;
|
import org.slf4j.Logger;
|
||||||
|
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
|
import java.nio.charset.StandardCharsets;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
@ -176,7 +176,7 @@ public class HttpTask extends AbstractTask {
|
|||||||
if (entity == null) {
|
if (entity == null) {
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
String webPage = EntityUtils.toString(entity, Bytes.UTF8_ENCODING);
|
String webPage = EntityUtils.toString(entity, StandardCharsets.UTF_8.name());
|
||||||
return webPage;
|
return webPage;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -264,7 +264,7 @@ public class HttpTask extends AbstractTask {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
StringEntity postingString = new StringEntity(jsonParam.toString(), Charsets.UTF_8);
|
StringEntity postingString = new StringEntity(jsonParam.toString(), Charsets.UTF_8);
|
||||||
postingString.setContentEncoding(Bytes.UTF8_ENCODING);
|
postingString.setContentEncoding(StandardCharsets.UTF_8.name());
|
||||||
postingString.setContentType(APPLICATION_JSON);
|
postingString.setContentType(APPLICATION_JSON);
|
||||||
builder.setEntity(postingString);
|
builder.setEntity(postingString);
|
||||||
}
|
}
|
||||||
|
@ -0,0 +1,135 @@
|
|||||||
|
/*
|
||||||
|
* Licensed to the Apache Software Foundation (ASF) under one or more
|
||||||
|
* contributor license agreements. See the NOTICE file distributed with
|
||||||
|
* this work for additional information regarding copyright ownership.
|
||||||
|
* The ASF licenses this file to You under the Apache License, Version 2.0
|
||||||
|
* (the "License"); you may not use this file except in compliance with
|
||||||
|
* the License. You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package org.apache.dolphinscheduler.server.utils;
|
||||||
|
|
||||||
|
import com.alibaba.fastjson.JSON;
|
||||||
|
import org.apache.dolphinscheduler.common.enums.CommandType;
|
||||||
|
import org.apache.dolphinscheduler.common.enums.DataType;
|
||||||
|
import org.apache.dolphinscheduler.common.enums.Direct;
|
||||||
|
import org.apache.dolphinscheduler.common.process.Property;
|
||||||
|
import org.junit.Before;
|
||||||
|
import org.junit.Test;
|
||||||
|
import org.slf4j.Logger;
|
||||||
|
import org.slf4j.LoggerFactory;
|
||||||
|
|
||||||
|
import java.util.Calendar;
|
||||||
|
import java.util.Date;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
|
import static org.junit.Assert.assertEquals;
|
||||||
|
import static org.junit.Assert.assertNull;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Test ParamUtils
|
||||||
|
*/
|
||||||
|
public class ParamUtilsTest {
|
||||||
|
|
||||||
|
private static final Logger logger = LoggerFactory.getLogger(ParamUtilsTest.class);
|
||||||
|
|
||||||
|
//Define global variables
|
||||||
|
public Map<String, Property> globalParams = new HashMap<>();
|
||||||
|
|
||||||
|
public Map<String, String> globalParamsMap = new HashMap<>();
|
||||||
|
|
||||||
|
public Map<String, Property> localParams = new HashMap<>();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Init params
|
||||||
|
* @throws Exception
|
||||||
|
*/
|
||||||
|
@Before
|
||||||
|
public void setUp() throws Exception {
|
||||||
|
|
||||||
|
Property property = new Property();
|
||||||
|
property.setProp("global_param");
|
||||||
|
property.setDirect(Direct.IN);
|
||||||
|
property.setType(DataType.VARCHAR);
|
||||||
|
property.setValue("${system.biz.date}");
|
||||||
|
globalParams.put("global_param", property);
|
||||||
|
|
||||||
|
globalParamsMap.put("global_param", "${system.biz.date}");
|
||||||
|
|
||||||
|
Property localProperty = new Property();
|
||||||
|
localProperty.setProp("local_param");
|
||||||
|
localProperty.setDirect(Direct.IN);
|
||||||
|
localProperty.setType(DataType.VARCHAR);
|
||||||
|
localProperty.setValue("${global_param}");
|
||||||
|
localParams.put("local_param", localProperty);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Test convert
|
||||||
|
*/
|
||||||
|
@Test
|
||||||
|
public void testConvert() {
|
||||||
|
|
||||||
|
//The expected value
|
||||||
|
String expected = "{\"global_param\":{\"direct\":\"IN\",\"prop\":\"global_param\",\"type\":\"VARCHAR\",\"value\":\"20191229\"},\"local_param\":{\"direct\":\"IN\",\"prop\":\"local_param\",\"type\":\"VARCHAR\",\"value\":\"20191229\"}}";
|
||||||
|
|
||||||
|
//The expected value when globalParams is null but localParams is not null
|
||||||
|
String expected1 = "{\"local_param\":{\"direct\":\"IN\",\"prop\":\"local_param\",\"type\":\"VARCHAR\",\"value\":\"20191229\"}}";
|
||||||
|
|
||||||
|
//Define expected date , the month is 0-base
|
||||||
|
Calendar calendar = Calendar.getInstance();
|
||||||
|
calendar.set(2019,11,30);
|
||||||
|
Date date = calendar.getTime();
|
||||||
|
|
||||||
|
//Invoke convert
|
||||||
|
Map<String, Property> paramsMap = ParamUtils.convert(globalParams, globalParamsMap, localParams, CommandType.START_PROCESS, date);
|
||||||
|
String result = JSON.toJSONString(paramsMap);
|
||||||
|
assertEquals(expected, result);
|
||||||
|
|
||||||
|
for (Map.Entry<String, Property> entry : paramsMap.entrySet()) {
|
||||||
|
|
||||||
|
String key = entry.getKey();
|
||||||
|
Property prop = entry.getValue();
|
||||||
|
logger.info(key + " : " + prop.getValue());
|
||||||
|
}
|
||||||
|
|
||||||
|
//Invoke convert with null globalParams
|
||||||
|
Map<String, Property> paramsMap1 = ParamUtils.convert(null, globalParamsMap, localParams, CommandType.START_PROCESS, date);
|
||||||
|
String result1 = JSON.toJSONString(paramsMap1);
|
||||||
|
assertEquals(expected1, result1);
|
||||||
|
|
||||||
|
//Null check, invoke convert with null globalParams and null localParams
|
||||||
|
Map<String, Property> paramsMap2 = ParamUtils.convert(null, globalParamsMap, null, CommandType.START_PROCESS, date);
|
||||||
|
assertNull(paramsMap2);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Test the overload method of convert
|
||||||
|
*/
|
||||||
|
@Test
|
||||||
|
public void testConvert1() {
|
||||||
|
|
||||||
|
//The expected value
|
||||||
|
String expected = "{\"global_param\":\"${system.biz.date}\"}";
|
||||||
|
|
||||||
|
//Invoke convert
|
||||||
|
Map<String, String> paramsMap = ParamUtils.convert(globalParams);
|
||||||
|
String result = JSON.toJSONString(paramsMap);
|
||||||
|
assertEquals(expected, result);
|
||||||
|
|
||||||
|
logger.info(result);
|
||||||
|
|
||||||
|
//Null check
|
||||||
|
Map<String, String> paramsMap1 = ParamUtils.convert(null);
|
||||||
|
assertNull(paramsMap1);
|
||||||
|
}
|
||||||
|
}
|
2
pom.xml
2
pom.xml
@ -675,12 +675,14 @@
|
|||||||
<include>**/api/service/WorkerGroupServiceTest.java</include>
|
<include>**/api/service/WorkerGroupServiceTest.java</include>
|
||||||
<include>**/api/service/AlertGroupServiceTest.java</include>
|
<include>**/api/service/AlertGroupServiceTest.java</include>
|
||||||
<include>**/api/service/ProjectServiceTest.java</include>
|
<include>**/api/service/ProjectServiceTest.java</include>
|
||||||
|
<include>**/api/service/UdfFuncServiceTest.java</include>
|
||||||
<include>**/alert/utils/ExcelUtilsTest.java</include>
|
<include>**/alert/utils/ExcelUtilsTest.java</include>
|
||||||
<include>**/alert/utils/FuncUtilsTest.java</include>
|
<include>**/alert/utils/FuncUtilsTest.java</include>
|
||||||
<include>**/alert/utils/JSONUtilsTest.java</include>
|
<include>**/alert/utils/JSONUtilsTest.java</include>
|
||||||
<include>**/alert/utils/PropertyUtilsTest.java</include>
|
<include>**/alert/utils/PropertyUtilsTest.java</include>
|
||||||
<include>**/server/utils/SparkArgsUtilsTest.java</include>
|
<include>**/server/utils/SparkArgsUtilsTest.java</include>
|
||||||
<include>**/server/utils/FlinkArgsUtilsTest.java</include>
|
<include>**/server/utils/FlinkArgsUtilsTest.java</include>
|
||||||
|
<include>**/server/utils/ParamUtilsTest.java</include>
|
||||||
<include>**/dao/mapper/AccessTokenMapperTest.java</include>
|
<include>**/dao/mapper/AccessTokenMapperTest.java</include>
|
||||||
<include>**/dao/mapper/AlertGroupMapperTest.java</include>
|
<include>**/dao/mapper/AlertGroupMapperTest.java</include>
|
||||||
<include>**/dao/mapper/AlertMapperTest.java</include>
|
<include>**/dao/mapper/AlertMapperTest.java</include>
|
||||||
|
Loading…
Reference in New Issue
Block a user